File size: 798 Bytes
a8e9b84 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
from DragMusic.core.mongo import mongodb
afkdb = mongodb.afk
async def is_afk(user_id: int) -> bool:
user = await afkdb.find_one({"user_id": user_id})
if not user:
return False, {}
return True, user["reason"]
async def add_afk(user_id: int, mode):
await afkdb.update_one(
{"user_id": user_id}, {"$set": {"reason": mode}}, upsert=True
)
async def remove_afk(user_id: int):
user = await afkdb.find_one({"user_id": user_id})
if user:
return await afkdb.delete_one({"user_id": user_id})
async def get_afk_users() -> list:
users = afkdb.find({"user_id": {"$gt": 0}})
if not users:
return []
users_list = []
for user in await users.to_list(length=1000000000):
users_list.append(user)
return users_list
|