wip
This commit is contained in:
@@ -1,126 +0,0 @@
|
||||
#!/usr/bin/python3
|
||||
|
||||
import os
|
||||
from configparser import ConfigParser
|
||||
import logging
|
||||
from systemd import journal
|
||||
import inspect
|
||||
from importlib import import_module
|
||||
from commands import Commands
|
||||
from threading import Lock
|
||||
|
||||
class CommandsLoader:
|
||||
def __init__(self, config: ConfigParser, log: logging.Handler, modpath: str):
|
||||
self.config = config
|
||||
self.modpath = modpath
|
||||
Commands.defmod = self.config['general'].get('default')
|
||||
self.log = log
|
||||
self.lock = Lock()
|
||||
self._load_commands()
|
||||
self._set_cmods()
|
||||
self._load_dependencies()
|
||||
|
||||
def _load_commands(self):
|
||||
cmds = {}
|
||||
with os.scandir(self.modpath) as it:
|
||||
for entry in it:
|
||||
if entry.name.startswith('cmds_') and entry.name.endswith('.py') and entry.is_file():
|
||||
obj = self._load_module(entry.name[:-3])
|
||||
if not obj is None:
|
||||
nmod = obj.nickname
|
||||
if nmod is None:
|
||||
nmod = (entry.name[:-3])[5:]
|
||||
cmds.update({nmod: obj})
|
||||
self.log.info('module "{}" loaded'.format(nmod))
|
||||
else:
|
||||
self.log.error(
|
||||
'The module "{}" could not be loaded'.format(entry.name[:-3]))
|
||||
self.cmods = cmds
|
||||
|
||||
def _load_module(self, name):
|
||||
module = import_module("."+name, 'commands')
|
||||
members = inspect.getmembers(module, inspect.isclass)
|
||||
conf = None
|
||||
if name[5:] in self.config.sections():
|
||||
conf = self.config[name[5:]]
|
||||
|
||||
obj = None
|
||||
for n, c in members:
|
||||
if issubclass(c, Commands) and (n != 'Commands'):
|
||||
obj = c(conf, self.log)
|
||||
obj.log = self.log
|
||||
obj.lock = self.lock
|
||||
break
|
||||
|
||||
return obj
|
||||
|
||||
def _set_cmods(self):
|
||||
for k, v in self.cmods.items():
|
||||
v.cmods = self.cmods
|
||||
|
||||
def _load_dependencies(self):
|
||||
for k, v in self.cmods.items():
|
||||
if hasattr(v, "dependencies"):
|
||||
for p in v.dependencies:
|
||||
if p in self.cmods.keys():
|
||||
setattr(v, p, self.cmods[p])
|
||||
else:
|
||||
self.log.error(
|
||||
'Dependency "{}" of module "{}" could not be satisfied'.format(p, k))
|
||||
|
||||
def start(self):
|
||||
for k, v in self.cmods.items():
|
||||
if v.threaded:
|
||||
v.start()
|
||||
|
||||
def join(self):
|
||||
for k, v in self.cmods.items():
|
||||
if v.threaded:
|
||||
v.join()
|
||||
|
||||
|
||||
class AppEngine:
|
||||
def __init__(self, app_name: str, conf_file : str="", mod_dir: str="", log_file: str="", debug: bool=False) -> None:
|
||||
self.app_name = app_name
|
||||
self.conf = ConfigParser()
|
||||
if conf_file != '':
|
||||
self.parse_config(conf_file)
|
||||
self.def_log(log_file)
|
||||
|
||||
if debug:
|
||||
self.log.setLevel(logging.DEBUG)
|
||||
else:
|
||||
self.log.setLevel(logging.WARNING)
|
||||
|
||||
def parse_config(self, cf: str):
|
||||
if os.path.exists(cf) and os.path.isfile(cf):
|
||||
self.conf.read(cf)
|
||||
else:
|
||||
raise Exception('Configuration file not found')
|
||||
|
||||
def def_log(self, log_file):
|
||||
self.log = logging.getLogger(self.app_name)
|
||||
if log_file != '':
|
||||
fname = log_file
|
||||
is_writeable = False
|
||||
|
||||
if not os.path.isabs(fname):
|
||||
fname = os.path.join(os.getcwd(), fname)
|
||||
|
||||
if os.access(os.path.dirname(fname), os.W_OK):
|
||||
is_writeable = True
|
||||
|
||||
if is_writeable:
|
||||
self.log.addHandler(logging.FileHandler(fname))
|
||||
else:
|
||||
self.log.addHandler(journal.JournalHandler())
|
||||
self.log.error('No write permissions: "{}"'.format(fname))
|
||||
|
||||
else:
|
||||
self.log.addHandler(journal.JournalHandler())
|
||||
|
||||
|
||||
def exec(self, modpath: str=""):
|
||||
cl = CommandsLoader(self.conf, self.log, modpath)
|
||||
cl.start()
|
||||
cl.join()
|
||||
Reference in New Issue
Block a user