Viewing file:  prober.py (2 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# Copyright 2015 Canonical, Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from probert.storage import Storage
from probert.network import Network
class Prober():
def __init__(self, options, results={}):
self.options = options
self.results = results
''' build a list of probe_ methods of this class,
excluding probe_all so we don't recurse.
This allows probe_all method to call all probe_
methods as we add it without maintaining a list
in the code.
'''
exclude = ['probe_all']
self.probes = [getattr(self, fn) for fn in
filter(lambda x: callable(getattr(self, x)) and
x.startswith('probe_') and
x not in exclude, dir(self))]
def probe(self):
# find out what methods to call by looking options
for fn in [x for x in dir(self.options)
if self.options.__getattribute__(x) is True]:
getattr(self, fn)()
def probe_all(self):
for fn in self.probes:
fn()
def probe_storage(self):
storage = Storage()
results = storage.probe()
self.results['storage'] = results
def probe_network(self):
network = Network()
results = network.probe()
self.results['network'] = results
def get_results(self):
return self.results
|