88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
import datetime
|
|
import motor.motor_asyncio
|
|
|
|
class Database:
|
|
def __init__(self, uri, database_name):
|
|
self._client = motor.motor_asyncio.AsyncIOMotorClient(uri)
|
|
self.db = self._client[database_name]
|
|
self.col = self.db.users
|
|
|
|
def new_user(self, id):
|
|
return dict(
|
|
id=id,
|
|
join_date=datetime.date.today().isoformat(),
|
|
notif=True,
|
|
ban_status=dict(
|
|
is_banned=False,
|
|
ban_duration=0,
|
|
banned_on=datetime.date.max.isoformat(),
|
|
ban_reason="",
|
|
),
|
|
)
|
|
|
|
async def add_user(self, id):
|
|
user = self.new_user(id)
|
|
await self.col.insert_one(user)
|
|
|
|
async def is_user_exist(self, id):
|
|
user = await self.col.find_one({"id": int(id)})
|
|
return True if user else False
|
|
|
|
async def total_users_count(self):
|
|
count = await self.col.count_documents({})
|
|
return count
|
|
|
|
async def get_all_users(self):
|
|
all_users = self.col.find({})
|
|
return all_users
|
|
|
|
async def delete_user(self, user_id):
|
|
await self.col.delete_many({"id": int(user_id)})
|
|
|
|
async def remove_ban(self, id):
|
|
ban_status = dict(
|
|
is_banned=False,
|
|
ban_duration=0,
|
|
banned_on=datetime.date.max.isoformat(),
|
|
ban_reason="",
|
|
)
|
|
await self.col.update_one({"id": id}, {"$set": {"ban_status": ban_status}})
|
|
|
|
async def ban_user(self, user_id, ban_duration, ban_reason):
|
|
ban_status = dict(
|
|
is_banned=True,
|
|
ban_duration=ban_duration,
|
|
banned_on=datetime.date.today().isoformat(),
|
|
ban_reason=ban_reason,
|
|
)
|
|
await self.col.update_one({"id": user_id}, {"$set": {"ban_status": ban_status}})
|
|
|
|
async def get_ban_status(self, id):
|
|
default = dict(
|
|
is_banned=False,
|
|
ban_duration=0,
|
|
banned_on=datetime.date.max.isoformat(),
|
|
ban_reason="",
|
|
)
|
|
user = await self.col.find_one({"id": int(id)})
|
|
return user.get("ban_status", default)
|
|
|
|
async def get_all_banned_users(self):
|
|
banned_users = self.col.find({"ban_status.is_banned": True})
|
|
return banned_users
|
|
|
|
async def set_notif(self, id, notif):
|
|
await self.col.update_one({"id": id}, {"$set": {"notif": notif}})
|
|
|
|
async def get_notif(self, id):
|
|
user = await self.col.find_one({"id": int(id)})
|
|
return user.get("notif", False)
|
|
|
|
async def get_all_notif_user(self):
|
|
notif_users = self.col.find({"notif": True})
|
|
return notif_users
|
|
|
|
async def total_notif_users_count(self):
|
|
count = await self.col.count_documents({"notif": True})
|
|
return count
|