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

RESTART_VIDEO command

Browse Source
This commit is contained in:
Devaev Maxim 2018-06-29 22:44:04 +03:00
parent 8fccd50717
commit 607e5a9877
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
kvmd/kvmd/server.py
View File

@ -55,6 +55,8 @@ class Server: # pylint: disable=too-many-instance-attributes
self.__system_tasks: List[asyncio.Task] = [] self.__system_tasks: List[asyncio.Task] = []
self.__restart_video = False
def run(self, host: str, port: int) -> None: def run(self, host: str, port: int) -> None:
self.__keyboard.start() self.__keyboard.start()
@ -122,6 +124,7 @@ class Server: # pylint: disable=too-many-instance-attributes
async def __stream_controller(self) -> None: async def __stream_controller(self) -> None:
prev = 0 prev = 0
shutdown_at = 0.0 shutdown_at = 0.0
while True: while True:
cur = len(self.__sockets) cur = len(self.__sockets)
if prev == 0 and cur > 0: if prev == 0 and cur > 0:
@ -132,6 +135,13 @@ class Server: # pylint: disable=too-many-instance-attributes
elif prev == 0 and cur == 0 and time.time() > shutdown_at: elif prev == 0 and cur == 0 and time.time() > shutdown_at:
if self.__streamer.is_running(): if self.__streamer.is_running():
await self.__streamer.stop() await self.__streamer.stop()
if self.__restart_video:
if self.__streamer.is_running():
await self.__streamer.stop()
await self.__streamer.start()
self.__restart_video = False
prev = cur prev = cur
await asyncio.sleep(0.1) await asyncio.sleep(0.1)
@ -168,6 +178,9 @@ class Server: # pylint: disable=too-many-instance-attributes
if method: if method:
await method() await method()
return None return None
elif command == "RESTART_VIDEO":
self.__restart_video = True
return None
get_logger().warning("Received an incorrect command: %r", command) get_logger().warning("Received an incorrect command: %r", command)
return "ERROR incorrect command" return "ERROR incorrect command"