class MegaUsers:
def __init__(self):
self.user_collection = MegaDB().db["users"]
async def insert_user(self, user_id: int):
if self.user_collection.count({"user_id": user_id}) > 0:
return False
else:
self.user_collection.insert_one({
"user_id": user_id,
"dld_settings": "default",
"custom_thumbnail": None
})
async def get_user(self, user_id: int):
return self.user_collection.find_one({"user_id": user_id})
async def update_dld_settings(self, user_id: int, data: str):
self.user_collection.update_one({"user_id": user_id},
{"$set": {
"dld_settings": data
}})
async def update_cst_thumb(self, user_id: int, thumb: typing.Union[bytes,
str]):
self.user_collection.update_one({"user_id": user_id},
{"$set": {
"custom_thumbnail": thumb
}})
class MegaUsers:
def __init__(self):
""" MegaUsers is the mongo collection for the documents that holds the details of the users such as their
download settings, gdrive and etc, for them users who uses the bot.
Functions:
insert_user: insert new documents, that contains the details of the new users who started using the bot.
get_user: return the document that contains the user_id for the the given telegram user id.
update_dld_settings: set/update the dld_settings field for the document with the given telegram user id.
update_cst_thumb: set/update the custom_thumbnail field for the document with the given telegram user id.
update_gdrive: set/update the gdrive_key and gdrive_key_location for the document with the given telegram
user id.
"""
self.user_collection = MegaDB().db["users"]
async def insert_user(self, user_id: int):
if self.user_collection.count({"user_id": user_id}) > 0:
return False
else:
self.user_collection.insert_one({
"user_id": user_id,
"dld_settings": "default",
"custom_thumbnail": None
})
async def get_user(self, user_id: int):
return self.user_collection.find_one({"user_id": user_id})
async def update_dld_settings(self, user_id: int, data: str):
self.user_collection.update_one({"user_id": user_id},
{"$set": {
"dld_settings": data
}})
async def update_cst_thumb(self, user_id: int, thumb: typing.Union[bytes,
str]):
self.user_collection.update_one({"user_id": user_id},
{"$set": {
"custom_thumbnail": thumb
}})
async def update_gdrive(self, user_id: int, key_file: typing.Union[bytes,
str],
key_location: str):
self.user_collection.update_one({"user_id": user_id}, {
"$set": {
"gdrive_key": key_file,
"gdrive_key_location": key_location
}
})
async def update_yt_cookie(self, user_id: int,
cookie_file: typing.Union[bytes, str],
cookie_file_location: str):
self.user_collection.update_one({"user_id": user_id}, {
"$set": {
"yt_cookie": cookie_file,
"yt_cookie_location": cookie_file_location
}
})
async def update_file_rename_settings(self, user_id: int, dld_type: str):
self.user_collection.update_one({"user_id": user_id},
{"$set": {
"f_rename_type": dld_type
}})
async def update_seedr_username(self, user_id: int, seedr_username: str):
self.user_collection.update_one(
{"user_id": user_id}, {"$set": {
"seedr_username": seedr_username
}})
async def update_seedr_paswd(self, user_id: int, seedr_passwd: str):
self.user_collection.update_one(
{"user_id": user_id}, {"$set": {
"seedr_passwd": seedr_passwd
}})