refactoring and extension loading fix

This commit is contained in:
Vinicius Moreira
2019-08-12 18:24:27 -03:00
parent 1951a545c8
commit f432bfdd06
2 changed files with 28 additions and 25 deletions

View File

@@ -16,7 +16,7 @@ SUGGESTIONS_LIMIT = 6
class GenericApplicationManager(ApplicationManager): class GenericApplicationManager(ApplicationManager):
def __init__(self, managers: List[ApplicationManager], disk_loader_factory: DiskCacheLoaderFactory, app_args: Namespace, locale_keys: dict): def __init__(self, managers: List[ApplicationManager], disk_loader_factory: DiskCacheLoaderFactory, app_args: Namespace, locale_keys: dict):
super(GenericApplicationManager, self).__init__(app_args=app_args, app_cache=None, locale_keys=locale_keys, fpakman_root_dir=ROOT_DIR, http_client=None) super(GenericApplicationManager, self).__init__(app_args=app_args, app_cache=None, locale_keys=locale_keys, app_root_dir=ROOT_DIR, http_client=None)
self.managers = managers self.managers = managers
self.map = {m.get_app_type(): m for m in self.managers} self.map = {m.get_app_type(): m for m in self.managers}
self.disk_loader_factory = disk_loader_factory self.disk_loader_factory = disk_loader_factory

View File

@@ -8,13 +8,15 @@ from bauh_api.abstract.controller import ApplicationManager
from bauh_api.util.cache import Cache from bauh_api.util.cache import Cache
from bauh_api.util.http import HttpClient from bauh_api.util.http import HttpClient
from bauh import ROOT_DIR from bauh import ROOT_DIR, __app_name__
from bauh.util import util from bauh.util import util
ignore_modules = {'fpakman_api'} ignore_modules = {'{}_api'.format(__app_name__)}
ext_pattern = '{}_'.format(__app_name__)
def find_manager(member): def find_manager(member):
if not isinstance(member, str):
if inspect.isclass(member) and inspect.getmro(member)[1].__name__ == 'ApplicationManager': if inspect.isclass(member) and inspect.getmro(member)[1].__name__ == 'ApplicationManager':
return member return member
elif inspect.ismodule(member) and member.__name__ not in ignore_modules: elif inspect.ismodule(member) and member.__name__ not in ignore_modules:
@@ -28,10 +30,11 @@ def load_managers(caches: List[Cache], cache_map: Dict[type, Cache], locale_keys
managers = [] managers = []
for m in pkgutil.iter_modules(): for m in pkgutil.iter_modules():
if m.ispkg and m.name and m.name not in ignore_modules and m.name.startswith('fpakman_'): if m.ispkg and m.name and m.name not in ignore_modules and m.name.startswith(ext_pattern):
module = pkgutil.find_loader(m.name).load_module() module = pkgutil.find_loader(m.name).load_module()
manager_class = find_manager(module) if hasattr(module, 'controller'):
manager_class = find_manager(module.controller)
if manager_class: if manager_class:
locale_path = '{}/resources/locale'.format(module.__path__[0]) locale_path = '{}/resources/locale'.format(module.__path__[0])