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

souleyez/storage/osint.py

@@ -21,7 +21,7 @@ class OsintManager:
         target: str = None,
         summary: str = None,
         content: str = None,
-        metadata: str = None
+        metadata: str = None,
     ) -> int:
         """
         Add OSINT data to the database.
@@ -42,7 +42,7 @@ class OsintManager:
         # Check if this exact entry already exists
         existing = self.db.execute_one(
             "SELECT id FROM osint_data WHERE engagement_id = ? AND data_type = ? AND value = ?",
-            (engagement_id, data_type, value)
+            (engagement_id, data_type, value),
         )
 
         if existing:
@@ -66,32 +66,28 @@ class OsintManager:
                 params.append(metadata)
 
             if updates:
-                params.append(existing['id'])
+                params.append(existing["id"])
                 self.db.execute(
                     f"UPDATE osint_data SET {', '.join(updates)} WHERE id = ?",
-                    tuple(params)
+                    tuple(params),
                 )
-            return existing['id']
+            return existing["id"]
 
         # Insert new data
-        data = {
-            'engagement_id': engagement_id,
-            'data_type': data_type,
-            'value': value
-        }
+        data = {"engagement_id": engagement_id, "data_type": data_type, "value": value}
 
         if source:
-            data['source'] = source
+            data["source"] = source
         if target:
-            data['target'] = target
+            data["target"] = target
         if summary:
-            data['summary'] = summary
+            data["summary"] = summary
         if content:
-            data['content'] = content
+            data["content"] = content
         if metadata:
-            data['metadata'] = metadata
+            data["metadata"] = metadata
 
-        return self.db.insert('osint_data', data)
+        return self.db.insert("osint_data", data)
 
     def bulk_add_osint_data(
         self,
@@ -99,7 +95,7 @@ class OsintManager:
         data_type: str,
         values: List[str],
         source: str = None,
-        target: str = None
+        target: str = None,
     ) -> int:
         """
         Add multiple OSINT data entries of the same type.
@@ -119,7 +115,7 @@ class OsintManager:
             # Check if exists
             existing = self.db.execute_one(
                 "SELECT id FROM osint_data WHERE engagement_id = ? AND data_type = ? AND value = ?",
-                (engagement_id, data_type, value)
+                (engagement_id, data_type, value),
             )
             if not existing:
                 self.add_osint_data(engagement_id, data_type, value, source, target)
@@ -132,10 +128,7 @@ class OsintManager:
         return self.db.execute_one(query, (osint_id,))
 
     def list_osint_data(
-        self,
-        engagement_id: int,
-        data_type: str = None,
-        source: str = None
+        self, engagement_id: int, data_type: str = None, source: str = None
     ) -> List[Dict[str, Any]]:
         """
         List OSINT data with optional filters.
@@ -180,8 +173,8 @@ class OsintManager:
 
         summary = {}
         for row in results:
-            data_type = row.get('data_type', 'unknown')
-            count = row.get('count', 0)
+            data_type = row.get("data_type", "unknown")
+            count = row.get("count", 0)
             summary[data_type] = count
 
         return summary