linted code

src/handler.py

@@ -11,7 +11,7 @@ import json
 import os
 
 from database import get_database
-from config import Config,get_config
+from config import Config, get_config
 from firewall.fwtype import FWType
 
 # imports for the __init_subclass__ method, do not remove pls
@@ -100,7 +100,6 @@ class Handler(BaseHTTPRequestHandler):
             error_codes = [400, 401, 403, 404, 500, 502, 503]
         return random.choice(error_codes)
 
-
     def _handle_sql_endpoint(self, path: str) -> bool:
         """
         Handle SQL injection honeypot endpoints.
@@ -245,7 +244,6 @@ class Handler(BaseHTTPRequestHandler):
         user_agent = self.headers.get("User-Agent", "")
         post_data = ""
 
-
         base_path = urlparse(self.path).path
 
         if base_path in ["/api/search", "/api/sql", "/api/database"]:
@@ -516,7 +514,11 @@ class Handler(BaseHTTPRequestHandler):
             self.end_headers()
             try:
                 stats = self.tracker.get_stats()
-                self.wfile.write(generate_dashboard(stats, self.config.dashboard_secret_path).encode())
+                self.wfile.write(
+                    generate_dashboard(
+                        stats, self.config.dashboard_secret_path
+                    ).encode()
+                )
             except BrokenPipeError:
                 pass
             except Exception as e:
@@ -563,7 +565,6 @@ class Handler(BaseHTTPRequestHandler):
             self.end_headers()
             try:
 
-
                 page = int(query_params.get("page", ["1"])[0])
                 page_size = int(query_params.get("page_size", ["25"])[0])
                 sort_by = query_params.get("sort_by", ["total_requests"])[0]
@@ -602,7 +603,6 @@ class Handler(BaseHTTPRequestHandler):
             self.end_headers()
             try:
 
-
                 # Parse query parameters
                 parsed_url = urlparse(self.path)
                 query_params = parse_qs(parsed_url.query)
@@ -766,7 +766,7 @@ class Handler(BaseHTTPRequestHandler):
                 result = db.get_top_ips_paginated(
                     page=page,
                     page_size=page_size,
-pathsort_by=sort_by,
+                    pathsort_by=sort_by,
                     sort_order=sort_order,
                 )
                 self.wfile.write(json.dumps(result).encode())
@@ -896,12 +896,12 @@ pathsort_by=sort_by,
 
         # API endpoint for downloading malicious IPs blocklist file
         if (
-            self.config.dashboard_secret_path and
-            request_path == f"{self.config.dashboard_secret_path}/api/get_banlist"
+            self.config.dashboard_secret_path
+            and request_path == f"{self.config.dashboard_secret_path}/api/get_banlist"
         ):
 
             # get fwtype from request params
-            fwtype = query_params.get("fwtype",["iptables"])[0]
+            fwtype = query_params.get("fwtype", ["iptables"])[0]
 
             # Query distinct suspicious IPs
             results = (
@@ -921,7 +921,10 @@ pathsort_by=sort_by,
 
             self.send_response(200)
             self.send_header("Content-type", "text/plain")
-            self.send_header("Content-Disposition", f'attachment; filename="{fwtype}.txt"',)
+            self.send_header(
+                "Content-Disposition",
+                f'attachment; filename="{fwtype}.txt"',
+            )
             self.send_header("Content-Length", str(len(banlist)))
             self.end_headers()
             self.wfile.write(banlist.encode())