Skip to content

Commit f193d29

Browse files
committed
Readd websession and fix typing
1 parent d1d8096 commit f193d29

File tree

4 files changed

+11
-10
lines changed

4 files changed

+11
-10
lines changed

plugwise/__init__.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -43,19 +43,20 @@ def __init__(
4343
self,
4444
host: str,
4545
password: str,
46+
websession: aiohttp.ClientSession,
4647
username: str = DEFAULT_USERNAME,
4748
port: int = DEFAULT_PORT,
4849
timeout: float = DEFAULT_TIMEOUT,
49-
websession: aiohttp.ClientSession | None = None,
50+
5051
) -> None:
5152
"""Set the constructor for this class."""
5253
super().__init__(
5354
host,
5455
password,
56+
websession,
5557
username,
5658
port,
5759
timeout,
58-
websession,
5960
)
6061

6162
self._smile_api: SmileAPI | SmileLegacyAPI
@@ -64,7 +65,7 @@ def __init__(
6465
self._user = username
6566
self._port = port
6667
self._timeout = timeout
67-
# self._websession = websession
68+
self._websession = websession
6869

6970
self._cooling_present = False
7071
self._elga = False
@@ -141,10 +142,10 @@ async def connect(self) -> bool:
141142
self.smile_name,
142143
self.smile_type,
143144
self.smile_version,
145+
self._websession,
144146
self._user,
145147
self._port,
146148
self._timeout,
147-
# self._websession,
148149
)
149150
if self.smile_legacy:
150151
self._smile_api = SmileLegacyAPI(
@@ -166,10 +167,10 @@ async def connect(self) -> bool:
166167
self.smile_type,
167168
self.smile_version,
168169
self.smile_zigbee_mac_address,
170+
self._websession,
169171
self._user,
170172
self._port,
171173
self._timeout,
172-
# self._websession,
173174
)
174175

175176
# Update all endpoints on first connect

plugwise/helper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,10 +104,10 @@ def __init__(
104104
self,
105105
host: str,
106106
password: str,
107+
websession: ClientSession | None,
107108
username: str,
108109
port: int,
109110
timeout: float,
110-
websession: ClientSession | None,
111111
) -> None:
112112
"""Set the constructor for this class."""
113113
if not websession:

plugwise/legacy/smile.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,19 +57,19 @@ def __init__(
5757
smile_type: str,
5858
smile_version: tuple[str, semver.version.Version],
5959
smile_zigbee_mac_address: str | None,
60+
websession: aiohttp.ClientSession,
6061
username: str = DEFAULT_USERNAME,
6162
port: int = DEFAULT_PORT,
6263
timeout: float = DEFAULT_TIMEOUT,
63-
websession: aiohttp.ClientSession | None = None,
6464
) -> None:
6565
"""Set the constructor for this class."""
6666
super().__init__(
6767
host,
6868
password,
69+
websession,
6970
username,
7071
port,
7172
timeout,
72-
websession,
7373
)
7474
SmileLegacyData.__init__(self)
7575

plugwise/smile.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,20 +63,20 @@ def __init__(
6363
smile_name: str,
6464
smile_type: str,
6565
smile_version: tuple[str, semver.version.Version],
66+
websession: aiohttp.ClientSession,
6667
username: str = DEFAULT_USERNAME,
6768
port: int = DEFAULT_PORT,
6869
timeout: float = DEFAULT_TIMEOUT,
69-
websession: aiohttp.ClientSession | None = None,
7070

7171
) -> None:
7272
"""Set the constructor for this class."""
7373
super().__init__(
7474
host,
7575
password,
76+
websession,
7677
username,
7778
port,
7879
timeout,
79-
websession,
8080
)
8181
SmileData.__init__(self)
8282

0 commit comments

Comments
 (0)