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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
# First Aid Kit - diagnostic and repair tool for Linux
# Copyright (C) 2007 Martin Sivak <msivak@redhat.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
from pyfirstaidkit.plugins import Plugin,Flow
from pyfirstaidkit.reporting import PLUGIN
from pyfirstaidkit.returns import *
from pyfirstaidkit.issue import SimpleIssue
from pyfirstaidkit.configuration import Config
from pyfirstaidkit.utils import spawnvch
import os
class Sample1Plugin(Plugin):
"""Discover information about the system"""
name = "Discovery"
description = "Discover properties of the system"
version = "0.0.1"
author = "Martin Sivak"
flows = Flow.init(Plugin)
flows["fix"] = flows["diagnose"]
def __init__(self, *args, **kwargs):
Plugin.__init__(self, *args, **kwargs)
self._issue = SimpleIssue(self.name, "Discovering system properties failed")
def prepare(self):
self._issue.set(reporting = self._reporting, origin = self, level = PLUGIN)
self._result=ReturnSuccess
def diagnose(self):
#architecture and cpus
(unamestdout, unamestderr) = spawnvch(executable = "/bin/uname", args = ["uname", "-a"], chroot = Config.system.root).communicate("")
self._info.uname = unamestdout.split("\n")[0]
#memory
(freestdout, freestderr) = spawnvch(executable = "/usr/bin/free", args = ["free"], chroot = Config.system.root).communicate("")
freedata = freestdout.split("\n")
self._info.memory = freedata[1].split()[1]
self._info.swap = freedata[3].split()[1]
#pci
pcilist = []
(lspcistdout, lspcistderr) = spawnvch(executable = "/sbin/lspci", args = ["lspci"], chroot = Config.system.root).communicate("")
for l in lspcistdout.split("\n"):
try:
(id, name) = l.split(" ", 1)
setattr(self._info, "_".join(["pci", id]), name)
pcilist.append(id)
except:
pass
self._info.pci = " ".join(pcilist)
#usb
if os.path.exists(os.path.join(Config.system.root, "/proc/bus/usb")):
self._info.usb = "True"
else:
self._info.usb = "False"
#scsi
if os.path.exists(os.path.join(Config.system.root, "/proc/scsi/device_info")):
self._info.scsi = "True"
else:
self._info.scsi = "False"
#ide
if os.path.exists(os.path.join(Config.system.root, "/proc/ide")):
self._info.ide = "True"
else:
self._info.ide = "False"
#partitions
partitionlist = []
for l in open("/proc/partitions").readlines()[2:]:
try:
(major, minor, blocks, name) = l.split()
if name.startswith("ram"):
continue
setattr(self._info, "_".join(["partition", name]), blocks)
partitionlist.append(name)
except:
continue
self._info.partition = " ".join(partitionlist)
#net
self._dependencies.provide("discovery")
self._issue.set(checked = True, happened = False, reporting = self._reporting, origin = self, level = PLUGIN)
self._result=ReturnSuccess
def clean(self):
self._result=ReturnSuccess
def get_plugin():
return Sample1Plugin
|