updated onelens backend python client

Author: pranjal-astuto

onelens_backend_client_v2/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  <stdin>
-#   timestamp: 2026-01-19T10:11:23+00:00
+#   timestamp: 2026-01-21T10:22:35+00:00
 
 from __future__ import annotations
 
@@ -1981,6 +1981,18 @@ class GetAgentRunConfigRequest(BaseModel):
     )
 
 
+class GetAggregatedPolicyStatsAPIRequest(BaseModel):
+    pass
+
+
+class GetAggregatedPolicyStatsRequest(BaseModel):
+    tenant_id: UUID = Field(..., title="Tenant Id")
+
+
+class GetAggregatedPolicyStatsResponse(BaseModel):
+    stats: Dict[str, Any] = Field(..., title="Stats")
+
+
 class GetAggregatedTicketsStatsResponse(BaseModel):
     stat_name: str = Field(..., title="Stat Name")
     total: float = Field(..., title="Total")
@@ -4182,6 +4194,11 @@ class PolicyTicketStatus(str, Enum):
     DEPRECATED = "DEPRECATED"
 
 
+class PolicyType(str, Enum):
+    POLICY = "POLICY"
+    INSIGHTS = "INSIGHTS"
+
+
 class PostNaviraLogsRequest(BaseModel):
     pagination: Optional[PaginationParams] = Field(
         {"page": 1, "page_size": 10},
@@ -5289,6 +5306,12 @@ class ResponseEnableTenantPolicyResponse(BaseModel):
     status_code: Optional[int] = Field(200, title="Status Code")
 
 
+class ResponseGetAggregatedPolicyStatsResponse(BaseModel):
+    data: GetAggregatedPolicyStatsResponse
+    message: Optional[str] = Field(None, title="Message")
+    status_code: Optional[int] = Field(200, title="Status Code")
+
+
 class ResponseGetAggregatedTicketsStatsResponse(BaseModel):
     data: GetAggregatedTicketsStatsResponse
     message: Optional[str] = Field(None, title="Message")
@@ -5907,6 +5930,27 @@ class SyncRecommendationUnitFromRepoResponse(BaseModel):
     )
 
 
+class SyncTenantPoliciesDataRequest(BaseModel):
+    policy_ids: List[UUID4] = Field(
+        ..., description="The ids of the policies.", title="Policy Ids"
+    )
+    tenant_id: UUID4 = Field(..., title="Tenant Id")
+
+
+class SyncTenantPoliciesDataResponse(BaseModel):
+    policies_synced: bool = Field(
+        ..., description="Whether the policies were synced", title="Policies Synced"
+    )
+    policies_synced_count: int = Field(
+        ..., description="The number of policies synced", title="Policies Synced Count"
+    )
+    policies_not_synced_count: int = Field(
+        ...,
+        description="The number of policies not synced",
+        title="Policies Not Synced Count",
+    )
+
+
 class SyncTicketDataRequest(BaseModel):
     ticket_ids: List[UUID] = Field(
         ..., description="The unique identifiers of the tickets", title="Ticket Ids"
@@ -7274,6 +7318,10 @@ class UploadedFileDetails(BaseModel):
     )
 
 
+class UpsertAggregatedPoliciesResponse(BaseModel):
+    pass
+
+
 class UpsertAggregatedTicketsResponse(BaseModel):
     pass
 
@@ -8076,6 +8124,33 @@ class AggregatedBucketMetrics(BaseModel):
     total_objects_sum: int = Field(..., title="Total Objects Sum")
 
 
+class AggregatedPoliciesMixin(BaseModel):
+    policy_ol_id: UUID = Field(
+        ..., description="The unique identifier of the policy", title="Policy Ol Id"
+    )
+    title: str = Field(..., description="The title of the policy", title="Title")
+    alias: str = Field(..., description="The alias of the policy", title="Alias")
+    description: str = Field(
+        ..., description="The description of the policy", title="Description"
+    )
+    category: PolicyCategory = Field(..., description="The category of the policy")
+    subcategory: OnelensModelsServiceInterfacesPoliciesCommonsSubcategory = Field(
+        ..., description="The subcategory of the policy"
+    )
+    system_state: TenantPolicySystemState = Field(
+        ..., description="The system state of the policy"
+    )
+    state: TenantPolicyState = Field(..., description="The state of the policy")
+    service: Union[AwsService, GcpService] = Field(
+        ..., description="The service of the policy", title="Service"
+    )
+    provider: Provider = Field(..., description="The provider of the policy")
+    policy_type: PolicyType = Field(..., description="The type of the policy")
+    last_run_on: Optional[datetime] = Field(
+        None, description="The last run on date of the policy", title="Last Run On"
+    )
+
+
 class AggregatedStorageMetrics(BaseModel):
     ol_id: str = Field(..., title="Ol Id")
     metrics_by_storage: List[StorageMetrics] = Field(..., title="Metrics By Storage")
@@ -9907,6 +9982,11 @@ class GetActionTypeResponse(BaseModel):
     pagination: PaginationFields = Field(..., description="Pagination fields")
 
 
+class GetAggregatedPoliciesAPIResponse(BaseModel):
+    pagination: PaginationFields = Field(..., description="Pagination fields.")
+    data: List[Dict[str, Any]] = Field(..., title="Data")
+
+
 class GetAggregatedTicketsAPIResponse(BaseModel):
     pagination: PaginationFields = Field(..., description="Pagination fields.")
     data: List[Dict[str, Any]] = Field(..., title="Data")
@@ -12596,6 +12676,12 @@ class ResponseFetchTenantUsersResponse(BaseModel):
     status_code: Optional[int] = Field(200, title="Status Code")
 
 
+class ResponseGetAggregatedPoliciesAPIResponse(BaseModel):
+    data: GetAggregatedPoliciesAPIResponse
+    message: Optional[str] = Field(None, title="Message")
+    status_code: Optional[int] = Field(200, title="Status Code")
+
+
 class ResponseGetAggregatedTicketsAPIResponse(BaseModel):
     data: GetAggregatedTicketsAPIResponse
     message: Optional[str] = Field(None, title="Message")
@@ -14234,6 +14320,15 @@ class UpdateUserResponse(BaseModel):
     id: UUID4 = Field(..., description="Unique identifier for the user", title="Id")
 
 
+class UpsertAggregatedPoliciesRequest(BaseModel):
+    tenant_id: UUID = Field(
+        ..., description="The unique identifier of the tenant", title="Tenant Id"
+    )
+    aggregated_policies: List[AggregatedPoliciesMixin] = Field(
+        ..., description="The aggregated policies", title="Aggregated Policies"
+    )
+
+
 class UpsertAggregatedTicketsRequest(BaseModel):
     tenant_id: UUID = Field(
         ..., description="The unique identifier of the tenant", title="Tenant Id"
@@ -15223,6 +15318,29 @@ class GetAgentRunConfigResponse(BaseModel):
     )
 
 
+class GetAggregatedPoliciesAPIRequest(BaseModel):
+    pagination: Optional[PaginationParams] = Field(
+        {"page": 1, "page_size": 10},
+        description="Pagination parameters for the request.",
+    )
+    filters: Optional[
+        List[OnelensDomainUtilitiesRepositoriesDynamicFiltersFilterCriteria]
+    ] = Field(None, title="Filters")
+    sort_criteria: Optional[SortCriteria] = None
+
+
+class GetAggregatedPoliciesRequest(BaseModel):
+    pagination: Optional[PaginationParams] = Field(
+        {"page": 1, "page_size": 10},
+        description="Pagination parameters for the request.",
+    )
+    filters: Optional[
+        List[OnelensDomainUtilitiesRepositoriesDynamicFiltersFilterCriteria]
+    ] = Field(None, title="Filters")
+    sort_criteria: Optional[SortCriteria] = None
+    tenant_id: UUID = Field(..., title="Tenant Id")
+
+
 class GetAggregatedTicketsAPIRequest(BaseModel):
     pagination: Optional[PaginationParams] = Field(
         {"page": 1, "page_size": 10},
@@ -19904,17 +20022,17 @@ class CreateCustomTenantTicketRequestMixin(BaseModel):
         title="Post Policy Execution Last Run",
     )
     last_run_id: Optional[UUID4] = Field(
-        "d42994fb-4730-48ec-93b3-8c53da91553d",
+        "36acc280-dc7a-44a5-90e1-b6812172b7ab",
         description="Id of the last policy violation/anomaly run",
         title="Last Run Id",
     )
     last_run_at: Optional[datetime] = Field(
-        "2026-01-19T15:40:47.753395",
+        "2026-01-21T15:52:05.844398",
         description="Datetime of the last policy violation/anomaly run",
         title="Last Run At",
     )
     first_run_at: Optional[datetime] = Field(
-        "2026-01-19T15:40:47.753410",
+        "2026-01-21T15:52:05.844412",
         description="Datetime of the first policy violation/anomaly run",
         title="First Run At",
     )

onelens_backend_client_v2/rpc/__init__.py

@@ -208,6 +208,10 @@
     AggregatedTicketsServiceRpcHandler,
 )
 
+from onelens_backend_client_v2.rpc.aggregated_policies_service_rpc_handler import (
+    AggregatedPoliciesServiceRpcHandler,
+)
+
 
 __all__ = [
     "TenantVerifyServiceRpcHandler",
@@ -263,4 +267,5 @@
     "AuthServiceRpcHandler",
     "TenantOnboardingServiceRpcHandler",
     "AggregatedTicketsServiceRpcHandler",
+    "AggregatedPoliciesServiceRpcHandler",
 ]