Commit 11be96a5 authored by Adrian Pino's avatar Adrian Pino
Browse files

Satisfy linters

parent b1008cb4

src/sunrise6g_opensdk/network/clients/open5gcore/client.py

+13 −9
Original line number Diff line number Diff line 
# -*- coding: utf-8 -*-

from pydantic import ValidationError



from sunrise6g_opensdk import logger

from sunrise6g_opensdk.network.core.network_interface import (

    NetworkManagementInterface,

    build_flows

    build_flows,

)



from ...core import schemas



log = logger.get_logger('Open5GCore') # Usage of brand name

log = logger.get_logger("Open5GCore")  # Usage of brand name



qos_support_map = {

    "qos-e": 1,  # ToDo
@@ -16,6 +18,7 @@ qos_support_map = { 
    "qos-l": 9,  # ToDo not yet available in Nokia RAN

}





class NetworkManager(NetworkManagementInterface):

    def __init__(self, base_url: str, scs_as_id: str):

        if not base_url:
@@ -25,11 +28,12 @@ class NetworkManager(NetworkManagementInterface): 


        self.base_url = base_url

        self.scs_as_id = scs_as_id



    def core_specific_qod_validation(self, session_info: schemas.CreateSession):

        qos_key = session_info.qosProfile.root.strip().lower()



        if qos_key not in qos_support_map:

            supported = ', '.join(qos_support_map.keys())

            supported = ", ".join(qos_support_map.keys())

            raise ValidationError(

                f"Unsupported QoS profile '{session_info.qosProfile.root}'. "

                f"Supported profiles for Open5GCore are: {supported}"
@@ -42,4 +46,4 @@ class NetworkManager(NetworkManagementInterface): 
    ) -> None:

        flow_id = qos_support_map[session_info.qosProfile.root]

        subscription.flowInfo = build_flows(flow_id, session_info)

        subscription.ueIpv4Addr = '192.168.6.1' #ToDo
 
 No newline at end of file
 
        subscription.ueIpv4Addr = "192.168.6.1"  # ToDo

tests/network/test_cases.py

+1 −1
Original line number Diff line number Diff line
@@ -20,5 +20,5 @@ test_cases = [ 
            "base_url": "http://192.168.11.80:8080",

            "scs_as_id": "0001",

        }

    }

    },

]