59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
|
from typing import Dict, List, Union
|
||
|
|
|
||
|
|
from driver.database.dblocal import db
|
||
|
|
|
||
|
|
pytgdb = db.pytg
|
||
|
|
admindb = db.admin
|
||
|
|
|
||
|
|
|
||
|
|
async def get_active_chats() -> list:
|
||
|
|
chats = pytgdb.find({"chat_id": {"$lt": 0}})
|
||
|
|
if not chats:
|
||
|
|
return []
|
||
|
|
chats_list = []
|
||
|
|
for chat in await chats.to_list(length=1000000000):
|
||
|
|
chats_list.append(chat)
|
||
|
|
return chats_list
|
||
|
|
|
||
|
|
|
||
|
|
async def is_active_chat(chat_id: int) -> bool:
|
||
|
|
chat = await pytgdb.find_one({"chat_id": chat_id})
|
||
|
|
if not chat:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
async def add_active_chat(chat_id: int):
|
||
|
|
is_served = await is_active_chat(chat_id)
|
||
|
|
if is_served:
|
||
|
|
return
|
||
|
|
return await pytgdb.insert_one({"chat_id": chat_id})
|
||
|
|
|
||
|
|
|
||
|
|
async def remove_active_chat(chat_id: int):
|
||
|
|
is_served = await is_active_chat(chat_id)
|
||
|
|
if not is_served:
|
||
|
|
return
|
||
|
|
return await pytgdb.delete_one({"chat_id": chat_id})
|
||
|
|
|
||
|
|
|
||
|
|
async def is_music_playing(chat_id: int) -> bool:
|
||
|
|
chat = await admindb.find_one({"chat_id_toggle": chat_id})
|
||
|
|
if not chat:
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
async def music_on(chat_id: int):
|
||
|
|
is_on = await is_music_playing(chat_id)
|
||
|
|
if is_on:
|
||
|
|
return
|
||
|
|
return await admindb.delete_one({"chat_id_toggle": chat_id})
|
||
|
|
|
||
|
|
|
||
|
|
async def music_off(chat_id: int):
|
||
|
|
is_on = await is_music_playing(chat_id)
|
||
|
|
if not is_on:
|
||
|
|
return
|
||
|
|
return await admindb.insert_one({"chat_id_toggle": chat_id})
|