mirror of
https://github.com/searxng/searxng
synced 2024-01-01 19:24:07 +01:00
54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
import threading
|
|
from typing import Optional
|
|
|
|
from . import shared_abstract
|
|
|
|
|
|
class SimpleSharedDict(shared_abstract.SharedDict):
|
|
|
|
__slots__ = ('d',)
|
|
|
|
def __init__(self):
|
|
self.d = {}
|
|
|
|
def get_int(self, key: str) -> Optional[int]:
|
|
return self.d.get(key, None)
|
|
|
|
def set_int(self, key: str, value: int, expire: Optional[int] = None):
|
|
self.d[key] = value
|
|
if expire:
|
|
self._expire(key, expire)
|
|
|
|
def get_str(self, key: str) -> Optional[str]:
|
|
return self.d.get(key, None)
|
|
|
|
def set_str(self, key: str, value: str, expire: Optional[int] = None):
|
|
self.d[key] = value
|
|
if expire:
|
|
self._expire(key, expire)
|
|
|
|
def _expire(self, key: str, expire: int):
|
|
t = threading.Timer(expire, lambda k, d: d.pop(k), args=[key, self.d])
|
|
t.daemon = True
|
|
t.start()
|
|
|
|
|
|
def run_locked(func, *args):
|
|
# SimpleSharedDict is not actually shared, so no locking needed
|
|
return func(*args)
|
|
|
|
|
|
def schedule(delay, func, *args):
|
|
def call_later():
|
|
t = threading.Timer(delay, wrapper)
|
|
t.daemon = True
|
|
t.start()
|
|
|
|
def wrapper():
|
|
call_later()
|
|
func(*args)
|
|
|
|
call_later()
|
|
return True
|