from sqlalchemy import create_engine from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import sessionmaker SQL_ALCHEMY_DATABASE_URL_PGSQL = "postgresql+psycopg2://postgres:159357@localhost/general" engine = create_engine( SQL_ALCHEMY_DATABASE_URL_PGSQL #, connect_args = {"check_same_thread": False} ) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind= engine) Base = declarative_base() def get_db(): db = SessionLocal() try: yield db finally: db.close()