Skip to content
Snippets Groups Projects
Commit d2f393e8 authored by Simon Will's avatar Simon Will
Browse files

Add config switch for populating the database

parent c3b54d98
No related branches found
No related tags found
No related merge requests found
...@@ -12,3 +12,5 @@ MACRONS_FILE = os.environ.get('MACRONS_FILE', ...@@ -12,3 +12,5 @@ MACRONS_FILE = os.environ.get('MACRONS_FILE',
MORPHEUS_DIR = os.environ.get('MORPHEUS_DIR', MORPHEUS_DIR = os.environ.get('MORPHEUS_DIR',
os.path.join(ROOT, 'morpheus')) os.path.join(ROOT, 'morpheus'))
MODE = os.environ.get('AZM_MODE', 'run') MODE = os.environ.get('AZM_MODE', 'run')
POPULATE_DATABASE = False
...@@ -38,7 +38,8 @@ class WordList: ...@@ -38,7 +38,8 @@ class WordList:
def __init__(self, form_analyses: Dict[str, Set[FormAnalysis]] = None, def __init__(self, form_analyses: Dict[str, Set[FormAnalysis]] = None,
unknown_forms: Set[str] = None, unknown_forms: Set[str] = None,
session_factory: sessionmaker = SESSION_FACTORY) -> None: session_factory: sessionmaker = SESSION_FACTORY,
populate_database: bool = POPULATE_DATABASE) -> None:
"""Initialize a WordList. """Initialize a WordList.
:param form_analyses: Mapping of forms to form analyses. :param form_analyses: Mapping of forms to form analyses.
...@@ -49,6 +50,8 @@ class WordList: ...@@ -49,6 +50,8 @@ class WordList:
self.unknown_forms = unknown_forms or set() self.unknown_forms = unknown_forms or set()
self.session_factory = session_factory self.session_factory = session_factory
self._session = self.session_factory() self._session = self.session_factory()
if populate_database:
self.populate_database()
def get_morphtags(self, form: str) -> Set[str]: def get_morphtags(self, form: str) -> Set[str]:
"""Get the morphtags of a form. """Get the morphtags of a form.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment