collectors_dir settings

Author: DevAlone

.gitignore

@@ -9,4 +9,4 @@ migrations
 alembic.ini
 alembic
 env/
-local_checkers/
+local/

collectors_list.py

@@ -1,13 +1,14 @@
+from models import session, CollectorState
+from proxy_py import settings
+
 import os
 import importlib.util
 
-from models import session, CollectorState
-
 
 collectors = {}
 
 
-for root, dirs, files in os.walk("collectors"):
+for root, dirs, files in os.walk(settings.COLLECTORS_DIR):
     for file in files:
         if file.endswith(".py"):
             file_path = os.path.join(root, file)
@@ -43,18 +44,18 @@
 #     return collectorState
 
 
-def get_collector_of_module_name(module_name : str):
+def get_collector_of_module_name(module_name: str):
     if module_name not in collectors:
         raise CollectorNotFoundException()
 
     return collectors[module_name]
 
 
-async def load_collector(state : CollectorState):
+async def load_collector(state: CollectorState):
     collector = get_collector_of_module_name(state.identifier)
     await collector.load_state(state)
     return collector
 
 
 class CollectorNotFoundException(BaseException):
-    pass
+    pass

proxy_py/_settings.py

@@ -13,6 +13,8 @@
 
 DEBUG = True
 
+COLLECTORS_DIR = 'collectors'
+
 # fetcher settings
 
 CONCURRENT_TASKS_COUNT = 128