Reformat the code to follow PEP 8

Author: ayanrajpoot10

bugscanx/__init__.py

@@ -1,6 +1,8 @@
+import os
 import threading
-from rich import print
 from pyfiglet import Figlet
+from rich import print
+
 
 def import_modules():
     def task():
@@ -9,15 +11,17 @@ def task():
             from bugscanx.modules.scrapers.subfinder import subfinder
         except Exception:
             pass
-    
+
     threading.Thread(target=task, daemon=True).start()
 
+
 figlet = Figlet(font="calvin_s")
 
+
 def clear_screen():
-    import os
     os.system('cls' if os.name == 'nt' else 'clear')
 
+
 def banner():
     clear_screen()
     print("""
@@ -28,11 +32,14 @@ def banner():
     [bold magenta]  Tᴇʟᴇɢʀᴀᴍ: @BᴜɢSᴄᴀɴX   [/bold magenta]
     """)
 
+
 def text_ascii(text, color="bold magenta", indentation=2):
     clear_screen()
     ascii_banner = figlet.renderText(text)
-    shifted_banner = "\n".join((" " * indentation) + line for line in ascii_banner.splitlines())
+    shifted_banner = "\n".join((" " * indentation) + line 
+                              for line in ascii_banner.splitlines())
     print(f"[{color}]{shifted_banner}[/{color}]")
     print()
 
+
 import_modules()

bugscanx/handler/runner.py

@@ -2,42 +2,52 @@ def run_1():
     from bugscanx.modules.scanners_pro import host_scanner_pro
     host_scanner_pro.main()
 
+
 def run_2():
     from bugscanx.modules.scanners import host_scanner
     host_scanner.main()
 
+
 def run_3():
     from bugscanx.modules.scanners import cidr_scanner
     cidr_scanner.main()
 
+
 def run_4():
     from bugscanx.modules.scrapers.subfinder import subfinder
     subfinder.main()
 
+
 def run_5():
     from bugscanx.modules.scrapers.iplookup import iplookup
     iplookup.main()
 
+
 def run_6():
     from bugscanx.modules.others import file_toolkit
     file_toolkit.main()
 
+
 def run_7():
     from bugscanx.modules.scanners import open_port
     open_port.main()
 
+
 def run_8():
     from bugscanx.modules.others import dns_records
     dns_records.main()
 
+
 def run_9():
     from bugscanx.modules.others import host_info
     host_info.main()
 
+
 def run_10():
     from bugscanx.modules.others import script_help
     script_help.main()
 
+
 def run_11():
     from bugscanx.modules.others import script_updater
     script_updater.main()

bugscanx/main.py

@@ -14,26 +14,33 @@
     '9': ("HOST INFO", "bold blue"),
     '10': ("HELP", "bold yellow"),
     '11': ("UPDATE", "bold magenta"),
-    '12': ("EXIT", "bold red")
+    '12': ("EXIT", "bold red"),
 }
 
+
 def main():
     try:
         while True:
             banner()
-            print('\n'.join(f"[{color}] [{k}]{' ' if len(k)==1 else ''} {desc}" 
-                        for k, (desc, color) in MENU_OPTIONS.items()))
+            menu_items = (
+                f"[{color}] [{k}]{' ' if len(k)==1 else ''} {desc}"
+                for k, (desc, color) in MENU_OPTIONS.items()
+            )
+            print('\n'.join(menu_items))
 
             choice = input("\n \033[36m[-]  Your Choice: \033[0m")
             if choice not in MENU_OPTIONS:
                 continue
-                
+
             if choice == '12':
                 return
-                
+
             text_ascii(MENU_OPTIONS[choice][0])
             try:
-                module = __import__('bugscanx.handler.runner', fromlist=[f'run_{choice}'])
+                module = __import__(
+                    'bugscanx.handler.runner',
+                    fromlist=[f'run_{choice}']
+                )
                 getattr(module, f'run_{choice}')()
                 print("\n[yellow] Press Enter to continue...", end="")
                 input()

bugscanx/modules/others/dns_records.py

@@ -1,43 +1,61 @@
 import os
-from rich import print
+from typing import List, Optional
+
 from dns import resolver as dns_resolver
-from bugscanx.utils.common import get_input, get_confirm
+from rich import print
+
+from bugscanx.utils.common import get_confirm, get_input
+
 
-def configure_resolver(custom_nameservers=None):
+def configure_resolver(custom_nameservers: Optional[List[str]] = None):
     dns_obj = dns_resolver.Resolver()
-    
+
     if custom_nameservers and len(custom_nameservers) > 0:
         dns_obj.nameservers = custom_nameservers
-        print(f"[yellow]Using custom DNS servers: {', '.join(custom_nameservers)}[/yellow]")
+        print(
+            f"[yellow]Using custom DNS servers: "
+            f"{', '.join(custom_nameservers)}[/yellow]"
+        )
         return dns_obj
-    
-    is_termux = 'com.termux' in os.environ.get('PREFIX', '') or os.path.exists('/data/data/com.termux')
-    
+
+    is_termux = "com.termux" in os.environ.get(
+        "PREFIX", ""
+    ) or os.path.exists("/data/data/com.termux")
+
     if is_termux:
-        dns_obj.nameservers = ['8.8.8.8', '8.8.4.4', '1.1.1.1']
-    
+        dns_obj.nameservers = ["8.8.8.8", "8.8.4.4", "1.1.1.1"]
+
     return dns_obj
 
-def resolve_and_print(domain, record_type, custom_nameservers=None):
+
+def resolve_and_print(
+    domain: str, record_type: str, custom_nameservers: Optional[List[str]] = None
+):
     print(f"\n[green] {record_type} Records:[/green]")
     try:
         dns_obj = configure_resolver(custom_nameservers)
         answers = dns_obj.resolve(domain, record_type)
         found = False
         for answer in answers:
             found = True
-            if record_type == 'MX':
-                print(f"[cyan]- {answer.exchange} (priority: {answer.preference})[/cyan]")
-            elif record_type == 'SOA':
+            if record_type == "MX":
+                print(
+                    f"[cyan]- {answer.exchange} "
+                    f"(priority: {answer.preference})[/cyan]"
+                )
+            elif record_type == "SOA":
                 print(f"[cyan]- Primary NS: {answer.mname}")
                 print(f"  Responsible: {answer.rname}")
                 print(f"  Serial: {answer.serial}")
                 print(f"  Refresh: {answer.refresh}")
                 print(f"  Retry: {answer.retry}")
                 print(f"  Expire: {answer.expire}")
                 print(f"  Minimum TTL: {answer.minimum}[/cyan]")
-            elif record_type == 'SRV':
-                print(f"[cyan]- Priority: {answer.priority}, Weight: {answer.weight}")
+            elif record_type == "SRV":
+                print(
+                    f"[cyan]- Priority: {answer.priority}, "
+                    f"Weight: {answer.weight}"
+                )
                 print(f"  Port: {answer.port}, Target: {answer.target}[/cyan]")
             else:
                 print(f"[cyan]- {answer.to_text()}[/cyan]")
@@ -48,42 +66,51 @@ def resolve_and_print(domain, record_type, custom_nameservers=None):
     except Exception:
         print(f"[yellow] Error fetching {record_type} record[/yellow]")
 
-def nslookup(domain, custom_nameservers=None):
+
+def nslookup(domain: str, custom_nameservers: Optional[List[str]] = None):
     print(f"[cyan]\n Performing NSLOOKUP for: {domain}[/cyan]")
-    
+
     record_types = [
-        'A',
-        'AAAA',
-        'CNAME',
-        'MX',
-        'NS',
-        'TXT',
-        'SOA',
-        'PTR',
-        'SRV',
-        'CAA',
-        'DNSKEY',
-        'TLSA'
+        "A",
+        "AAAA",
+        "CNAME",
+        "MX",
+        "NS",
+        "TXT",
+        "SOA",
+        "PTR",
+        "SRV",
+        "CAA",
+        "DNSKEY",
+        "TLSA",
     ]
-    
+
     for record_type in record_types:
         resolve_and_print(domain, record_type, custom_nameservers)
 
+
 def main():
     domain = get_input("Enter the domain to lookup")
     if not domain:
         print("[red] Please enter a valid domain.[/red]")
         return
-    
+
     use_custom_dns = get_confirm(" Want to use custom DNS servers?")
     custom_nameservers = None
-    
+
     if use_custom_dns:
-        nameservers_input = get_input("Enter DNS servers separated by commas (e.g., 8.8.8.8,1.1.1.1)")
+        nameservers_input = get_input(
+            "Enter DNS servers separated by commas (e.g., 8.8.8.8,1.1.1.1)"
+        )
         if nameservers_input:
-            custom_nameservers = [server.strip() for server in nameservers_input.split(',')]
-            print(f"[cyan] Will use custom DNS servers: {', '.join(custom_nameservers)}[/cyan]")
-    
+            custom_nameservers = [
+                server.strip() for server in nameservers_input.split(",")
+            ]
+            print(
+                f"[cyan] Will use custom DNS servers: "
+                f"{', '.join(custom_nameservers)}[/cyan]"
+            )
+
     try:
         nslookup(domain, custom_nameservers)
     except Exception as e: