diff --git a/portal/__init__.py b/portal/__init__.py index 642054e1eae3ce4ca477b4d960c33113987b6bb5..99937d5f6c96730cb3a75a2f8c00ab45734ba5c5 100644 --- a/portal/__init__.py +++ b/portal/__init__.py @@ -9,7 +9,7 @@ from flask_sqlalchemy import SQLAlchemy from .db_config import get_database_uri, get_uri_for_sqlite -db = SQLAlchemy() +db: SQLAlchemy = SQLAlchemy() from .model import * diff --git a/portal/model/FlaskConfigEntry.py b/portal/model/FlaskConfigEntry.py index 378aed8320af2ee381777e3eb8bcbee363bd60b1..bfd3cd9243325bfe148e217d6fbe76d5e915109e 100644 --- a/portal/model/FlaskConfigEntry.py +++ b/portal/model/FlaskConfigEntry.py @@ -1,9 +1,12 @@ from flask import Flask +from flask_sqlalchemy import DefaultMeta from portal import db +BaseModel: DefaultMeta = db.Model -class FlaskConfigEntry(db.Model): + +class FlaskConfigEntry(BaseModel): """ A configuration entry for Flask (a key-value pair) as persisted in a database """ diff --git a/portal/model/Retrievable.py b/portal/model/Retrievable.py index 1b7c6495c41b7e1aace3d3628fa627b34ae3b56a..b15aa785dfc5b6f29957927fb13c2beda329b501 100644 --- a/portal/model/Retrievable.py +++ b/portal/model/Retrievable.py @@ -1,12 +1,16 @@ from abc import abstractmethod from typing import Any, List, Optional, Self +from flask_sqlalchemy import DefaultMeta +from sqlalchemy import Column from flask import g from portal import db +BaseModel: DefaultMeta = db.Model -class Retrievable(db.Model): + +class Retrievable(BaseModel): """ A type whose instances can be retrieved from the database. @@ -17,7 +21,7 @@ class Retrievable(db.Model): __abstract__ = True @classmethod - def get_canonical_order_column(cls) -> Optional[db.Column]: + def get_canonical_order_column(cls) -> Optional[Column]: """ Return the colum by which instances of this type should be canonically ordered when retrieving them all from the database.