diff options
Diffstat (limited to 'archivist/cli.py')
| -rw-r--r-- | archivist/cli.py | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/archivist/cli.py b/archivist/cli.py index 4316705..f73a7a4 100644 --- a/archivist/cli.py +++ b/archivist/cli.py @@ -19,14 +19,11 @@ def db(): @db.command() def init(): """Initialize the database, if not done already.""" - from .model import create_tables, Prefix, db + from .model import create_tables, db + from .prefixes import register_prefixes create_tables() with db.atomic(): - Prefix.create(name='year', builtin=True) - Prefix.create(name='month', builtin=True) - Prefix.create(name='day', builtin=True) - Prefix.create(name='date', builtin=True, pseudo=True) - Prefix.create(name='direction', builtin=True, pseudo=True) + register_prefixes() @db.command() @click.confirmation_option(prompt="Are you sure you want to drop the database?") |