hemna
/
aprsd
mirror of https://github.com/craigerl/aprsd.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Added type setting on pluging.py for mypy

This commit is contained in:
Hemna 2024-03-20 09:28:01 -04:00
parent d80277c9d8
commit 9c77ca26be
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
aprsd/plugin.py
View File

@ -65,7 +65,7 @@ class APRSDPluginBase(metaclass=abc.ABCMeta):
self.threads = self.create_threads() or [] self.threads = self.create_threads() or []
self.start_threads() self.start_threads()
def start_threads(self): def start_threads(self) -> None:
if self.enabled and self.threads: if self.enabled and self.threads:
if not isinstance(self.threads, list): if not isinstance(self.threads, list):
self.threads = [self.threads] self.threads = [self.threads]
@ -90,10 +90,10 @@ class APRSDPluginBase(metaclass=abc.ABCMeta):
) )
@property @property
def message_count(self): def message_count(self) -> int:
return self.message_counter return self.message_counter
def help(self): def help(self) -> str:
return "Help!" return "Help!"
@abc.abstractmethod @abc.abstractmethod
@ -118,11 +118,11 @@ class APRSDPluginBase(metaclass=abc.ABCMeta):
thread.stop() thread.stop()
@abc.abstractmethod @abc.abstractmethod
def filter(self, packet: packets.core.Packet): def filter(self, packet: type[packets.core.Packet]) -> str | packets.core.MessagePacket:
pass pass
@abc.abstractmethod @abc.abstractmethod
def process(self, packet: packets.core.Packet): def process(self, packet: type[packets.core.Packet]):
"""This is called when the filter passes.""" """This is called when the filter passes."""
@ -154,7 +154,7 @@ class APRSDWatchListPluginBase(APRSDPluginBase, metaclass=abc.ABCMeta):
LOG.warning("Watch list enabled, but no callsigns set.") LOG.warning("Watch list enabled, but no callsigns set.")
@hookimpl @hookimpl
def filter(self, packet: packets.core.Packet): def filter(self, packet: type[packets.core.Packet]) -> str | packets.core.MessagePacket:
result = packets.NULL_MESSAGE result = packets.NULL_MESSAGE
if self.enabled: if self.enabled:
wl = watch_list.WatchList() wl = watch_list.WatchList()
@ -206,7 +206,7 @@ class APRSDRegexCommandPluginBase(APRSDPluginBase, metaclass=abc.ABCMeta):
self.enabled = True self.enabled = True
@hookimpl @hookimpl
def filter(self, packet: packets.core.MessagePacket): def filter(self, packet: packets.core.MessagePacket) -> str | packets.core.MessagePacket:
LOG.info(f"{self.__class__.__name__} called") LOG.info(f"{self.__class__.__name__} called")
if not self.enabled: if not self.enabled:
result = f"{self.__class__.__name__} isn't enabled" result = f"{self.__class__.__name__} isn't enabled"