Sourceloader should import dynamically
This commit is contained in:
parent
071018cbac
commit
edc91c2279
@ -8,7 +8,7 @@ from FourmiCrawler.sources.source import Source
|
|||||||
class SourceLoader:
|
class SourceLoader:
|
||||||
sources = []
|
sources = []
|
||||||
|
|
||||||
def __init__(self, rel_dir="FourmiCrawler/sources"):
|
def __init__(self, rel_dir="../FourmiCrawler/sources"):
|
||||||
"""
|
"""
|
||||||
The initiation of a SourceLoader, selects and indexes a directory for usable sources.
|
The initiation of a SourceLoader, selects and indexes a directory for usable sources.
|
||||||
:param rel_dir: A relative path to a directory.
|
:param rel_dir: A relative path to a directory.
|
||||||
@ -18,7 +18,7 @@ class SourceLoader:
|
|||||||
known_parser = set()
|
known_parser = set()
|
||||||
|
|
||||||
for py in [f[:-3] for f in os.listdir(path) if f.endswith('.py') and f != '__init__.py']:
|
for py in [f[:-3] for f in os.listdir(path) if f.endswith('.py') and f != '__init__.py']:
|
||||||
mod = __import__('.'.join([rel_dir.replace("/", "."), py]), fromlist=[py])
|
mod = __import__('.'.join([rel_dir.replace("../", "").replace("/", "."), py]), fromlist=[py])
|
||||||
classes = [getattr(mod, x) for x in dir(mod) if inspect.isclass(getattr(mod, x))]
|
classes = [getattr(mod, x) for x in dir(mod) if inspect.isclass(getattr(mod, x))]
|
||||||
for cls in classes:
|
for cls in classes:
|
||||||
if issubclass(cls, Source) and cls not in known_parser:
|
if issubclass(cls, Source) and cls not in known_parser:
|
||||||
|
Reference in New Issue
Block a user