def init_db() -> None: try: # Try to create session to check if DB is awake db = SessionLocal() db.execute("SELECT 1") except Exception as e: logger.error(e) raise e
def check_db() -> None: try: # Try to create session to check if DB is awake db = SessionLocal() db.execute("SELECT 1") except Exception: logger.exception("Init failed") raise
def init(): try: db = SessionLocal() # Try to create session to check if DB is awake db.execute("SELECT 1") # Wait for API to be awake, run one simple tests to authenticate except Exception as e: logger.error(e) raise e
def init() -> None: logger.error(f'{settings.SQLALCHEMY_DATABASE_URI}') logger.error('hello c pre start') try: # Try to create session to check if DB is awake db = SessionLocal() db.execute("SELECT 1") except Exception as e: logger.error(e) raise e
def init() -> None: try: db = SessionLocal() # Try to create session to check if DB is awake db.execute("SELECT 1") ctx.database_connection = True logger.info("Database connection is working") except Exception as e: ctx.database_connection = False logger.error(e) logger.error("Could not connect to database")
def init() -> None: try: db = SessionLocal() # Try to create session to check if DB is awake db.execute("SELECT 1") if not os.path.isdir(settings.UPLOADED_FILES_FOLDER): create_uploaded_files_folder(settings.UPLOADED_FILES_FOLDER) except Exception as e: logger.error(e) raise e
def create_table(): engine = create_engine(config.SQLALCHEMY_DATABASE_URI, ) metadata = MetaData() video_caption = Table( "video_caption", metadata, Column("title", String, primary_key=True), Column("caption", JSONB), ) print("Creating tables") metadata.drop_all(engine) metadata.create_all(engine) db = SessionLocal() db.execute( f"CREATE INDEX ON video_caption((video_caption.caption->'caption'));")
import pandas as pd import os from app.db.session import SessionLocal, engine db = SessionLocal() sql = """ SELECT TABLE_NAME FROM information_schema.`TABLES` WHERE TABLE_SCHEMA = 'DWDB' AND TABLE_NAME != 'alembic_version' """ rows = db.execute(sql).fetchall() for table in rows: table_name = table[0] sql = f""" SELECT * FROM DWDB.{table_name } """ df = pd.read_sql(sql, con=engine) file_path = os.getcwd() + f"/init_data/{table_name}.csv" print(file_path) df.to_csv(file_path, index=0)