15 lines
600 B
Python
15 lines
600 B
Python
import os
|
|
import asyncpg
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine, session
|
|
import dotenv
|
|
|
|
# connection = psycopg2.connect(*(os.getenv(key) for key in ["DATABASE", "DB_HOST", "DB_USER", "DB_PASSWORD"]))
|
|
# connection.autocommit = True
|
|
|
|
dotenv.load_dotenv(".env")
|
|
DATABASE_URL = (f"postgresql+asyncpg://{os.getenv('DB_USER')}:{os.getenv('DB_PASSWORD')}@"
|
|
f"{os.getenv('DB_HOST')}/{os.getenv('DATABASE')}")
|
|
print(DATABASE_URL)
|
|
engine = create_async_engine(DATABASE_URL, echo=True)
|
|
async_session_ = async_sessionmaker(bind=engine, expire_on_commit=False)
|