diff options
Diffstat (limited to 'searx/shared')
-rw-r--r-- | searx/shared/shared_abstract.py | 10 | ||||
-rw-r--r-- | searx/shared/shared_simple.py | 9 | ||||
-rw-r--r-- | searx/shared/shared_uwsgi.py | 9 |
3 files changed, 16 insertions, 12 deletions
diff --git a/searx/shared/shared_abstract.py b/searx/shared/shared_abstract.py index b4b15bea..af4be30a 100644 --- a/searx/shared/shared_abstract.py +++ b/searx/shared/shared_abstract.py @@ -1,20 +1,22 @@ # SPDX-License-Identifier: AGPL-3.0-or-later +# pyright: strict from abc import ABC, abstractmethod +from typing import Optional class SharedDict(ABC): @abstractmethod - def get_int(self, key): + def get_int(self, key: str) -> Optional[int]: pass @abstractmethod - def set_int(self, key, value): + def set_int(self, key: str, value: int): pass @abstractmethod - def get_str(self, key): + def get_str(self, key: str) -> Optional[str]: pass @abstractmethod - def set_str(self, key, value): + def set_str(self, key: str, value: str): pass diff --git a/searx/shared/shared_simple.py b/searx/shared/shared_simple.py index 0bf13a2a..2b9d4c2d 100644 --- a/searx/shared/shared_simple.py +++ b/searx/shared/shared_simple.py @@ -1,6 +1,7 @@ # SPDX-License-Identifier: AGPL-3.0-or-later import threading +from typing import Optional from . import shared_abstract @@ -12,16 +13,16 @@ class SimpleSharedDict(shared_abstract.SharedDict): def __init__(self): self.d = {} - def get_int(self, key): + def get_int(self, key: str) -> Optional[int]: return self.d.get(key, None) - def set_int(self, key, value): + def set_int(self, key: str, value: int): self.d[key] = value - def get_str(self, key): + def get_str(self, key: str) -> Optional[str]: return self.d.get(key, None) - def set_str(self, key, value): + def set_str(self, key: str, value: str): self.d[key] = value diff --git a/searx/shared/shared_uwsgi.py b/searx/shared/shared_uwsgi.py index 592e24a4..4a6b0a15 100644 --- a/searx/shared/shared_uwsgi.py +++ b/searx/shared/shared_uwsgi.py @@ -1,6 +1,7 @@ # SPDX-License-Identifier: AGPL-3.0-or-later import time +from typing import Optional import uwsgi # pylint: disable=E0401 from . import shared_abstract @@ -9,25 +10,25 @@ _last_signal = 10 class UwsgiCacheSharedDict(shared_abstract.SharedDict): - def get_int(self, key): + def get_int(self, key: str) -> Optional[int]: value = uwsgi.cache_get(key) if value is None: return value else: return int.from_bytes(value, 'big') - def set_int(self, key, value): + def set_int(self, key: str, value: int): b = value.to_bytes(4, 'big') uwsgi.cache_update(key, b) - def get_str(self, key): + def get_str(self, key: str) -> Optional[str]: value = uwsgi.cache_get(key) if value is None: return value else: return value.decode('utf-8') - def set_str(self, key, value): + def set_str(self, key: str, value: str): b = value.encode('utf-8') uwsgi.cache_update(key, b) |