fix version

Author: he2ss

crowdsec_tracker_api/services/cves.py

@@ -1,89 +1,89 @@
 import json
 from types import NoneType
-from typing import Optional, Union, Annotated
+from typing import Annotated, Optional, Union
 
-from ..models import *
-from ..base_model import Page, Service
+from httpx import Auth
 from pydantic import BaseModel, Field
 from pydantic.fields import FieldInfo
-from httpx import Auth
+
+from ..base_model import Page, Service
 from ..http_client import HttpClient
+from ..models import *
+
 
 class Cves(Service):
-    def __init__(self, auth: Auth, base_url: str = "https://admin.api.crowdsec.net/v1") -> None:
-        super().__init__(base_url=base_url, auth=auth, user_agent="crowdsec_tracker_api/1.95.1")
-    
+    def __init__(
+        self, auth: Auth, base_url: str = "https://admin.api.crowdsec.net/v1"
+    ) -> None:
+        super().__init__(
+            base_url=base_url, auth=auth, user_agent="crowdsec_tracker_api/1.95.2"
+        )
+
     def get_cves(
         self,
         query: Optional[str] = None,
         sort_by: Optional[GetCVEsSortBy] = GetCVEsSortBy("rule_release_date"),
         sort_order: Optional[GetCVEsSortOrder] = GetCVEsSortOrder("desc"),
         page: int = 1,
         size: int = 50,
-    )-> GetCVEsResponsePage:
+    ) -> GetCVEsResponsePage:
         endpoint_url = "/cves"
         loc = locals()
         headers = {}
         params = json.loads(
-            CvesGetCvesQueryParameters(**loc).model_dump_json(
-                exclude_none=True
-            )
+            CvesGetCvesQueryParameters(**loc).model_dump_json(exclude_none=True)
         )
         path_params = {}
-        
+
         response = self.http_client.get(
             url=endpoint_url, path_params=path_params, params=params, headers=headers
         )
-        
+
         return GetCVEsResponsePage(_client=self, **response.json())
-    
+
     def get_cve(
         self,
         cve_id: str,
-    )-> GetCVEResponse:
+    ) -> GetCVEResponse:
         endpoint_url = "/cves/{cve_id}"
         loc = locals()
         headers = {}
         params = {}
         path_params = json.loads(
-            CvesGetCvePathParameters(**loc).model_dump_json(
-                exclude_none=True
-            )
+            CvesGetCvePathParameters(**loc).model_dump_json(exclude_none=True)
         )
-        
+
         response = self.http_client.get(
             url=endpoint_url, path_params=path_params, params=params, headers=headers
         )
-        
+
         return GetCVEResponse(**response.json())
-    
+
     def download_cve_ips(
         self,
         cve_id: str,
-    )-> str:
+    ) -> str:
         endpoint_url = "/cves/{cve_id}/ips-download"
         loc = locals()
         headers = {}
         params = {}
         path_params = json.loads(
-            CvesDownloadCveIpsPathParameters(**loc).model_dump_json(
-                exclude_none=True
-            )
+            CvesDownloadCveIpsPathParameters(**loc).model_dump_json(exclude_none=True)
         )
-        
+
         response = self.http_client.get(
             url=endpoint_url, path_params=path_params, params=params, headers=headers
         )
-        
+
         return response.text
-    
+
     def get_cve_ips_details(
         self,
         cve_id: str,
         since: Optional[str] = "14d",
         page: int = 1,
         size: int = 50,
-    )-> GetCVEIPsResponsePage:
+    ) -> GetCVEIPsResponsePage:
         endpoint_url = "/cves/{cve_id}/ips-details"
         loc = locals()
         headers = {}
@@ -93,23 +93,21 @@ def get_cve_ips_details(
             )
         )
         path_params = json.loads(
-            CvesGetCveIpsDetailsPathParameters(**loc).model_dump_json(
-                exclude_none=True
-            )
+            CvesGetCveIpsDetailsPathParameters(**loc).model_dump_json(exclude_none=True)
         )
-        
+
         response = self.http_client.get(
             url=endpoint_url, path_params=path_params, params=params, headers=headers
         )
-        
+
         return GetCVEIPsResponsePage(_client=self, **response.json())
-    
+
     def get_cve_subscribed_integrations(
         self,
         cve_id: str,
         page: int = 1,
         size: int = 50,
-    )-> GetCVESubscribedIntegrationsResponsePage:
+    ) -> GetCVESubscribedIntegrationsResponsePage:
         endpoint_url = "/cves/{cve_id}/integrations"
         loc = locals()
         headers = {}
@@ -123,13 +121,13 @@ def get_cve_subscribed_integrations(
                 exclude_none=True
             )
         )
-        
+
         response = self.http_client.get(
             url=endpoint_url, path_params=path_params, params=params, headers=headers
         )
-        
+
         return GetCVESubscribedIntegrationsResponsePage(_client=self, **response.json())
-    
+
     def subscribe_integration_to_cve(
         self,
         request: SubscribeCVEIntegrationRequest,
@@ -144,18 +142,22 @@ def subscribe_integration_to_cve(
                 exclude_none=True
             )
         )
-        
-        payload = json.loads(
-            request.model_dump_json(
-                exclude_none=True
-            )
-        ) if "request" in loc else None
+
+        payload = (
+            json.loads(request.model_dump_json(exclude_none=True))
+            if "request" in loc
+            else None
+        )
         response = self.http_client.post(
-            url=endpoint_url, path_params=path_params, params=params, headers=headers, json=payload
+            url=endpoint_url,
+            path_params=path_params,
+            params=params,
+            headers=headers,
+            json=payload,
         )
-        
+
         return None
-    
+
     def unsubscribe_integration_from_cve(
         self,
         cve_id: str,
@@ -170,35 +172,30 @@ def unsubscribe_integration_from_cve(
                 exclude_none=True
             )
         )
-        
+
         response = self.http_client.delete(
             url=endpoint_url, path_params=path_params, params=params, headers=headers
         )
-        
+
         return None
-    
+
     def get_cve_timeline(
         self,
         cve_id: str,
         since_days: SinceOptions,
-    )-> list[TimelineItem]:
+    ) -> list[TimelineItem]:
         endpoint_url = "/cves/{cve_id}/timeline"
         loc = locals()
         headers = {}
         params = json.loads(
-            CvesGetCveTimelineQueryParameters(**loc).model_dump_json(
-                exclude_none=True
-            )
+            CvesGetCveTimelineQueryParameters(**loc).model_dump_json(exclude_none=True)
         )
         path_params = json.loads(
-            CvesGetCveTimelinePathParameters(**loc).model_dump_json(
-                exclude_none=True
-            )
+            CvesGetCveTimelinePathParameters(**loc).model_dump_json(exclude_none=True)
         )
-        
+
         response = self.http_client.get(
             url=endpoint_url, path_params=path_params, params=params, headers=headers
         )
-        
+
         return [TimelineItem(**item) for item in response.json()]
-