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

refactoring

Browse Source
This commit is contained in:
Maxim Devaev 2021-07-30 06:53:09 +03:00
parent 1aaea37f54
commit 7d7f4965de
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
kvmd/apps/kvmd/http.py
View File

@ -31,8 +31,12 @@ from typing import Dict
from typing import Callable from typing import Callable
from typing import Optional from typing import Optional
import aiohttp from aiohttp.web import BaseRequest
import aiohttp.web from aiohttp.web import Request
from aiohttp.web import Response
from aiohttp.web import StreamResponse
from aiohttp.web import Application
from aiohttp.web import run_app
try: try:
from aiohttp.web import AccessLogger # type: ignore from aiohttp.web import AccessLogger # type: ignore
@ -138,9 +142,9 @@ def make_json_response(
status: int=200, status: int=200,
set_cookies: Optional[Dict[str, str]]=None, set_cookies: Optional[Dict[str, str]]=None,
wrap_result: bool=True, wrap_result: bool=True,
) -> aiohttp.web.Response: ) -> Response:
response = aiohttp.web.Response( response = Response(
text=json.dumps(({ text=json.dumps(({
"ok": (status == 200), "ok": (status == 200),
"result": (result or {}), "result": (result or {}),
@ -154,7 +158,7 @@ def make_json_response(
return response return response
def make_json_exception(err: Exception, status: Optional[int]=None) -> aiohttp.web.Response: def make_json_exception(err: Exception, status: Optional[int]=None) -> Response:
name = type(err).__name__ name = type(err).__name__
msg = str(err) msg = str(err)
if isinstance(err, HttpError): if isinstance(err, HttpError):
@ -168,13 +172,13 @@ def make_json_exception(err: Exception, status: Optional[int]=None) -> aiohttp.w
}, status=status) }, status=status)
async def start_streaming(request: aiohttp.web.Request, content_type: str) -> aiohttp.web.StreamResponse: async def start_streaming(request: Request, content_type: str) -> StreamResponse:
response = aiohttp.web.StreamResponse(status=200, reason="OK", headers={"Content-Type": content_type}) response = StreamResponse(status=200, reason="OK", headers={"Content-Type": content_type})
await response.prepare(request) await response.prepare(request)
return response return response
async def stream_json(response: aiohttp.web.StreamResponse, result: Dict) -> None: async def stream_json(response: StreamResponse, result: Dict) -> None:
await response.write(json.dumps(result).encode("utf-8") + b"\r\n") await response.write(json.dumps(result).encode("utf-8") + b"\r\n")
@ -182,14 +186,14 @@ async def stream_json(response: aiohttp.web.StreamResponse, result: Dict) -> Non
_REQUEST_AUTH_INFO = "_kvmd_auth_info" _REQUEST_AUTH_INFO = "_kvmd_auth_info"
def _format_P(request: aiohttp.web.BaseRequest, *_, **__) -> str: # type: ignore # pylint: disable=invalid-name def _format_P(request: BaseRequest, *_, **__) -> str: # type: ignore # pylint: disable=invalid-name
return (getattr(request, _REQUEST_AUTH_INFO, None) or "-") return (getattr(request, _REQUEST_AUTH_INFO, None) or "-")
AccessLogger._format_P = staticmethod(_format_P) # type: ignore # pylint: disable=protected-access AccessLogger._format_P = staticmethod(_format_P) # type: ignore # pylint: disable=protected-access
def set_request_auth_info(request: aiohttp.web.BaseRequest, info: str) -> None: def set_request_auth_info(request: BaseRequest, info: str) -> None:
setattr(request, _REQUEST_AUTH_INFO, info) setattr(request, _REQUEST_AUTH_INFO, info)
@ -218,7 +222,7 @@ class HttpServer:
else: else:
socket_kwargs = {"host": host, "port": port} socket_kwargs = {"host": host, "port": port}
aiohttp.web.run_app( run_app(
app=self._make_app(), app=self._make_app(),
shutdown_timeout=1, shutdown_timeout=1,
access_log_format=access_log_format, access_log_format=access_log_format,
@ -226,7 +230,7 @@ class HttpServer:
**socket_kwargs, **socket_kwargs,
) )
async def _make_app(self) -> aiohttp.web.Application: async def _make_app(self) -> Application:
raise NotImplementedError raise NotImplementedError
def __run_app_print(self, text: str) -> None: def __run_app_print(self, text: str) -> None: