Skip to content

Commit aae4fa4

Browse files
tmunzer-AIDEtmunzer-AIDE
committed
fix typo
1 parent c5b2c1d commit aae4fa4

16 files changed

Lines changed: 134 additions & 133 deletions

File tree

‎src/mistapi/device_utils/__tools/__ws_wrapper.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -147,13 +147,13 @@ class WebSocketWrapper:
147147

148148
def __init__(
149149
self,
150-
apissession: APISession,
150+
apisession: APISession,
151151
util_response: UtilResponse,
152152
timeout: int = 10,
153153
max_duration: int = 60,
154154
on_message: Callable[[dict], None] | None = None,
155155
) -> None:
156-
self.apissession = apissession
156+
self.apisession = apisession
157157
self.util_response = util_response
158158
self.timers = {
159159
Timer.TIMEOUT.value: {

‎src/mistapi/device_utils/__tools/arp.py‎

Lines changed: 13 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
"""
1212

1313
from collections.abc import Callable
14+
1415
from mistapi import APISession as _APISession
1516
from mistapi.__logger import logger as LOGGER
1617
from mistapi.api.v1.sites import devices
@@ -20,7 +21,7 @@
2021

2122

2223
def retrieve_ap_arp_table(
23-
apissession: _APISession,
24+
apisession: _APISession,
2425
site_id: str,
2526
device_id: str,
2627
node: Node | None = None,
@@ -64,19 +65,19 @@ def retrieve_ap_arp_table(
6465
body["node"] = node.value
6566
util_response = UtilResponse()
6667
return WebSocketWrapper(
67-
apissession, util_response, timeout=timeout, on_message=on_message
68+
apisession, util_response, timeout=timeout, on_message=on_message
6869
).start_with_trigger(
6970
trigger_fn=lambda: devices.arpFromDevice(
70-
apissession, site_id=site_id, device_id=device_id, body=body
71+
apisession, site_id=site_id, device_id=device_id, body=body
7172
),
7273
ws_factory_fn=lambda _trigger: DeviceCmdEvents(
73-
apissession, site_id=site_id, device_ids=[device_id]
74+
apisession, site_id=site_id, device_ids=[device_id]
7475
),
7576
)
7677

7778

7879
def retrieve_ssr_arp_table(
79-
apissession: _APISession,
80+
apisession: _APISession,
8081
site_id: str,
8182
device_id: str,
8283
node: Node | None = None,
@@ -120,19 +121,19 @@ def retrieve_ssr_arp_table(
120121
body["node"] = node.value
121122
util_response = UtilResponse()
122123
return WebSocketWrapper(
123-
apissession, util_response, timeout=timeout, on_message=on_message
124+
apisession, util_response, timeout=timeout, on_message=on_message
124125
).start_with_trigger(
125126
trigger_fn=lambda: devices.arpFromDevice(
126-
apissession, site_id=site_id, device_id=device_id, body=body
127+
apisession, site_id=site_id, device_id=device_id, body=body
127128
),
128129
ws_factory_fn=lambda _trigger: DeviceCmdEvents(
129-
apissession, site_id=site_id, device_ids=[device_id]
130+
apisession, site_id=site_id, device_ids=[device_id]
130131
),
131132
)
132133

133134

134135
def retrieve_junos_arp_table(
135-
apissession: _APISession,
136+
apisession: _APISession,
136137
site_id: str,
137138
device_id: str,
138139
ip: str | None = None,
@@ -190,12 +191,12 @@ def retrieve_junos_arp_table(
190191
body["port_id"] = port_id
191192
util_response = UtilResponse()
192193
return WebSocketWrapper(
193-
apissession, util_response, timeout=timeout, on_message=on_message
194+
apisession, util_response, timeout=timeout, on_message=on_message
194195
).start_with_trigger(
195196
trigger_fn=lambda: devices.showSiteDeviceArpTable(
196-
apissession, site_id=site_id, device_id=device_id, body=body
197+
apisession, site_id=site_id, device_id=device_id, body=body
197198
),
198199
ws_factory_fn=lambda _trigger: DeviceCmdEvents(
199-
apissession, site_id=site_id, device_ids=[device_id]
200+
apisession, site_id=site_id, device_ids=[device_id]
200201
),
201202
)

‎src/mistapi/device_utils/__tools/bgp.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121

2222
def summary(
23-
apissession: _APISession,
23+
apisession: _APISession,
2424
site_id: str,
2525
device_id: str,
2626
timeout=5,
@@ -57,12 +57,12 @@ def summary(
5757
body: dict[str, str | list | int] = {"protocol": "bgp"}
5858
util_response = UtilResponse()
5959
return WebSocketWrapper(
60-
apissession, util_response, timeout=timeout, on_message=on_message
60+
apisession, util_response, timeout=timeout, on_message=on_message
6161
).start_with_trigger(
6262
trigger_fn=lambda: devices.showSiteDeviceBgpSummary(
63-
apissession, site_id=site_id, device_id=device_id, body=body
63+
apisession, site_id=site_id, device_id=device_id, body=body
6464
),
6565
ws_factory_fn=lambda _trigger: DeviceCmdEvents(
66-
apissession, site_id=site_id, device_ids=[device_id]
66+
apisession, site_id=site_id, device_ids=[device_id]
6767
),
6868
)

‎src/mistapi/device_utils/__tools/bpdu.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818

1919
def clear_error(
20-
apissession: _APISession,
20+
apisession: _APISession,
2121
site_id: str,
2222
device_id: str,
2323
port_ids: list[str],
@@ -51,8 +51,8 @@ def clear_error(
5151
)
5252
body: dict[str, str | list | int] = {"ports": port_ids}
5353
util_response = UtilResponse()
54-
return WebSocketWrapper(apissession, util_response).start_with_trigger(
54+
return WebSocketWrapper(apisession, util_response).start_with_trigger(
5555
trigger_fn=lambda: devices.clearBpduErrorsFromPortsOnSwitch(
56-
apissession, site_id=site_id, device_id=device_id, body=body
56+
apisession, site_id=site_id, device_id=device_id, body=body
5757
),
5858
)

‎src/mistapi/device_utils/__tools/dhcp.py‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222

2323
def release_dhcp_leases(
24-
apissession: _APISession,
24+
apisession: _APISession,
2525
site_id: str,
2626
device_id: str,
2727
macs: list[str] | None = None,
@@ -96,19 +96,19 @@ def release_dhcp_leases(
9696
body["port_id"] = port_id
9797
util_response = UtilResponse()
9898
return WebSocketWrapper(
99-
apissession, util_response, timeout=timeout, on_message=on_message
99+
apisession, util_response, timeout=timeout, on_message=on_message
100100
).start_with_trigger(
101101
trigger_fn=lambda: devices.releaseSiteDeviceDhcpLease(
102-
apissession, site_id=site_id, device_id=device_id, body=body
102+
apisession, site_id=site_id, device_id=device_id, body=body
103103
),
104104
ws_factory_fn=lambda _trigger: DeviceCmdEvents(
105-
apissession, site_id=site_id, device_ids=[device_id]
105+
apisession, site_id=site_id, device_ids=[device_id]
106106
),
107107
)
108108

109109

110110
def retrieve_dhcp_leases(
111-
apissession: _APISession,
111+
apisession: _APISession,
112112
site_id: str,
113113
device_id: str,
114114
network: str,
@@ -155,12 +155,12 @@ def retrieve_dhcp_leases(
155155
body["node"] = node.value
156156
util_response = UtilResponse()
157157
return WebSocketWrapper(
158-
apissession, util_response, timeout=timeout, on_message=on_message
158+
apisession, util_response, timeout=timeout, on_message=on_message
159159
).start_with_trigger(
160160
trigger_fn=lambda: devices.showSiteDeviceDhcpLeases(
161-
apissession, site_id=site_id, device_id=device_id, body=body
161+
apisession, site_id=site_id, device_id=device_id, body=body
162162
),
163163
ws_factory_fn=lambda _trigger: DeviceCmdEvents(
164-
apissession, site_id=site_id, device_ids=[device_id]
164+
apisession, site_id=site_id, device_ids=[device_id]
165165
),
166166
)

‎src/mistapi/device_utils/__tools/dns.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ class Node(Enum):
2222

2323
## NO DATA
2424
# def test_resolution(
25-
# apissession: _APISession,
25+
# apisession: _APISession,
2626
# site_id: str,
2727
# device_id: str,
2828
# node: Node | None = None,
@@ -64,7 +64,7 @@ class Node(Enum):
6464
# if hostname:
6565
# body["hostname"] = hostname
6666
# trigger = devices.testSiteSsrDnsResolution(
67-
# apissession,
67+
# apisession,
6868
# site_id=site_id,
6969
# device_id=device_id,
7070
# body=body,
@@ -73,9 +73,9 @@ class Node(Enum):
7373
# if trigger.status_code == 200:
7474
# LOGGER.info(trigger.data)
7575
# print(f"SSR DNS resolution command triggered for device {device_id}")
76-
# ws = DeviceCmdEvents(apissession, site_id=site_id, device_ids=[device_id])
76+
# ws = DeviceCmdEvents(apisession, site_id=site_id, device_ids=[device_id])
7777
# util_response = await WebSocketWrapper(
78-
# apissession, util_response, timeout=timeout, on_message=on_message
78+
# apisession, util_response, timeout=timeout, on_message=on_message
7979
# ).start(ws)
8080
# else:
8181
# LOGGER.error(

‎src/mistapi/device_utils/__tools/dot1x.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818

1919
def clear_sessions(
20-
apissession: _APISession,
20+
apisession: _APISession,
2121
site_id: str,
2222
device_id: str,
2323
port_ids: list[str],
@@ -51,8 +51,8 @@ def clear_sessions(
5151
)
5252
body: dict[str, str | list | int] = {"ports": port_ids}
5353
util_response = UtilResponse()
54-
return WebSocketWrapper(apissession, util_response).start_with_trigger(
54+
return WebSocketWrapper(apisession, util_response).start_with_trigger(
5555
trigger_fn=lambda: devices.clearSiteDeviceDot1xSession(
56-
apissession, site_id=site_id, device_id=device_id, body=body
56+
apisession, site_id=site_id, device_id=device_id, body=body
5757
),
5858
)

‎src/mistapi/device_utils/__tools/mac.py‎

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020

2121

2222
def clear_mac_table(
23-
apissession: _APISession,
23+
apisession: _APISession,
2424
site_id: str,
2525
device_id: str,
2626
mac_address: str | None = None,
@@ -34,7 +34,7 @@ def clear_mac_table(
3434
3535
PARAMS
3636
-----------
37-
apissession : mistapi.APISession
37+
apisession : mistapi.APISession
3838
The API session to use for the request.
3939
site_id : str
4040
UUID of the site where the device is located.
@@ -69,15 +69,15 @@ def clear_mac_table(
6969
if vlan_id:
7070
body["vlan_id"] = vlan_id
7171
util_response = UtilResponse()
72-
return WebSocketWrapper(apissession, util_response).start_with_trigger(
72+
return WebSocketWrapper(apisession, util_response).start_with_trigger(
7373
trigger_fn=lambda: devices.clearSiteDeviceMacTable(
74-
apissession, site_id=site_id, device_id=device_id, body=body
74+
apisession, site_id=site_id, device_id=device_id, body=body
7575
),
7676
)
7777

7878

7979
def retrieve_mac_table(
80-
apissession: _APISession,
80+
apisession: _APISession,
8181
site_id: str,
8282
device_id: str,
8383
mac_address: str | None = None,
@@ -93,7 +93,7 @@ def retrieve_mac_table(
9393
9494
PARAMS
9595
-----------
96-
apissession : mistapi.APISession
96+
apisession : mistapi.APISession
9797
The API session to use for the request.
9898
site_id : str
9999
UUID of the site where the device is located.
@@ -134,19 +134,19 @@ def retrieve_mac_table(
134134
body["vlan_id"] = vlan_id
135135
util_response = UtilResponse()
136136
return WebSocketWrapper(
137-
apissession, util_response, timeout=timeout, on_message=on_message
137+
apisession, util_response, timeout=timeout, on_message=on_message
138138
).start_with_trigger(
139139
trigger_fn=lambda: devices.showSiteDeviceMacTable(
140-
apissession, site_id=site_id, device_id=device_id, body=body
140+
apisession, site_id=site_id, device_id=device_id, body=body
141141
),
142142
ws_factory_fn=lambda _trigger: DeviceCmdEvents(
143-
apissession, site_id=site_id, device_ids=[device_id]
143+
apisession, site_id=site_id, device_ids=[device_id]
144144
),
145145
)
146146

147147

148148
def clear_learned_mac(
149-
apissession: _APISession,
149+
apisession: _APISession,
150150
site_id: str,
151151
device_id: str,
152152
port_ids: list[str],
@@ -158,7 +158,7 @@ def clear_learned_mac(
158158
159159
PARAMS
160160
-----------
161-
apissession: mistapi.APISession
161+
apisession: mistapi.APISession
162162
The API session to use for the request.
163163
site_id : str
164164
UUID of the site where the device is located.
@@ -180,8 +180,8 @@ def clear_learned_mac(
180180
)
181181
body: dict[str, str | list | int] = {"ports": port_ids}
182182
util_response = UtilResponse()
183-
return WebSocketWrapper(apissession, util_response).start_with_trigger(
183+
return WebSocketWrapper(apisession, util_response).start_with_trigger(
184184
trigger_fn=lambda: devices.clearAllLearnedMacsFromPortOnSwitch(
185-
apissession, site_id=site_id, device_id=device_id, body=body
185+
apisession, site_id=site_id, device_id=device_id, body=body
186186
),
187187
)

0 commit comments

Comments
 (0)