Implement BaseModel class, TimestampMixin, TzDateTime type; cleanup Task model

migrations/env.py

@@ -26,16 +26,22 @@ def process_revision_directives(_context, _revision, directives):
         directives[:] = []
 
 
+context_parameters = {
+    'target_metadata': BaseModel.metadata,
+    'process_revision_directives': process_revision_directives,
+    'user_module_prefix': 'tofu_types.',
+}
+
+
 def run_migrations_offline():
     """
     Run migrations in 'offline' mode, which does not require an actual database engine and can be used to generate SQL scripts.
     """
     context.configure(
         url=app.config.sqlalchemy_database_uri,
-        target_metadata=BaseModel.metadata,
-        process_revision_directives=process_revision_directives,
         literal_binds=True,
         dialect_opts={"paramstyle": "named"},
+        **context_parameters,
     )
 
     with context.begin_transaction():
@@ -49,8 +55,7 @@ def run_migrations_online():
     with db.engine.connect() as connection:
         context.configure(
             connection=connection,
-            target_metadata=BaseModel.metadata,
-            process_revision_directives=process_revision_directives,
+            **context_parameters,
         )
 
         with context.begin_transaction():