products.py

import json
from types import NoneType
from typing import Optional, Union, Annotated

from ..models import *
from ..base_model import Page, Service
from pydantic import BaseModel, Field
from pydantic.fields import FieldInfo
from httpx import Auth
from ..http_client import HttpClient

class Products(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_service_api/v0.15.27")
    
    def get_products(
        self,
        query: Optional[str] = None,
        page: int = 1,
        size: int = 50,
    )-> LookupListWithStatsResponsePage:
        endpoint_url = "/products"
        loc = locals()
        headers = {}
        params = json.loads(
            ProductsGetProductsQueryParameters(**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 LookupListWithStatsResponsePage(_client=self, **response.json())
    
    def get_product_impact(
        self,
        product: str,
        sort_by: Optional[GetCVEsSortBy] = GetCVEsSortBy("rule_release_date"),
        sort_order: Optional[GetCVEsSortOrder] = GetCVEsSortOrder("desc"),
        page: int = 1,
        size: int = 50,
    )-> LookupImpactResponsePage:
        endpoint_url = "/products/{product}"
        loc = locals()
        headers = {}
        params = json.loads(
            ProductsGetProductImpactQueryParameters(**loc).model_dump_json(
                exclude_none=True
            )
        )
        path_params = json.loads(
            ProductsGetProductImpactPathParameters(**loc).model_dump_json(
                exclude_none=True
            )
        )
        
        response = self.http_client.get(
            url=endpoint_url, path_params=path_params, params=params, headers=headers
        )
        
        return LookupImpactResponsePage(_client=self, **response.json())