Added type hints

vil · vil · commit 5ed995ece75e · 2024-07-02T20:18:27.000+03:00
diff --git a/helper/printer.py b/helper/printer.py
@@ -18,7 +18,7 @@
 from colorama import Fore, Style
 
 
-def print_colored(message, color, prefix, *args, **kwargs):
+def print_colored(message, color, prefix, *args, **kwargs) -> None:
     """
     Print colored message with specified color and prefix.
 
@@ -31,29 +31,30 @@ def print_colored(message, color, prefix, *args, **kwargs):
     print(f"{color}{prefix}{Style.RESET_ALL} {message}", *args, **kwargs)
 
 
-def info(message, *args, **kwargs):
+def info(message, *args, **kwargs) -> None:
     print_colored(message, Fore.LIGHTBLUE_EX, "[*]", *args, **kwargs)
 
 
-def success(message, *args, **kwargs):
+def success(message, *args, **kwargs) -> None:
     print_colored(message, Fore.LIGHTGREEN_EX, "[+]", *args, **kwargs)
 
 
-def error(message, *args, **kwargs):
+def error(message, *args, **kwargs) -> None:
     print_colored(message, Fore.LIGHTRED_EX, "[!]", *args, **kwargs)
 
 
-def warning(message, *args, **kwargs):
+def warning(message, *args, **kwargs) -> None:
     print_colored(message, Fore.LIGHTYELLOW_EX, "[-]", *args, **kwargs)
 
 
-def debug(message, *args, **kwargs):
+def debug(message, *args, **kwargs) -> None:
     print_colored(message, Fore.LIGHTMAGENTA_EX, "[>]", *args, **kwargs)
 
 
-def nonprefix(message, *args, **kwargs):
+def nonprefix(message, *args, **kwargs) -> None:
     print(message, *args, **kwargs)
 
-def inp(prompt, *args, **kwargs):
+
+def inp(prompt, *args, **kwargs) -> None:
     print_colored(prompt, Fore.LIGHTBLUE_EX, "[?]", end="", *args, **kwargs)
     return input()
diff --git a/helper/randomuser.py b/helper/randomuser.py
@@ -439,8 +439,8 @@ class GetUser:
     """
     Return a random user agent from the list.
     """
-    def __init__(self):
+    def __init__(self) -> str:
         self.user_agent = random.choice(users)
 
-    def __str__(self):
+    def __str__(self) -> str:
         return self.user_agent
diff --git a/helper/timer.py b/helper/timer.py
@@ -23,7 +23,7 @@ def timer(func):
     """
     A timer decorator to measure the execution time of a function.
     """
-    def wrapper(*args, **kwargs):
+    def wrapper(*args, **kwargs) -> str:
         start_time = time.time()
         result = func(*args, **kwargs)
         end_time = time.time()
diff --git a/helper/url_helper.py b/helper/url_helper.py
@@ -40,7 +40,7 @@ def read_local_content(path):
         return None
 
 # I hate pyinstaller.
-def resource_path(relative_path):
+def resource_path(relative_path) -> str:
     if hasattr(sys, '_MEIPASS'):
         return os.path.join(sys._MEIPASS, relative_path)
     return os.path.join(os.path.abspath("."), relative_path)
diff --git a/utils/caesar_cipher.py b/utils/caesar_cipher.py
@@ -27,7 +27,7 @@ class CaesarCipher:
     :param shift: The shift to use for the encryption or decryption.
     :param mode: The mode to use for the encryption or decryption.
     """
-    def __init__(self, text: str, mode: str):
+    def __init__(self, text: str, mode: str) -> None:
         self.text = text
         self.mode = mode
 
@@ -48,14 +48,14 @@ def __init__(self, text: str, mode: str):
             printer.error("Invalid mode, please choose either 'encrypt' , 'decrypt' or 'bruteforce'..!")
 
     @staticmethod
-    def get_key():
+    def get_key() -> int:
         shift = int(printer.inp("Enter a number of shifts (0 to 25) : \t"))
         if shift < 0 or shift > 25:
             printer.error("Invalid shift number, please choose a number between 0 and 25..!")
         return shift
 
     @staticmethod
-    def caesar_encrypt(text, shift):
+    def caesar_encrypt(text, shift) -> str:
         encrypted_text = ""
         for char in text:
             if char.isalpha():
@@ -70,7 +70,7 @@ def caesar_encrypt(text, shift):
         return encrypted_text
 
     @staticmethod
-    def caesar_decrypt(encrypted_text, shift):
+    def caesar_decrypt(encrypted_text, shift) -> str:
         decrypted_text = ""
         for char in encrypted_text:
             if char.isalpha():
@@ -85,7 +85,7 @@ def caesar_decrypt(encrypted_text, shift):
         return decrypted_text
 
     @staticmethod
-    def brute_force(encrypted_text):
+    def brute_force(encrypted_text) -> None:
         for i in range(26):
             decrypted_text = ""
             for char in encrypted_text:
diff --git a/utils/cybercrime_int.py b/utils/cybercrime_int.py
@@ -15,7 +15,7 @@
  along with this program.  If not, see <https://www.gnu.org/licenses/>.
  """
 
-import json, time, requests
+import time, requests
 from helper import printer, timer
 from helper import randomuser
 from colorama import Style
@@ -28,7 +28,7 @@ class Scan:
     :param target: email or a domain
     """
     @timer.timer
-    def __init__(self, target):
+    def __init__(self, target) -> None:
         try:
             if '@' in target:
                 url = f"https://cavalier.hudsonrock.com/api/json/v2/osint-tools/search-by-email?email={target}"
diff --git a/utils/dirbuster.py b/utils/dirbuster.py
@@ -27,7 +27,7 @@ class Scan:
     :param domain: url to scan
     """
     @timer.timer
-    def __init__(self, domain):
+    def __init__(self, domain) -> None:
         self.domain = domain
         self.url_set = set()
 
@@ -37,7 +37,7 @@ def __init__(self, domain):
         printer.success(f"Scan Completed..! There were {Style.BRIGHT}{len(self.url_set)}{Style.RESET_ALL} valid URLs!")
 
     @staticmethod
-    def get_wordlist():
+    def get_wordlist() -> str:
         """
         Reads the wordlist from the url and returns a list of names
 
@@ -49,7 +49,7 @@ def get_wordlist():
         except requests.exceptions.ConnectionError:
             return None
 
-    async def fetch_url(self, session, path):
+    async def fetch_url(self, session, path) -> None:
         """
         Fetches the url and checks if it is valid
 
@@ -63,7 +63,7 @@ async def fetch_url(self, session, path):
                 printer.success(f"{len(self.url_set) + 1} Valid URL(s) found : {Style.BRIGHT}{url}{Style.RESET_ALL}")
                 self.url_set.add(url)
 
-    async def scan_async(self, paths):
+    async def scan_async(self, paths) -> None:
         """
         Scans the url asynchronously
 
@@ -73,7 +73,7 @@ async def scan_async(self, paths):
             tasks = [self.fetch_url(session, path) for path in paths]
             await asyncio.gather(*tasks, return_exceptions=True)
 
-    def scan_urls(self):
+    def scan_urls(self) -> None:
         paths = self.get_wordlist()
         if paths is None:
             printer.error("Connection Error..!")
diff --git a/utils/email_search.py b/utils/email_search.py
@@ -29,7 +29,7 @@ class Holehe:
     :param email: The email address to search for.
     """
     @timer.timer
-    def __init__(self, email):
+    def __init__(self, email) -> None:
         printer.info(f"Trying to find sites where {Style.BRIGHT}{email}{Style.RESET_ALL} is used, thanks to holehe.")
         time.sleep(1)
         try:
@@ -49,7 +49,7 @@ def __init__(self, email):
             printer.error(f"Unexpected error : {e}")
 
     @staticmethod
-    def _format_output(output):
+    def _format_output(output) -> str:
         lines = output.split("\n")[4:-4]
         for i, line in enumerate(lines):
             if "[+]" in line:
diff --git a/utils/fake_info_generator.py b/utils/fake_info_generator.py
@@ -27,7 +27,7 @@ class Generate:
     Thanks to Faker, https://pypi.org/project/Faker/
     """
     @timer.timer
-    def __init__(self):
+    def __init__(self) -> None:
         fake = Faker()
         printer.info("Generating fake information...")
         time.sleep(1)
diff --git a/utils/ig_scrape.py b/utils/ig_scrape.py
@@ -26,8 +26,7 @@ class Scrape:
     :param target: The username of the account to scrape.
     """
     @timer.timer
-    def __init__(self, target):
-        
+    def __init__(self, target) -> None:
         printer.info(f"Trying to scrape information about {Style.BRIGHT}{target}{Style.RESET_ALL}...")
 
         try:
@@ -40,7 +39,7 @@ def __init__(self, target):
         
 
     # Function to print user information
-    def print_user_info(self, data):
+    def print_user_info(self, data) -> None:
         readable_data = { # Format
             'Username': data.get('username', 'N/A'),
             'Full Name': data.get('full_name', 'N/A'),
diff --git a/utils/ip_lookup.py b/utils/ip_lookup.py
@@ -28,7 +28,7 @@ class Lookup:
     :param ip: The ip address to search for.
     """
     @timer.timer
-    def __init__(self, ip):
+    def __init__(self, ip) -> None:
         try:
             ip = socket.gethostbyname(ip)
             url = f"https://ipinfo.io/{ip}/json"
diff --git a/utils/local_user_enum.py b/utils/local_user_enum.py
@@ -8,7 +8,7 @@ class Scan:
     Scans for local accounts on the system and their information.
     """
     @timer.timer
-    def __init__(self):
+    def __init__(self) -> None:
         if os.name == "nt":
             printer.info("Windows system detected..!\n")
             time.sleep(1)
diff --git a/utils/phonenumber_lookup.py b/utils/phonenumber_lookup.py
@@ -28,7 +28,7 @@ class LookUp:
     :param no: The phone number.
     """
     @timer.timer
-    def __init__(self, no):
+    def __init__(self, no) -> None:
         try:
             ph_no = p.parse(no)
             country = p.region_code_for_country_code(ph_no.country_code)
diff --git a/utils/port_scanner.py b/utils/port_scanner.py
@@ -32,7 +32,7 @@ class Scan:
     :param port_range: The range of ports to scan.
     """
     @timer.timer
-    def __init__(self, ip, port_range):
+    def __init__(self, ip, port_range) -> None:
         try:
             printer.info(f"Scanning for open ports for {Style.BRIGHT}{ip}{Style.RESET_ALL} with the port range of {Style.BRIGHT}1-{port_range}{Style.RESET_ALL}...")
             if port_range > 1000:
@@ -45,7 +45,7 @@ def __init__(self, ip, port_range):
         except KeyboardInterrupt:
             printer.error("Cancelled..!")
 
-    def scan(self, ip, port_range):
+    def scan(self, ip, port_range) -> None:
         """
         Scans for open ports in a given IP address.
 
@@ -60,7 +60,7 @@ def scan(self, ip, port_range):
                     printer.success(result)
 
     @staticmethod
-    def scan_port(ip, port):
+    def scan_port(ip, port) -> None:
         """
         Scans an individual port of a given IP address.
 
diff --git a/utils/search_username.py b/utils/search_username.py
@@ -29,15 +29,15 @@ class Search:
     :param username: The username to search for.
     """
     @timer.timer
-    def __init__(self, username):
+    def __init__(self, username) -> None:
         self.username = username
         try:
             self.scan(self.username)
         except KeyboardInterrupt:
             printer.error("Cancelled..!")
             pass
 
-    def scan(self, username):
+    def scan(self, username) -> str:
         """
         Scans for the given username across many different sites.
 
@@ -61,7 +61,7 @@ def scan(self, username):
 
         return user_json
 
-    async def make_requests(self, username):
+    async def make_requests(self, username) -> None:
         """
         Makes the requests to the sites.
 
@@ -75,7 +75,7 @@ async def make_requests(self, username):
             await asyncio.gather(*tasks)
 
     @staticmethod
-    async def make_request(session, u, username):
+    async def make_request(session, u, username) -> None:
         url = u["url"].format(username=username)
         json_body = None
         headers = {"User-Agent": f"{randomuser.GetUser()}"}
diff --git a/utils/smsbomber.py b/utils/smsbomber.py
@@ -29,7 +29,7 @@ class SMSBomber:
     :param count: number of SMS to send
     :param throttle: throttle time between each SMS
     """
-    def __init__(self, number, count, throttle):
+    def __init__(self, number, count, throttle) -> None:
         self.number = number
         self.count = int(count)
         self.throttle = int(throttle)
@@ -58,7 +58,7 @@ def __init__(self, number, count, throttle):
             printer.error(f"Error : {e}")
 
     @timer.timer
-    def start(self):
+    def start(self) -> None:
         successes = 0
         fails = 0
         try:
diff --git a/utils/web_scrape.py b/utils/web_scrape.py
@@ -29,7 +29,7 @@ class Scrape:
     
     :param url: url of the website.
     """
-    def __init__(self, url):
+    def __init__(self, url) -> None:
         self.url = url
         self.base_url = urlparse(url).netloc
         self.scraped_links = set()
@@ -53,17 +53,17 @@ def __init__(self, url):
         except KeyboardInterrupt:
             printer.error(f"Cancelled..!")
 
-    async def fetch(self, session, url):
+    async def fetch(self, session, url) -> str:
         headers = {"User-Agent": f"{randomuser.GetUser()}"}
         async with session.get(url, headers=headers) as response:
             return await response.text()
 
-    async def parse_links(self, content, base_url):
+    async def parse_links(self, content, base_url) -> set:
         soup = BeautifulSoup(content, "html.parser")
         links = soup.find_all("a")
         return [(urljoin(base_url, link.get("href")), link.text) for link in links]
 
-    async def scrape_links(self, url, recursive=False):
+    async def scrape_links(self, url, recursive=False) -> None:
         async with aiohttp.ClientSession() as session:
             html_content = await self.fetch(session, url)
             links = await self.parse_links(html_content, url)
diff --git a/utils/webhook_spammer.py b/utils/webhook_spammer.py
@@ -31,7 +31,7 @@ class Spam:
     :param throttle_interval: Time interval between sending messages (in seconds).
     """
     @timer.timer
-    def __init__(self, url, amount, message, username, throttle_interval=1):
+    def __init__(self, url, amount, message, username, throttle_interval=1) -> None:
         self.url = url
         self.amount = amount
         self.message = message
@@ -46,14 +46,14 @@ def __init__(self, url, amount, message, username, throttle_interval=1):
         except KeyboardInterrupt:
             printer.error("Cancelled..!")
 
-    async def send_message(self, session, data):
+    async def send_message(self, session, data) -> None:
         async with session.post(self.url, json=data) as response:
             if response.status == 204:
                 return True
             else:
                 return False
 
-    async def send_messages(self):
+    async def send_messages(self) -> None:
         data = {
             "content": str(self.message),
             "username": str(self.username),
diff --git a/utils/websearch.py b/utils/websearch.py
diff --git a/utils/whois_lookup.py b/utils/whois_lookup.py
diff --git a/utils/wifi_finder.py b/utils/wifi_finder.py
diff --git a/utils/wifi_vault.py b/utils/wifi_vault.py
