This commit is contained in:
levina 2021-11-14 06:53:08 +07:00 committed by GitHub
parent c6f7dc0457
commit 1356c93862
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,29 +1,20 @@
import os
import re
from sqlalchemy import create_engine
from config import DATABASE_URL as db
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import scoped_session, sessionmaker
from config import DATABASE_URL as db
if db and db.startswith("postgres://"):
app = db.replace("postgres://", "postgresql://", 1)
BASE = declarative_base()
def start() -> scoped_session:
engine = create_engine(app)
BASE.metadata.bind = engine
BASE.metadata.create_all(engine)
return scoped_session(sessionmaker(bind=engine, autoflush=False))
SESSION = start()