main.py

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from cassandra.cqlengine.management import sync_table
from routes import (access_api, dictionary_mapper_api, data_dictionary_api, data_dictionary_usl_api, configuration_api,
                    usl_data_transmission_api, site_configuration_api, user_management)
from models.models import (AccessCredentials, MappedVariables, DataDictionaries, DataDictionaryTerms,
                           USLConfig, SchedulesConfig, SiteConfig, TransmissionHistory, SchedulesLog, UniversalDictionaryConfig)
from models.usl_models import (DataDictionariesUSL, DataDictionaryTermsUSL, DictionaryChangeLog,
                               UniversalDictionaryFacilityPulls, UniversalDictionaryTokens)
from database.user_db import UserBase, user_engine, SessionLocal
from utils.user_utils import seed_default_user

app = FastAPI()

origins = [
    "*",
]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


@app.on_event("startup")
async def startup_event():
    db = SessionLocal()
    try:
        seed_default_user(db)
    finally:
        db.close()

    sync_table(AccessCredentials)
    sync_table(MappedVariables)
    sync_table(DataDictionaries)
    sync_table(DataDictionaryTerms)
    sync_table(DataDictionariesUSL)
    sync_table(DataDictionaryTermsUSL)
    sync_table(DictionaryChangeLog)
    sync_table(UniversalDictionaryConfig)
    sync_table(UniversalDictionaryFacilityPulls)
    sync_table(UniversalDictionaryTokens)
    sync_table(SiteConfig)
    sync_table(TransmissionHistory)

UserBase.metadata.create_all(bind=user_engine)

app.include_router(access_api.router, tags=['Access'], prefix='/api/db_access')
app.include_router(dictionary_mapper_api.router, tags=['Mapper'], prefix='/api/dictionary_mapper')
app.include_router(usl_data_transmission_api.router, tags=['Transmission'], prefix='/api/usl_data')
app.include_router(data_dictionary_api.router, tags=['Data Dictionary'], prefix='/api/data_dictionary')
app.include_router(configuration_api.router, tags=['App Configuration'], prefix='/api/config')
app.include_router(site_configuration_api.router, tags=['Site Configuration'], prefix='/api/site_config')
app.include_router(data_dictionary_usl_api.router, tags=['USL Data Dictionary'], prefix='/api/usl/data_dictionary')
# TODO: MOVE READ TO ELSEWHERE
# app.include_router(text2sql_api.router, tags=['Text2SQL'], prefix='/api/text2sql')


@app.get("/api/healthchecker")
def root():
    return {"message": "Welcome to data map, we are up and running"}


# Run the FastAPI application
if __name__ == "__main__":
    import uvicorn

    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)