[all] pylint refinements
Improve code by bringing up the pylint score a bit. see #23
This commit is contained in:
parent
252260c249
commit
0c7884d170
4 changed files with 10 additions and 8 deletions
|
@ -7,7 +7,7 @@ import importlib
|
|||
import bumblebee.error
|
||||
import bumblebee.modules
|
||||
|
||||
def modules():
|
||||
def all_modules():
|
||||
"""Return a list of available modules"""
|
||||
result = []
|
||||
path = os.path.dirname(bumblebee.modules.__file__)
|
||||
|
@ -28,7 +28,7 @@ class Module(object):
|
|||
def __init__(self, engine, config={}, widgets=[]):
|
||||
self.name = self.__module__.split(".")[-1]
|
||||
self._config = config
|
||||
if not "name" in self._config:
|
||||
if "name" not in self._config:
|
||||
self._config["name"] = self.name
|
||||
self._widgets = []
|
||||
if widgets:
|
||||
|
@ -68,7 +68,7 @@ class Engine(object):
|
|||
|
||||
def load_module(self, module_name, config_name=None):
|
||||
"""Load specified module and return it as object"""
|
||||
if config_name == None:
|
||||
if config_name is None:
|
||||
config_name = module_name
|
||||
try:
|
||||
module = importlib.import_module("bumblebee.modules.{}".format(module_name))
|
||||
|
|
|
@ -6,13 +6,16 @@ this module
|
|||
"""
|
||||
|
||||
class Store(object):
|
||||
"""Interface for storing and retrieving simple values"""
|
||||
def __init__(self):
|
||||
self._data = {}
|
||||
|
||||
def set(self, key, value):
|
||||
"""Set 'key' to 'value', overwriting 'key' if it exists already"""
|
||||
self._data[key] = value
|
||||
|
||||
def get(self, key, default=None):
|
||||
"""Return the current value of 'key', or 'default' if 'key' is not set"""
|
||||
return self._data.get(key, default)
|
||||
|
||||
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|
||||
|
|
|
@ -3,8 +3,7 @@
|
|||
import unittest
|
||||
import importlib
|
||||
|
||||
from bumblebee.modules.cpu import Module
|
||||
from bumblebee.engine import modules
|
||||
from bumblebee.engine import all_modules
|
||||
from bumblebee.config import Config
|
||||
from tests.util import assertWidgetAttributes, MockEngine
|
||||
|
||||
|
@ -13,9 +12,9 @@ class TestGenericModules(unittest.TestCase):
|
|||
engine = MockEngine()
|
||||
config = Config()
|
||||
self.objects = {}
|
||||
for mod in modules():
|
||||
for mod in all_modules():
|
||||
cls = importlib.import_module("bumblebee.modules.{}".format(mod["name"]))
|
||||
self.objects[mod["name"]] = getattr(cls, "Module")(engine, { "config": config })
|
||||
self.objects[mod["name"]] = getattr(cls, "Module")(engine, {"config": config})
|
||||
|
||||
def test_widgets(self):
|
||||
for mod in self.objects:
|
||||
|
|
|
@ -19,7 +19,7 @@ class TestModule(unittest.TestCase):
|
|||
self.anyConfigName = "cfg"
|
||||
self.anotherConfigName = "cfg2"
|
||||
self.anyModule = Module(engine=None, widgets=self.widget, config={
|
||||
"name": self.anyConfigName, "config": self.config
|
||||
"name": self.anyConfigName, "config": self.config
|
||||
})
|
||||
self.anotherModule = Module(engine=None, widgets=self.widget, config={
|
||||
"name": self.anotherConfigName, "config": self.config
|
||||
|
|
Loading…
Reference in a new issue