GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2025-12-13 01:30:31 +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) return dict(self.__params)
async def get_state(self) -> Dict: async def get_state(self) -> Dict:
self.__ensure_session() session = self.__ensure_session()
url = "http://%s:%d/state" % (self.__host, self.__port) url = "http://%s:%d/state" % (self.__host, self.__port)
state = None state = None
try: 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() response.raise_for_status()
state = (await response.json())["result"] state = (await response.json())["result"]
except (aiohttp.ClientConnectionError, aiohttp.ServerConnectionError): except (aiohttp.ClientConnectionError, aiohttp.ServerConnectionError):
@ -158,12 +158,13 @@ class Streamer: # pylint: disable=too-many-instance-attributes
await self.__http_session.close() await self.__http_session.close()
self.__http_session = None self.__http_session = None
def __ensure_session(self) -> None: def __ensure_session(self) -> aiohttp.ClientSession:
if not self.__http_session: if not self.__http_session:
if self.__unix_path: if self.__unix_path:
self.__http_session = aiohttp.ClientSession(connector=aiohttp.UnixConnector(path=self.__unix_path)) self.__http_session = aiohttp.ClientSession(connector=aiohttp.UnixConnector(path=self.__unix_path))
else: else:
self.__http_session = aiohttp.ClientSession() self.__http_session = aiohttp.ClientSession()
return self.__http_session
async def __inner_start(self) -> None: async def __inner_start(self) -> None:
assert not self.__streamer_task assert not self.__streamer_task