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