souleyez 2.43.29__py3-none-any.whl → 2.43.32__py3-none-any.whl

souleyez/storage/web_paths.py

@@ -18,7 +18,7 @@ class WebPathsManager:
         url: str,
         status_code: int = None,
         content_length: int = None,
-        redirect: str = None
+        redirect: str = None,
     ) -> int:
         """
         Add a web path to the database.
@@ -35,50 +35,45 @@ class WebPathsManager:
         """
         # Check if this exact path already exists for this host
         existing = self.db.execute_one(
-            "SELECT id FROM web_paths WHERE host_id = ? AND url = ?",
-            (host_id, url)
+            "SELECT id FROM web_paths WHERE host_id = ? AND url = ?", (host_id, url)
         )
 
         if existing:
             # Update if status or length changed
-            if status_code is not None or content_length is not None or redirect is not None:
+            if (
+                status_code is not None
+                or content_length is not None
+                or redirect is not None
+            ):
                 update_data = {}
                 if status_code is not None:
-                    update_data['status_code'] = status_code
+                    update_data["status_code"] = status_code
                 if content_length is not None:
-                    update_data['content_length'] = content_length
+                    update_data["content_length"] = content_length
                 if redirect is not None:
-                    update_data['redirect'] = redirect
+                    update_data["redirect"] = redirect
 
                 if update_data:
-                    set_clause = ', '.join([f"{k} = ?" for k in update_data.keys()])
-                    values = list(update_data.values()) + [existing['id']]
+                    set_clause = ", ".join([f"{k} = ?" for k in update_data.keys()])
+                    values = list(update_data.values()) + [existing["id"]]
                     self.db.execute(
-                        f"UPDATE web_paths SET {set_clause} WHERE id = ?",
-                        tuple(values)
+                        f"UPDATE web_paths SET {set_clause} WHERE id = ?", tuple(values)
                     )
-            return existing['id']
+            return existing["id"]
 
         # Insert new path
-        data = {
-            'host_id': host_id,
-            'url': url
-        }
+        data = {"host_id": host_id, "url": url}
 
         if status_code is not None:
-            data['status_code'] = status_code
+            data["status_code"] = status_code
         if content_length is not None:
-            data['content_length'] = content_length
+            data["content_length"] = content_length
         if redirect is not None:
-            data['redirect'] = redirect
+            data["redirect"] = redirect
 
-        return self.db.insert('web_paths', data)
+        return self.db.insert("web_paths", data)
 
-    def bulk_add_web_paths(
-        self,
-        host_id: int,
-        paths: List[Dict[str, Any]]
-    ) -> int:
+    def bulk_add_web_paths(self, host_id: int, paths: List[Dict[str, Any]]) -> int:
         """
         Add multiple web paths for a host.
 
@@ -94,15 +89,15 @@ class WebPathsManager:
             # Check if exists
             existing = self.db.execute_one(
                 "SELECT id FROM web_paths WHERE host_id = ? AND url = ?",
-                (host_id, path.get('url'))
+                (host_id, path.get("url")),
             )
             if not existing:
                 self.add_web_path(
                     host_id,
-                    path.get('url'),
-                    path.get('status_code'),
-                    path.get('size'),  # gobuster uses 'size' field
-                    path.get('redirect')  # add redirect support
+                    path.get("url"),
+                    path.get("status_code"),
+                    path.get("size"),  # gobuster uses 'size' field
+                    path.get("redirect"),  # add redirect support
                 )
                 count += 1
         return count
@@ -113,10 +108,7 @@ class WebPathsManager:
         return self.db.execute_one(query, (path_id,))
 
     def list_web_paths(
-        self,
-        host_id: int = None,
-        engagement_id: int = None,
-        status_code: int = None
+        self, host_id: int = None, engagement_id: int = None, status_code: int = None
     ) -> List[Dict[str, Any]]:
         """
         List web paths with optional filters.
@@ -150,7 +142,11 @@ class WebPathsManager:
             params = []
 
         if status_code is not None:
-            query += " AND wp.status_code = ?" if engagement_id or host_id else " AND status_code = ?"
+            query += (
+                " AND wp.status_code = ?"
+                if engagement_id or host_id
+                else " AND status_code = ?"
+            )
             params.append(status_code)
 
         query += " ORDER BY wp.created_at DESC"
@@ -175,8 +171,8 @@ class WebPathsManager:
 
         summary = {}
         for row in results:
-            status = str(row.get('status_code', 'unknown'))
-            count = row.get('count', 0)
+            status = str(row.get("status_code", "unknown"))
+            count = row.get("count", 0)
             summary[status] = count
 
         return summary