majjo/Vocard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Return the state of update db method

Browse Source
This commit is contained in:
Choco
2023-09-20 19:54:24 +08:00
parent 8eeb2d665a
commit bfc57f70e0
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
function.py
View File

@@ -54,7 +54,7 @@ def get_settings(guild_id:int) -> dict:
GUILD_SETTINGS[guild_id] = settings or {} GUILD_SETTINGS[guild_id] = settings or {}
return settings return settings
def update_settings(guild_id:int, data: dict, mode="set") -> None: def update_settings(guild_id:int, data: dict, mode="set") -> bool:
settings = get_settings(guild_id) settings = get_settings(guild_id)
for key, value in data.items(): for key, value in data.items():
@@ -65,9 +65,10 @@ def update_settings(guild_id:int, data: dict, mode="set") -> None:
case "unset": case "unset":
GUILD_SETTINGS[guild_id].pop(key) GUILD_SETTINGS[guild_id].pop(key)
case _: case _:
return return False
SETTINGS_DB.update_one({"_id":guild_id}, {f"${mode}":data}) result = SETTINGS_DB.update_one({"_id":guild_id}, {f"${mode}":data})
return result.modified_count > 0
def open_json(path: str) -> dict: def open_json(path: str) -> dict:
try: try:
@@ -198,7 +199,7 @@ async def create_account(ctx: Union[commands.Context, discord.Interaction]) -> N
except: except:
pass pass
async def get_playlist(user_id:int, dType:str=None, dId:str=None) -> dict: async def get_playlist(user_id:int, dType:str=None, dId:str=None) -> bool:
user = PLAYLISTS_DB.find_one({"_id":user_id}, {"_id": 0}) user = PLAYLISTS_DB.find_one({"_id":user_id}, {"_id": 0})
if not user: if not user:
return None return None
@@ -211,7 +212,9 @@ async def get_playlist(user_id:int, dType:str=None, dId:str=None) -> dict:
async def update_playlist(user_id:int, data:dict, *, mode:str="set", update_cache: bool=False) -> None: async def update_playlist(user_id:int, data:dict, *, mode:str="set", update_cache: bool=False) -> None:
if update_cache: if update_cache:
PLAYLIST_NAME.pop(str(user_id), None) PLAYLIST_NAME.pop(str(user_id), None)
PLAYLISTS_DB.update_one({"_id":user_id}, {f"${mode}": data}) result = PLAYLISTS_DB.update_one({"_id":user_id}, {f"${mode}": data})
return result.modified_count > 0
async def update_inbox(user_id:int, data:dict) -> None: async def update_inbox(user_id:int, data:dict) -> bool:
return PLAYLISTS_DB.update_one({"_id":user_id}, {"$push":{'inbox':data}}) result = PLAYLISTS_DB.update_one({"_id":user_id}, {"$push":{'inbox':data}})
return result.modified_count > 0