GlowSenseAI / check_all_missing_columns.py
Tayyaba11's picture
deploy backend
6e08e39
Raw
History Blame Contribute Delete
952 Bytes
from sqlalchemy import inspect, MetaData
from database import engine
from models import Base
import models # Ensure all models are imported so they register with Base.metadata
def check_missing_columns():
inspector = inspect(engine)
db_tables = inspector.get_table_names()
missing_dict = {}
for table_name in Base.metadata.tables.keys():
if table_name not in db_tables:
print(f"Table missing: {table_name}")
continue
model_columns = Base.metadata.tables[table_name].columns.keys()
db_columns = [col['name'] for col in inspector.get_columns(table_name)]
missing = [col for col in model_columns if col not in db_columns]
if missing:
print(f"Table '{table_name}' is missing columns: {missing}")
missing_dict[table_name] = missing
return missing_dict
if __name__ == "__main__":
check_missing_columns()