added data_sql for saving data
created data_sql for user data
This commit is contained in:
parent
f502dd6a2b
commit
af807b5bec
37
driver/database/data_sql.py
Normal file
37
driver/database/data_sql.py
Normal file
@ -0,0 +1,37 @@
|
||||
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 chatlists():
|
||||
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