Python SessionLocal.execute Beispiele

Beispiel #1

Datei: tests_pre_start.py Projekt: elben10/base-project

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

Beispiel #2

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

Beispiel #3

Datei: tests_pre_start.py Projekt: plydot/fps-farm-aggregator

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

Beispiel #4

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

Beispiel #5

Datei: database.py Projekt: FabriceSalvaire/poc-portal-backend

 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")

Beispiel #6

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

Beispiel #7

Datei: caption_migrator.py Projekt: davipaula/vocable-me

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'));")

Beispiel #8

Datei: export_data.py Projekt: woodslee/vue-element-admin-fastapi

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)