X-Git-Url: http://www.average.org/gitweb/?p=loctrkd.git;a=blobdiff_plain;f=loctrkd%2Fcommon.py;h=162fe0db07abbbd5a14e0cfbc429d93d193eb9d5;hp=941a93e008dbf793cad604d0df1fd86a81809b63;hb=2cf0fd9d215dda17eae4261ab7967367f6aa0028;hpb=6eff65f7b03bc66a479df0fd694250e1e0b7c5ae diff --git a/loctrkd/common.py b/loctrkd/common.py index 941a93e..162fe0d 100644 --- a/loctrkd/common.py +++ b/loctrkd/common.py @@ -11,7 +11,7 @@ from sys import argv, stderr, stdout from typing import Any, cast, Dict, List, Optional, Tuple, Union from types import SimpleNamespace -from .protomodule import ProtoModule +from .protomodule import ProtoClass, ProtoModule CONF = "/etc/loctrkd.conf" pmods: List[ProtoModule] = [] @@ -71,6 +71,22 @@ def pmod_for_proto(proto: str) -> Optional[ProtoModule]: return None +def pmod_by_name(pmodname: str) -> Optional[ProtoModule]: + for pmod in pmods: + if pmod.__name__.split(".")[-1] == pmodname: + return pmod + return None + + +def make_response( + pmodname: str, cmd: str, imei: str, **kwargs: Any +) -> Optional[ProtoClass.Out]: + pmod = pmod_by_name(pmodname) + if pmod is None: + return None + return pmod.make_response(cmd, imei, **kwargs) + + def parse_message(proto: str, packet: bytes, is_incoming: bool = True) -> Any: pmod = pmod_for_proto(proto) return pmod.parse_message(packet, is_incoming) if pmod else None