mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2025-12-29 09:44:50 +00:00
31 lines
807 B
Python
31 lines
807 B
Python
from database import dbname
|
|
|
|
usersdb = dbname.users
|
|
|
|
|
|
async def is_afk(user_id: int) -> bool:
|
|
user = await usersdb.find_one({"user_id": user_id})
|
|
return (True, user["reason"]) if user else (False, {})
|
|
|
|
|
|
async def add_afk(user_id: int, mode):
|
|
await usersdb.update_one({"user_id": user_id}, {"$set": {
|
|
"reason": mode
|
|
}},
|
|
upsert=True)
|
|
|
|
|
|
async def remove_afk(user_id: int):
|
|
user = await usersdb.find_one({"user_id": user_id})
|
|
if user:
|
|
return await usersdb.delete_one({"user_id": user_id})
|
|
|
|
|
|
async def get_afk_users() -> list:
|
|
users = usersdb.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
|