GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2025-12-12 17:20:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
Devaev Maxim 2019-04-08 17:03:19 +03:00
parent d2559c26b6
commit dda30309a4
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
kvmd/apps/kvmd/streamer.py
View File

@ -117,11 +117,11 @@ class Streamer: # pylint: disable=too-many-instance-attributes
return dict(self.__params)
async def get_state(self) -> Dict:
self.__ensure_session()
session = self.__ensure_session()
url = "http://%s:%d/state" % (self.__host, self.__port)
state = None
try:
async with self.__http_session.get(url, timeout=self.__timeout) as response: # type: ignore
async with session.get(url, timeout=self.__timeout) as response:
response.raise_for_status()
state = (await response.json())["result"]
except (aiohttp.ClientConnectionError, aiohttp.ServerConnectionError):
@ -158,12 +158,13 @@ class Streamer: # pylint: disable=too-many-instance-attributes
await self.__http_session.close()
self.__http_session = None
def __ensure_session(self) -> None:
def __ensure_session(self) -> aiohttp.ClientSession:
if not self.__http_session:
if self.__unix_path:
self.__http_session = aiohttp.ClientSession(connector=aiohttp.UnixConnector(path=self.__unix_path))
else:
self.__http_session = aiohttp.ClientSession()
return self.__http_session
async def __inner_start(self) -> None:
assert not self.__streamer_task