2020-01-25 14:20:25 +01:00
|
|
|
"""Store interface
|
|
|
|
|
|
|
|
|
|
Allows arbitrary classes to offer a simple get/set
|
|
|
|
|
store interface by deriving from the Store class in
|
|
|
|
|
this module
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
class Store(object):
|
|
|
|
|
"""Interface for storing and retrieving simple values"""
|
|
|
|
|
def __init__(self):
|
2020-02-08 13:45:52 +01:00
|
|
|
super(Store, self).__init__()
|
2020-01-25 14:20:25 +01:00
|
|
|
self._data = {}
|
|
|
|
|
|
|
|
|
|
def set(self, key, value):
|
|
|
|
|
"""Set 'key' to 'value', overwriting 'key' if it exists already"""
|
|
|
|
|
self._data[key] = { 'value': value, 'used': False }
|
|
|
|
|
|
|
|
|
|
def unused_keys(self):
|
|
|
|
|
return [ key for key, value in self._data.items() if value['used'] == False ]
|
|
|
|
|
|
|
|
|
|
def get(self, key, default=None):
|
|
|
|
|
"""Return the current value of 'key', or 'default' if 'key' is not set"""
|
|
|
|
|
if key in self._data:
|
|
|
|
|
self._data[key]['used'] = True
|
|
|
|
|
return self._data.get(key, { 'value': default })['value']
|
|
|
|
|
|
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|
