document-processor/app/db/init_db.py

19 lines
593 B
Python

from app.db.base import Base
from app.db.session import engine
# Import models so Base.metadata knows about all tables
from app.models.document import Document # noqa: F401
from app.models.document_version import DocumentVersion # noqa: F401
from app.models.text_version import TextVersion # noqa: F401
from app.models.extracted_field import ExtractedField # noqa: F401
from app.models.layer1_candidate import Layer1Candidate # noqa: F401
def init_db() -> None:
Base.metadata.create_all(bind=engine)
if __name__ == "__main__":
init_db()
print("Database tables created.")