Commit d2f393e8 authored by Simon Will's avatar Simon Will
Browse files

Add config switch for populating the database

parent c3b54d98

allzweckmesser/config.py

+2 −0
Original line number Diff line number Diff line
@@ -12,3 +12,5 @@ MACRONS_FILE = os.environ.get('MACRONS_FILE', 
MORPHEUS_DIR = os.environ.get('MORPHEUS_DIR',

                              os.path.join(ROOT, 'morpheus'))

MODE = os.environ.get('AZM_MODE', 'run')



POPULATE_DATABASE = False

allzweckmesser/wordlist.py

+4 −1
Original line number Diff line number Diff line
@@ -38,7 +38,8 @@ class WordList: 


    def __init__(self, form_analyses: Dict[str, Set[FormAnalysis]] = 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.



        :param form_analyses: Mapping of forms to form analyses.
@@ -49,6 +50,8 @@ class WordList: 
        self.unknown_forms = unknown_forms or set()

        self.session_factory = session_factory

        self._session = self.session_factory()

        if populate_database:

            self.populate_database()



    def get_morphtags(self, form: str) -> Set[str]:

        """Get the morphtags of a form.