python 擷取記憶體的狀态

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

<code>#!/usr/bin/env python</code>

<code>import</code> <code>subprocess</code>

<code>try</code><code>:</code>

<code>    </code><code>import</code> <code>simplejson as json</code>

<code>except</code><code>:</code>

<code>    </code><code>import</code> <code>json</code>

<code>memory_status </code><code>=</code> <code>{} </code>

<code>p </code><code>=</code> <code>subprocess.Popen(</code><code>"dmidecode --type system | perl -alne '/Manufacturer:\s(.*)/ and print $1'"</code><code>, shell</code><code>=</code><code>True</code><code>, stdout</code><code>=</code><code>subprocess.PIPE)</code>

<code>vender </code><code>=</code> <code>p.stdout.readline().strip()</code>

<code>if</code> <code>vender </code><code>=</code><code>=</code> <code>'HP'</code><code>:</code>

<code>    </code><code>mem_slot </code><code>=</code> <code>[]</code>

<code>    </code><code>mem_status </code><code>=</code> <code>[]</code>

<code>    </code><code>p </code><code>=</code> <code>subprocess.call(</code><code>"rpm -qa | grep hp-health"</code><code>, shell</code><code>=</code><code>True</code><code>, stdout</code><code>=</code><code>subprocess.PIPE)</code>

<code>    </code><code>if</code> <code>p !</code><code>=</code> <code>0</code><code>:</code>

<code>        </code><code>subprocess.call(</code><code>"rpm -ivh http://downloads.linux.hp.com/downloads/ManagementComponentPack/centos/6/x86_64/9.40/hp-health-9.40-1602.44.rhel6.x86_64.rpm"</code><code>, shell</code><code>=</code><code>True</code><code>, stdout</code><code>=</code><code>subprocess.PIPE, stderr</code><code>=</code><code>subprocess.PIPE)</code>

<code>    </code><code>p </code><code>=</code> <code>subprocess.Popen(</code><code>"hpasmcli -s 'show dimm'"</code><code>, shell</code><code>=</code><code>True</code><code>, stdout</code><code>=</code><code>subprocess.PIPE)</code>

<code>    </code><code>for</code> <code>line </code><code>in</code> <code>p.stdout.readlines():</code>

<code>        </code><code>a </code><code>=</code> <code>{}</code>

<code>        </code><code>data </code><code>=</code> <code>line.strip().split(</code><code>":"</code><code>)</code>

<code>        </code><code>if</code> <code>len</code><code>(data) </code><code>=</code><code>=</code> <code>2</code><code>:</code>

<code>            </code><code>a[data[</code><code>0</code><code>].strip()] </code><code>=</code> <code>data[</code><code>1</code><code>].strip()</code>

<code>        </code> 

<code>        </code><code>if</code> <code>'Status'</code> <code>in</code> <code>a.keys():</code>

<code>            </code><code>status </code><code>=</code> <code>a[</code><code>'Status'</code><code>]</code>

<code>                        </code><code>if</code> <code>status </code><code>=</code><code>=</code> <code>"Ok"</code><code>:</code>

<code>                                </code><code>status </code><code>=</code> <code>0</code>

<code>                                </code><code>mem_status.append(status)</code>

<code>    </code><code>length </code><code>=</code> <code>len</code><code>(mem_status)</code>

<code>    </code><code>if</code> <code>length:</code>

<code>        </code><code>for</code> <code>i </code><code>in</code> <code>range</code><code>(</code><code>0</code><code>, length):</code>

<code>            </code><code>slot </code><code>=</code> <code>"slot_%d"</code>  <code>%</code> <code>(i)</code>

<code>            </code><code>mem_slot.append(slot)</code>

<code>    </code><code>memory_status </code><code>=</code> <code>dict</code><code>(</code><code>zip</code><code>(mem_slot, mem_status))</code>

<code>else</code><code>:</code>

<code>    </code><code>p </code><code>=</code> <code>subprocess.Popen(</code><code>"omreport chassis Memory"</code><code>, shell</code><code>=</code><code>True</code><code>, stdout</code><code>=</code><code>subprocess.PIPE)</code>

<code>        </code><code>if</code> <code>'Index'</code> <code>in</code> <code>a.keys():</code>

<code>                </code><code>if</code> <code>a[</code><code>'Index'</code><code>]:</code>

<code>                </code><code>slot </code><code>=</code> <code>"slot_"</code> <code>+</code> <code>a[</code><code>'Index'</code><code>]</code>

<code>                </code><code>mem_slot.append(slot)      </code>

<code>        </code><code>elif</code> <code>'Status'</code> <code>in</code> <code>a.keys():</code>

<code>            </code><code>if</code> <code>status !</code><code>=</code> <code>"Unknown"</code> <code>and</code> <code>status </code><code>=</code><code>=</code> <code>"Ok"</code><code>:</code>

<code>                </code><code>status </code><code>=</code> <code>0</code>

<code>                </code><code>mem_status.append(status)</code>

<code>            </code> 

<code>print</code> <code>json.dumps(memory_status, indent</code><code>=</code><code>4</code><code>)</code>

本文轉自 Art_Hero 51CTO部落格，原文連結：http://blog.51cto.com/curran/1675331，如需轉載請自行聯系原作者