diff --git a/src/plugins/ncm/__init__.py b/src/plugins/ncm/__init__.py index e92d535..f004b2e 100644 --- a/src/plugins/ncm/__init__.py +++ b/src/plugins/ncm/__init__.py @@ -41,7 +41,7 @@ async def song_is_open(event: Union[GroupMessageEvent, PrivateMessageEvent]) -> setting.insert({"group_id": event.group_id, "song": False, "list": False}) return False - elif isinstance(event, PrivateMessageEvent): + if isinstance(event, PrivateMessageEvent): if info := setting.search(Q["user_id"] == event.user_id): return info[0]["song"] setting.insert({"user_id": event.user_id, "song": True, "list": True}) @@ -55,27 +55,24 @@ async def playlist_is_open( info = setting.search(Q["group_id"] == event.group_id) if info: return info[0]["list"] - else: - setting.insert({"group_id": event.group_id, - "song": False, "list": False}) - return False + setting.insert({"group_id": event.group_id, + "song": False, "list": False}) + return False elif isinstance(event, PrivateMessageEvent): info = setting.search(Q["user_id"] == event.user_id) if info: return info[0]["list"] - else: - setting.insert({"user_id": event.user_id, - "song": True, "list": True}) - return True + setting.insert({"user_id": event.user_id, + "song": True, "list": True}) + return True async def check_search() -> bool: info = setting.search(Q["global"] == "search") if info: return info[0]["value"] - else: - setting.insert({"global": "search", "value": True}) - return True + setting.insert({"global": "search", "value": True}) + return True async def music_set_rule(event: Union[GroupMessageEvent, PrivateMessageEvent]) -> bool: