Break27/kvmd
1
0
Fork 0
mirror of https://github.com/Break27/kvmd.git synced 2026-02-06 18:36:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: action parsing

Browse Source
This commit is contained in:
Break27 2023-10-09 19:56:25 +08:00
parent 1da10b3996
commit 6966050e5b
1 changed files with 13 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
kvmd/apps/kvmd/remote.py
View File

@ -19,7 +19,11 @@ class RemoteHost:
self.name = name
self.address = address
self.encoding = encoding
self.actions = kwargs
self.actions = {
k[3:].upper() : v for k, v in kwargs.items()
if k.startswith('on_')
}
self.online = False
self.last_seen = 0
@ -29,19 +33,9 @@ class RemoteHost:
"name": self.name,
"online": self.online,
"last_seen": self.last_seen,
"actions": self.parse_action(),
"actions": [ x for x in self.actions ],
}
def parse_action(self, action=None) -> list[str] | str:
if not action:
return [
key[3:].upper() for key, value in self.actions.items()
if key.startswith('on_') and value
]
command = self.actions.get("on_%s" % action.lower())
if not command: raise NotImplementedError()
return command
async def ping(self) -> bool:
process = await asyncio.create_subprocess_shell(
f"ping -i 0.5 -c 2 -w 2 {self.address}",
@ -69,7 +63,7 @@ class RemoteControl:
@staticmethod
def from_dict(hosts: list[dict], **kwargs):
roll = [RemoteHost(**host) for host in hosts]
roll = [ RemoteHost(**host) for host in hosts ]
return RemoteControl(roll, **kwargs)
async def get_state(self) -> list[dict]:
@ -81,7 +75,10 @@ class RemoteControl:
if hostname not in self.__hosts:
raise KeyError(hostname)
host = self.__hosts.get(hostname)
command = host.parse_action(action)
if action not in host.actions:
raise NotImplementedError(action)
command = host.actions.get(action)
if command.startswith("ssh"):
head, tail = command.split(' ', 1)
@ -109,6 +106,6 @@ class RemoteControl:
async def poll_state(self) -> AsyncGenerator[list, None]:
while True:
state = await self.update()
if state: yield state
status = await self.update()
if status: yield status
await self.__notifier.wait(self.__timeout)