not now
This commit is contained in:
parent
c9144f7b93
commit
1b93dd163a
@ -1,21 +0,0 @@
|
|||||||
import os
|
|
||||||
import re
|
|
||||||
|
|
||||||
from driver.veez import user as app
|
|
||||||
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
|
|
||||||
|
|
||||||
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()
|
|
@ -1,34 +0,0 @@
|
|||||||
import threading
|
|
||||||
from driver.database import BASE, SESSION
|
|
||||||
from sqlalchemy import Column, String, UnicodeText
|
|
||||||
|
|
||||||
|
|
||||||
class Chats(BASE):
|
|
||||||
__tablename__ = "chats"
|
|
||||||
chat_id = Column(String(14), primary_key=True)
|
|
||||||
chat_name = Column(UnicodeText)
|
|
||||||
|
|
||||||
def __init__(self, chat_id, chat_name=None):
|
|
||||||
self.chat_id = chat_id
|
|
||||||
self.chat_name = chat_name
|
|
||||||
|
|
||||||
Chats.__table__.create(checkfirst=True)
|
|
||||||
|
|
||||||
CHATS_LOCK = threading.RLock()
|
|
||||||
CHATS_DATA = set()
|
|
||||||
|
|
||||||
def del_chat(chat_id):
|
|
||||||
with CHATS_LOCK:
|
|
||||||
chat = SESSION.query(Chats).get(str(chat_id))
|
|
||||||
if chat:
|
|
||||||
SESSION.delete(chat)
|
|
||||||
SESSION.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def chatdata():
|
|
||||||
global CHAT_ID
|
|
||||||
try:
|
|
||||||
CHAT_ID = {int(x.chat_id) for x in SESSION.query(Chats).all()}
|
|
||||||
return CHAT_ID
|
|
||||||
finally:
|
|
||||||
SESSION.close()
|
|
Loading…
Reference in New Issue
Block a user