github action adjustements

mIcHyAmRaNe · Jul 12, 2024 · e52b436 · e52b436
1 parent b332530
commit e52b436
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 14 deletions.
diff --git a/.github/workflows/build-debian.yml b/.github/workflows/build-debian.yml
@@ -26,6 +26,10 @@ jobs:
           tar czf okadminfinder_${{ github.ref_name }}.orig.tar.gz okadminfinder
           ls -ll
 
+      - name: Clean before build
+        run: |
+          fakeroot debian/rules clean
+
       - name: Build Debian package
         run: |
           dpkg-buildpackage -rfakeroot -uc -us

diff --git a/okadminfinder/_classes.py b/okadminfinder/_classes.py
@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
+
 from os.path import dirname, isfile
 from sys import exit
 import contextlib
@@ -67,7 +68,9 @@ def get_links(dns=False, wordlist=None, debug=False):
         links = []
 
         if dns and not wordlist:
-            links_path = str(dirname(meta.__file__) + "/LinkFile/subdomains-top1million-5000.txt")
+            links_path = str(
+                dirname(meta.__file__) + "/LinkFile/subdomains-top1million-5000.txt"
+            )
         elif wordlist:
             if not isfile(wordlist):
                 print(red, "\n\tError: The wordlist file does not exist.", RESET_ALL)
@@ -90,7 +93,7 @@ def get_links(dns=False, wordlist=None, debug=False):
                     line = f"{{}}/{line}"
                     links.append(line.replace("\n", ""))
             if debug:
-                okadminfinder.get_debug("LINKS: ",links)
+                okadminfinder.get_debug("LINKS: ", links)
         return links
 
     def create_link(website, dns=False, wordlist=None, debug=False):
@@ -110,7 +113,7 @@ def create_link(website, dns=False, wordlist=None, debug=False):
         host = url.host
         if url.port:
             host += f":{url.port}"
-        
+
         # Handle case where the URL starts with 'www.'
         if host.startswith("www."):
             website = host.replace("www.", "")
@@ -122,7 +125,7 @@ def create_link(website, dns=False, wordlist=None, debug=False):
             for n in okadminfinder.get_links(dns, wordlist, debug):
                 req_link = url.scheme + "://" + n.format(website)
                 reqlinks.append(req_link.replace("\n", ""))
-        
+
         # debug mode
         if debug:
             okadminfinder.get_debug("ReqLinks: ", reqlinks)
@@ -173,7 +176,9 @@ def proxy(self, prox, headers):
             exit(0)
         return proxies
 
-    async def url(self, website, headers, proxies, dns=False, wordlist=None, debug=False):
+    async def url(
+        self, website, headers, proxies, dns=False, wordlist=None, debug=False
+    ):
         try:
             if okadminfinder.check_url(website, headers, proxies):
                 print(
@@ -223,7 +228,7 @@ async def url(self, website, headers, proxies, dns=False, wordlist=None, debug=F
                         httpx.NetworkError,
                         httpx.ReadTimeout,
                         httpx.ConnectTimeout,
-                        httpx.ProxyError
+                        httpx.ProxyError,
                     ):
                         continue
                 pbar.close()
@@ -232,7 +237,11 @@ async def url(self, website, headers, proxies, dns=False, wordlist=None, debug=F
             print("\t╚═══[∞︎︎]", str(total_count), "Total pages scanned")
         except (httpx.RemoteProtocolError):
             pbar.close()
-            print(red, "\n\tSession Canceled: Server disconnected without sending a response.", RESET_ALL)
+            print(
+                red,
+                "\n\tSession Canceled: Server disconnected without sending a response.",
+                RESET_ALL,
+            )
             exit(0)
         except (KeyboardInterrupt):
             pbar.close()

diff --git a/okadminfinder/okadminfinder.py b/okadminfinder/okadminfinder.py
@@ -64,11 +64,7 @@ def main():
             help="Use randomly selected User-Agent",
         )
         parser.add_argument(
-            "-D",
-            "--debug",
-            action="store_true",
-            default=False,
-            help="Debug mode"
+            "-D", "--debug", action="store_true", default=False, help="Debug mode"
         )
         if len(sys.argv) <= 1:
             parser.print_usage()
@@ -89,7 +85,7 @@ def main():
                     quit(0)
                 elif args.tor is not False:
                     parser.print_usage()
-                    print(" \n use only Tor or only Proxy not both at a time") 
+                    print(" \n use only Tor or only Proxy not both at a time")
                     quit(0)
                 else:
                     if str(args.proxy)[0:5] == "socks":

diff --git a/tests/test_url.py b/tests/test_url.py
@@ -1,5 +1,7 @@
-import pytest
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
 
+import pytest
 from okadminfinder._classes import okadminfinder