CSVAgent / database.py
Quazim0t0's picture
Update database.py
a400f6e verified
raw
history blame
812 Bytes
from sqlalchemy import create_engine, MetaData, inspect
# Initialize the SQLite database
engine = create_engine("sqlite:///database.db")
metadata_obj = MetaData()
# Function to check existing tables
def get_existing_tables():
"""
Returns a list of tables currently in the database.
Returns:
list: List of table names.
"""
inspector = inspect(engine)
return inspector.get_table_names()
# Function to safely start the database
def initialize_database():
"""
Starts the database without failing if no SQL file is uploaded yet.
"""
tables = get_existing_tables()
if not tables:
print("No tables found. The database is waiting for an SQL file upload.")
else:
print(f"Database initialized with tables: {tables}")
initialize_database()