diff --git a/pyproject.toml b/pyproject.toml index d41a4ab..7328e03 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "pyironscales" -version = "0.1.2" +version = "0.1.3" authors = [ { name="Peter Annabel", email="peter.annabel@gmail.com" }, ] diff --git a/src/pyironscales/endpoints/ironscales/CompanyIdStatsEndpoint.py b/src/pyironscales/endpoints/ironscales/CompanyIdStatsEndpoint.py index c9ce398..c8dfbcf 100644 --- a/src/pyironscales/endpoints/ironscales/CompanyIdStatsEndpoint.py +++ b/src/pyironscales/endpoints/ironscales/CompanyIdStatsEndpoint.py @@ -1,5 +1,4 @@ from pyironscales.endpoints.base.base_endpoint import IronscalesEndpoint -from pyironscales.endpoints.ironscales.CompanyIdStatsRemediationStatusesEndpoint import CompanyIdStatsRemediationStatusesEndpoint from pyironscales.interfaces import ( IGettable, ) @@ -17,7 +16,6 @@ class CompanyIdStatsEndpoint( def __init__(self, client, parent_endpoint=None) -> None: IronscalesEndpoint.__init__(self, client, "stats/", parent_endpoint=parent_endpoint) IGettable.__init__(self, CompanyStatisticsAndLicense) - self.remediation_statuses = self._register_child_endpoint(CompanyIdStatsRemediationStatusesEndpoint(client, parent_endpoint=self)) def get( self, diff --git a/src/pyironscales/endpoints/ironscales/CompanyIdStatsRemediationStatusesEndpoint.py b/src/pyironscales/endpoints/ironscales/CompanyIdStatsRemediationStatusesEndpoint.py deleted file mode 100644 index 5544207..0000000 --- a/src/pyironscales/endpoints/ironscales/CompanyIdStatsRemediationStatusesEndpoint.py +++ /dev/null @@ -1,38 +0,0 @@ -from pyironscales.endpoints.base.base_endpoint import IronscalesEndpoint - -from pyironscales.interfaces import ( - IGettable, -) -from pyironscales.models.ironscales import CompanyStatsRemediationStatuses -from pyironscales.types import ( - JSON, - IronscalesRequestParams, -) - - -class CompanyIdStatsRemediationStatusesEndpoint( - IronscalesEndpoint, - IGettable[CompanyStatsRemediationStatuses, IronscalesRequestParams], -): - def __init__(self, client, parent_endpoint=None) -> None: - IronscalesEndpoint.__init__(self, client, "remediation-statuses/", parent_endpoint=parent_endpoint) - IGettable.__init__(self, CompanyStatsRemediationStatuses) - - def get( - self, - data: JSON | None = None, - params: IronscalesRequestParams | None = None, - ) -> CompanyStatsRemediationStatuses: - """ - Performs a GET request against the /company/{id}/stats/remediation-statuses/ endpoint. - - Parameters: - data (dict[str, Any]): The data to send in the request body. - params (dict[str, int | str]): The parameters to send in the request query string. - Returns: - CompanyStatsRemediationStatuses: The parsed response data. - """ - return self._parse_one( - CompanyStatsRemediationStatuses, - super()._make_request("GET", data=data, params=params).json(), - ) diff --git a/src/pyironscales/models/ironscales/__init__.py b/src/pyironscales/models/ironscales/__init__.py index 16a160c..b3ffa0d 100644 --- a/src/pyironscales/models/ironscales/__init__.py +++ b/src/pyironscales/models/ironscales/__init__.py @@ -154,12 +154,6 @@ class CompanyStatisticsAndLicense(IronscalesModel): activeMailboxes: bool | None = Field(default=None, alias="activeMailboxes") lastMailboxSyncDate: datetime | None = Field(default=None, alias="lastMailboxSyncDate") -class CompanyStatsRemediationStatuses(IronscalesModel): - phishing: dict[str, int] | None = Field(default=None, alias="Phishing") - spam: dict[str, int] | None = Field(default=None, alias="Spam") - safe: dict[str, int] | None = Field(default=None, alias="Safe") - unclassified: dict[str, int] | None = Field(default=None, alias="Unclassified") - class EscalatedEmails(IronscalesModel): arrival_date: datetime | None = Field(default=None, alias="ArrivalDate") incident_id: int | None = Field(default=None, alias="IncidentId")