diff --git a/aioruckus/ruckusapi.py b/aioruckus/ruckusapi.py index 0793d79..523fdba 100644 --- a/aioruckus/ruckusapi.py +++ b/aioruckus/ruckusapi.py @@ -7,7 +7,7 @@ import xmltodict from .exceptions import SchemaError -from .typing.policy import ArcApplication, ArcPolicy, AvpPort, DevicePolicy, Dpsk, Ip4Policy, Ip6Policy, L2Policy, L2Rule, PrecedencePolicy, Role, UrlBlockCategory, UrlFilter +from .typing.policy import ArcApplication, ArcPolicy, ArcPort, DevicePolicy, Dpsk, Ip4Policy, Ip6Policy, L2Policy, L2Rule, PrecedencePolicy, Role, UrlBlockCategory, UrlFilter from .abcsession import AbcSession, ConfigItem from .const import ERROR_POST_BADRESULT, URL_FILTERING_CATEGORIES, SystemStat @@ -197,7 +197,7 @@ async def get_arc_applications(self) -> list[ArcApplication | dict]: except KeyError: return [] - async def get_arc_ports(self) -> list[AvpPort | dict]: + async def get_arc_ports(self) -> list[ArcPort | dict]: """Return a list of Application Recognition & Control User Defined Ports""" try: return await self._get_conf(ConfigItem.AVPPORT_LIST, ["avpport"]) diff --git a/aioruckus/typing/policy.py b/aioruckus/typing/policy.py index 8dd657f..9a12cac 100644 --- a/aioruckus/typing/policy.py +++ b/aioruckus/typing/policy.py @@ -17,7 +17,7 @@ 'id': str, 'name': str, 'protocol': Literal['tcp', 'udp'] , 'src-ip': str, 'src-port': str, 'dst-ip': str, 'dst-port': str, 'netmask': str }, total=False) -AvpPort = TypedDict('AvpPort', { +ArcPort = TypedDict('ArcPort', { 'id': str, 'name': str, 'port': str, 'protocol': Literal['tcp', 'udp'] }, total=False)