added vpc flow log models

chetanbb-astuto · chetanbb-astuto · commit c15c7a0d4d6c · 2026-04-22T18:32:26.000+05:30
diff --git a/onelens_backend_client_v2/models.py b/onelens_backend_client_v2/models.py
@@ -22522,4 +22522,65 @@ class ResponseGetUserCustomTicketByIdResponse(BaseModel):
     status_code: Optional[int] = Field(200, title="Status Code")
 
 
+class VpcFlowLogViolationMixin(BaseModel):
+    raw_insight_id: str = Field(..., title="Raw Insight Id")
+    insight_type: str = Field(..., title="Insight Type")
+    account_id: str = Field(..., title="Account Id")
+    region: str = Field(..., title="Region")
+    vpc_id: Optional[str] = Field(None, title="Vpc Id")
+    resource: Optional[str] = Field(None, title="Resource")
+    source_ip: Optional[str] = Field(None, title="Source Ip")
+    destination: Optional[str] = Field(None, title="Destination")
+    destination_ip: Optional[str] = Field(None, title="Destination Ip")
+    service: Optional[str] = Field(None, title="Service")
+    transfer_type: Optional[str] = Field(None, title="Transfer Type")
+    source_az: Optional[str] = Field(None, title="Source Az")
+    destination_az: Optional[str] = Field(None, title="Destination Az")
+    bytes_transferred: Optional[int] = Field(None, title="Bytes Transferred")
+    networking_cost: Optional[float] = Field(None, title="Networking Cost")
+    potential_monthly_savings: Optional[float] = Field(0.0, title="Potential Monthly Savings")
+    recommendation_title: Optional[str] = Field(None, title="Recommendation Title")
+    recommendation_description: Optional[str] = Field(None, title="Recommendation Description")
+    effort_level: Optional[str] = Field(None, title="Effort Level")
+    start_date: Optional[date] = Field(None, title="Start Date")
+    end_date: Optional[date] = Field(None, title="End Date")
+    resource_ol_id: Optional[UUID] = Field(None, title="Resource Ol Id")
+    account_name: Optional[str] = Field(None, title="Account Name")
+    risk: Optional[str] = Field(None, title="Risk")
+    priority: Optional[str] = Field(None, title="Priority")
+    effort: Optional[str] = Field(None, title="Effort")
+
+
+class SyncVpcFlowLogViolationsRequest(BaseModel):
+    tenant_id: UUID = Field(..., title="Tenant Id")
+    request_id: str = Field(..., title="Request Id")
+    account_id: str = Field(..., title="Account Id")
+    region: str = Field(..., title="Region")
+    violations: List[VpcFlowLogViolationMixin] = Field(..., title="Violations")
+
+
+class SyncVpcFlowLogViolationsResponse(BaseModel):
+    inserted_violations: int = Field(..., title="Inserted Violations")
+    created_ticket_count: int = Field(..., title="Created Ticket Count")
+    pending_verification_count: int = Field(..., title="Pending Verification Count")
+
+
+class BulkUpdateVpcFlowLogViolationTicketsRequest(BaseModel):
+    tenant_id: UUID = Field(..., title="Tenant Id")
+    ticket_ids: List[UUID] = Field(..., title="Ticket Ids")
+    status: Optional[PolicyTicketStatus] = None
+    priority: Optional[Priority] = None
+    assigned_to: Optional[UUID] = Field(None, title="Assigned To")
+    updated_by: Optional[UUID] = Field(None, title="Updated By")
+    achieved_savings: Optional[float] = Field(None, title="Achieved Savings")
+    achieved_savings_on: Optional[datetime] = Field(None, title="Achieved Savings On")
+
+
+class BulkUpdateVpcFlowLogViolationTicketsResponse(BaseModel):
+    successful_ticket_ids: List[UUID] = Field(..., title="Successful Ticket Ids")
+    failed_ticket_ids: List[UUID] = Field(..., title="Failed Ticket Ids")
+    message: str = Field(..., title="Message")
+    status_code: int = Field(..., title="Status Code")
+
+
 GroupData.model_rebuild()
diff --git a/onelens_backend_client_v2/rpc/__init__.py b/onelens_backend_client_v2/rpc/__init__.py
@@ -224,6 +224,10 @@
     NetworkFlowAnalysisRequestsServiceRpcHandler,
 )
 
+from onelens_backend_client_v2.rpc.vpc_flow_log_violations_service_rpc_handler import (
+    VpcFlowLogViolationsServiceRpcHandler,
+)
+
 
 __all__ = [
     "AzureAdvisorTicketsServiceRpcHandler",
@@ -283,4 +287,5 @@
     "AggregatedTicketsServiceRpcHandler",
     "AggregatedPoliciesServiceRpcHandler",
     "NetworkFlowAnalysisRequestsServiceRpcHandler",
+    "VpcFlowLogViolationsServiceRpcHandler",
 ]
diff --git a/onelens_backend_client_v2/rpc/vpc_flow_log_violations_service_rpc_handler.py b/onelens_backend_client_v2/rpc/vpc_flow_log_violations_service_rpc_handler.py
@@ -0,0 +1,273 @@
+# VpcFlowLogViolationsServiceRpcHandler API
+
+from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
+from typing import Any, Dict, List, Optional, Tuple, Union
+from typing_extensions import Annotated
+
+
+from onelens_backend_client_v2.models import BulkUpdateVpcFlowLogViolationTicketsRequest
+
+
+from onelens_backend_client_v2.models import (
+    BulkUpdateVpcFlowLogViolationTicketsResponse,
+)
+
+
+from onelens_backend_client_v2.models import SyncVpcFlowLogViolationsRequest
+
+
+from onelens_backend_client_v2.models import SyncVpcFlowLogViolationsResponse
+
+
+from onelens_backend_client_v2.api_client import ApiClient, RequestSerialized
+
+
+class VpcFlowLogViolationsServiceRpcHandler:
+    """NOTE: This class is auto generated. Do not edit the class manually."""
+
+    def __init__(self, api_client=None) -> None:
+        if api_client is None:
+            api_client = ApiClient.get_default()
+        self.api_client = api_client
+
+    @validate_call
+    def bulk_update_vpc_flow_log_violation_tickets(
+        self,
+        request: BulkUpdateVpcFlowLogViolationTicketsRequest,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> BulkUpdateVpcFlowLogViolationTicketsResponse:
+        """Bulk Update Vpc Flow Log Violation Tickets
+
+
+
+        :param request: (required)
+        :type request: BulkUpdateVpcFlowLogViolationTicketsRequest
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """
+
+        _param = self._bulk_update_vpc_flow_log_violation_tickets_serialize(
+            request=request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "BulkUpdateVpcFlowLogViolationTicketsResponse",
+            "422": "HTTPValidationError",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+    def _bulk_update_vpc_flow_log_violation_tickets_serialize(
+        self,
+        request: BulkUpdateVpcFlowLogViolationTicketsRequest,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+        _host = None
+
+        _collection_formats: Dict[str, str] = {}
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[str, Union[str, bytes]] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the body parameter
+        if request is not None:
+            _body_params = request
+
+        # set the HTTP header `Accept`
+        _header_params["Accept"] = self.api_client.select_header_accept(
+            ["application/json"]
+        )
+
+        # set the HTTP header `Content-Type`
+        if _content_type:
+            _header_params["Content-Type"] = _content_type
+        else:
+            _default_content_type = self.api_client.select_header_content_type(
+                ["application/json"]
+            )
+            if _default_content_type is not None:
+                _header_params["Content-Type"] = _default_content_type
+
+        # authentication setting
+        _auth_settings: List[str] = []
+
+        return self.api_client.param_serialize(
+            method="POST",
+            resource_path="/rpc/vpc_flow_log_violations_service/bulk_update_vpc_flow_log_violation_tickets",
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth,
+        )
+
+    @validate_call
+    def sync_vpc_flow_log_violations(
+        self,
+        request: SyncVpcFlowLogViolationsRequest,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> SyncVpcFlowLogViolationsResponse:
+        """Sync Vpc Flow Log Violations
+
+
+
+        :param request: (required)
+        :type request: SyncVpcFlowLogViolationsRequest
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """
+
+        _param = self._sync_vpc_flow_log_violations_serialize(
+            request=request,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "SyncVpcFlowLogViolationsResponse",
+            "422": "HTTPValidationError",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+    def _sync_vpc_flow_log_violations_serialize(
+        self,
+        request: SyncVpcFlowLogViolationsRequest,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+        _host = None
+
+        _collection_formats: Dict[str, str] = {}
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[str, Union[str, bytes]] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the body parameter
+        if request is not None:
+            _body_params = request
+
+        # set the HTTP header `Accept`
+        _header_params["Accept"] = self.api_client.select_header_accept(
+            ["application/json"]
+        )
+
+        # set the HTTP header `Content-Type`
+        if _content_type:
+            _header_params["Content-Type"] = _content_type
+        else:
+            _default_content_type = self.api_client.select_header_content_type(
+                ["application/json"]
+            )
+            if _default_content_type is not None:
+                _header_params["Content-Type"] = _default_content_type
+
+        # authentication setting
+        _auth_settings: List[str] = []
+
+        return self.api_client.param_serialize(
+            method="POST",
+            resource_path="/rpc/vpc_flow_log_violations_service/sync_vpc_flow_log_violations",
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth,
+        )

Original file line number	Diff line number	Diff line change
`@@ -224,6 +224,10 @@`
`224`	`224`	`NetworkFlowAnalysisRequestsServiceRpcHandler,`
`225`	`225`	`)`
`226`	`226`
	`227`	`+from onelens_backend_client_v2.rpc.vpc_flow_log_violations_service_rpc_handler import (`
	`228`	`+ VpcFlowLogViolationsServiceRpcHandler,`
	`229`	`+)`
	`230`	`+`
`227`	`231`
`228`	`232`	`__all__ = [`
`229`	`233`	`"AzureAdvisorTicketsServiceRpcHandler",`
`@@ -283,4 +287,5 @@`
`283`	`287`	`"AggregatedTicketsServiceRpcHandler",`
`284`	`288`	`"AggregatedPoliciesServiceRpcHandler",`
`285`	`289`	`"NetworkFlowAnalysisRequestsServiceRpcHandler",`
	`290`	`+ "VpcFlowLogViolationsServiceRpcHandler",`
`286`	`291`	`]`