Commit 065b4078 authored by George Papathanail's avatar George Papathanail
Browse files

fix: align traffic_influence_service with srm_events removal and SRM zone rename

parent 8dcecf2c

src/open_exposure_gateway/application/services/traffic_influence_service.py

+4 −3
Original line number Diff line number Diff line
@@ -24,6 +24,7 @@ from open_exposure_gateway.core.exceptions import ( 
    DownstreamServiceException,

    NotFoundException,

)

from open_exposure_gateway.domain.edge_application_management import SRMOperationCompleted

from open_exposure_gateway.domain.models import (

    CallbackRegistration,

    Operation,
@@ -36,7 +37,7 @@ from open_exposure_gateway.domain.models import ( 
from open_exposure_gateway.domain.models import (

    TrafficInfluenceState as RecordTrafficInfluenceState,

)

from open_exposure_gateway.domain.srm_events import SRMOperationCompleted, SRMOperationStatus

from open_exposure_gateway.domain.quality_on_demand import SRMOperationStatus

from open_exposure_gateway.domain.traffic_influence import Subject

from open_exposure_gateway.ports.database.callbacks import CallbackRegistrationRepository

from open_exposure_gateway.ports.database.operations import OperationRepository
@@ -101,13 +102,13 @@ class TrafficInfluenceService: 
        if request.edgeCloudZoneId is not None:

            return str(request.edgeCloudZoneId)

        if request.edgeCloudRegion is not None:

            zones = await self.srm_client.get_resource_zones(

            zones = await self.srm_client.get_zones(

                region=request.edgeCloudRegion,

                status=None,

                x_correlator=x_correlator,

            )

            if zones:

                return zones[0].resource_zone_id

                return zones[0].id

        return None



    async def _replay_create(self, operation: Operation) -> TrafficInfluenceResponse: