create database for broadcast

added database structure for running broadcast function
This commit is contained in:
levina 2021-11-14 06:51:47 +07:00 committed by GitHub
parent db97c04c1f
commit c6f7dc0457
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -0,0 +1,29 @@
import os
import re
from sqlalchemy import create_engine
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()