sascha/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2025-02-15 09:28:16 +00:00
Code Issues Releases Wiki Activity

typing, missing awaits

Browse Source
This commit is contained in:
J. Nick Koston 2023-11-11 18:38:48 -06:00
parent b047038adb
commit 26d0f0ebea
No known key found for this signature in database
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
esphome/dashboard/dashboard.py
View File

@ -342,7 +342,7 @@ class EsphomeCommandWebSocket(tornado.websocket.WebSocketHandler):
return self._proc is not None and self._proc.returncode is None return self._proc is not None and self._proc.returncode is None
@websocket_method("stdin") @websocket_method("stdin")
async def handle_stdin(self, json_message): async def handle_stdin(self, json_message: dict[str, Any]) -> None:
if not self.is_process_active: if not self.is_process_active:
return return
data = json_message["data"] data = json_message["data"]
@ -351,7 +351,7 @@ class EsphomeCommandWebSocket(tornado.websocket.WebSocketHandler):
self._proc.stdin.write(data) self._proc.stdin.write(data)
@tornado.gen.coroutine @tornado.gen.coroutine
def _redirect_stdout(self): def _redirect_stdout(self) -> None:
reg = b"[\n\r]" reg = b"[\n\r]"
while True: while True:
@ -370,7 +370,7 @@ class EsphomeCommandWebSocket(tornado.websocket.WebSocketHandler):
_LOGGER.debug("> stdout: %s", data) _LOGGER.debug("> stdout: %s", data)
self.write_message({"event": "line", "data": data}) self.write_message({"event": "line", "data": data})
def _stdout_thread(self): def _stdout_thread(self) -> None:
if not self._use_popen: if not self._use_popen:
return return
while True: while True:
@ -383,13 +383,13 @@ class EsphomeCommandWebSocket(tornado.websocket.WebSocketHandler):
self._proc.wait(1.0) self._proc.wait(1.0)
self._queue.put_nowait(None) self._queue.put_nowait(None)
def _proc_on_exit(self, returncode): def _proc_on_exit(self, returncode: int) -> None:
if not self._is_closed: if not self._is_closed:
# Check if the proc was not forcibly closed # Check if the proc was not forcibly closed
_LOGGER.info("Process exited with return code %s", returncode) _LOGGER.info("Process exited with return code %s", returncode)
self.write_message({"event": "exit", "code": returncode}) self.write_message({"event": "exit", "code": returncode})
def on_close(self): def on_close(self) -> None:
# Check if proc exists (if 'start' has been run) # Check if proc exists (if 'start' has been run)
if self.is_process_active: if self.is_process_active:
_LOGGER.debug("Terminating process") _LOGGER.debug("Terminating process")
@ -466,13 +466,13 @@ class EsphomeRenameHandler(EsphomeCommandWebSocket):
class EsphomeUploadHandler(EsphomePortCommandWebSocket): class EsphomeUploadHandler(EsphomePortCommandWebSocket):
async def build_command(self, json_message: dict[str, Any]) -> list[str]: async def build_command(self, json_message: dict[str, Any]) -> list[str]:
"""Build the command to run.""" """Build the command to run."""
return self.run_command(["upload"], json_message) return await self.run_command(["upload"], json_message)
class EsphomeRunHandler(EsphomePortCommandWebSocket): class EsphomeRunHandler(EsphomePortCommandWebSocket):
async def build_command(self, json_message: dict[str, Any]) -> list[str]: async def build_command(self, json_message: dict[str, Any]) -> list[str]:
"""Build the command to run.""" """Build the command to run."""
return self.run_command(["run"], json_message) return await self.run_command(["run"], json_message)
class EsphomeCompileHandler(EsphomeCommandWebSocket): class EsphomeCompileHandler(EsphomeCommandWebSocket):