You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ORPA-pyOpenRPA/WPy32-3720/python-3.7.2/Lib/site-packages/partd/dict.py

67 lines
1.7 KiB

from .core import Interface
from threading import Lock
class Dict(Interface):
def __init__(self):
self.lock = Lock()
self.data = dict()
Interface.__init__(self)
def __getstate__(self):
return {'data': self.data}
def __setstate__(self, state):
Interface.__setstate__(self, state)
Dict.__init__(self)
self.data = state['data']
def append(self, data, lock=True, **kwargs):
if lock: self.lock.acquire()
try:
for k, v in data.items():
if k not in self.data:
self.data[k] = []
self.data[k].append(v)
finally:
if lock: self.lock.release()
def _get(self, keys, lock=True, **kwargs):
assert isinstance(keys, (list, tuple, set))
if lock:
self.lock.acquire()
try:
result = [b''.join(self.data.get(key, [])) for key in keys]
finally:
if lock:
self.lock.release()
return result
def _iset(self, key, value, lock=True):
""" Idempotent set """
if lock:
self.lock.acquire()
try:
self.data[key] = [value]
finally:
if lock:
self.lock.release()
def _delete(self, keys, lock=True):
if lock:
self.lock.acquire()
try:
for key in keys:
if key in self.data:
del self.data[key]
finally:
if lock:
self.lock.release()
def drop(self):
self._iset_seen.clear()
self.data.clear()
def __exit__(self, *args):
self.drop()