✨ get message info

Author: NoNormalCreeper

src/paser.py

@@ -58,6 +58,7 @@ def parse_arg(self, args):
         get_info_parser = subparsers.add_parser("get-info", help="Get the info of a user or group.")
         get_info_parser.add_argument("-u", "--user", help="user ID", default=None, required=False)
         get_info_parser.add_argument("-g", "--group", help="group ID", default=None, required=False)
+        get_info_parser.add_argument("-m", "--message", help="message ID", default=None, required=False)
 
         args_namespace = parser.parse_args()
         if args_namespace.operation == "send":
@@ -72,7 +73,7 @@ def parse_arg(self, args):
         elif args_namespace.operation == "get-image":
             asyncio.run(sender.get_image(args_namespace.name))
         elif args_namespace.operation == "get-info":
-            asyncio.run(sender.get_info(args_namespace.user, args_namespace.group))
+            asyncio.run(sender.get_info(args_namespace.user, args_namespace.group, args_namespace.message))
 
 
 

src/sender.py

@@ -82,15 +82,15 @@ async def get_image(self, file_name) -> None:
             {response_data['url']}"
             )
 
-    async def get_info(self, user_id: Optional[int], group_id: Optional[int]) -> None:
+    async def get_info(self, user_id: Optional[int], group_id: Optional[int], message_id: Optional[int]) -> None:
         if user_id and group_id:
             func = self._call_api('get_group_member_info', {"group_id": group_id, "user_id": user_id}, "get_info_by_cmd", print_response=False)
         elif user_id and not group_id:
             func = self._call_api('get_stranger_info', {"user_id": user_id}, "get_info_by_cmd", print_response=False)
         elif not user_id and group_id:
             func = self._call_api('get_group_info', {"group_id": group_id}, "get_info_by_cmd", print_response=False)
-        elif (not user_id) and (not group_id):
-            print("error: user or group ID is required")
+        elif message_id:
+            func = self._call_api('get_msg', {"message_id": message_id}, "get_info_by_cmd", print_response=False)
         response = await func
         print(f"\033[1;37mInfo <\033[0m \n{beautify_json(response['data'])}")