9bb8decbcf
Former-commit-id:133dc97f67[formerlya02aeb236c] [formerly9f19e3f712] [formerly06a8b51d6d[formerly9f19e3f712[formerly 64fa9254b946eae7e61bbc3f513b7c3696c4f54f]]] Former-commit-id:06a8b51d6dFormer-commit-id:377dcd10b9[formerly3360eb6c5f] Former-commit-id:8e80217e59
76 lines
1.9 KiB
Python
Executable file
76 lines
1.9 KiB
Python
Executable file
# -*- coding: utf-8 -*-
|
|
"""
|
|
cupoftee.db
|
|
~~~~~~~~~~~
|
|
|
|
A simple object database. As long as the server is not running in
|
|
multiprocess mode that's good enough.
|
|
|
|
:copyright: (c) 2009 by the Werkzeug Team, see AUTHORS for more details.
|
|
:license: BSD, see LICENSE for more details.
|
|
"""
|
|
from __future__ import with_statement
|
|
import gdbm
|
|
from threading import Lock
|
|
from pickle import dumps, loads
|
|
|
|
|
|
class Database(object):
|
|
|
|
def __init__(self, filename):
|
|
self.filename = filename
|
|
self._fs = gdbm.open(filename, 'cf')
|
|
self._local = {}
|
|
self._lock = Lock()
|
|
|
|
def __getitem__(self, key):
|
|
with self._lock:
|
|
return self._load_key(key)
|
|
|
|
def _load_key(self, key):
|
|
if key in self._local:
|
|
return self._local[key]
|
|
rv = loads(self._fs[key])
|
|
self._local[key] = rv
|
|
return rv
|
|
|
|
def __setitem__(self, key, value):
|
|
self._local[key] = value
|
|
|
|
def __delitem__(self, key, value):
|
|
with self._lock:
|
|
self._local.pop(key, None)
|
|
if self._fs.has_key(key):
|
|
del self._fs[key]
|
|
|
|
def __del__(self):
|
|
self.close()
|
|
|
|
def __contains__(self, key):
|
|
with self._lock:
|
|
try:
|
|
self._load_key(key)
|
|
except KeyError:
|
|
pass
|
|
return key in self._local
|
|
|
|
def setdefault(self, key, factory):
|
|
with self._lock:
|
|
try:
|
|
rv = self._load_key(key)
|
|
except KeyError:
|
|
self._local[key] = rv = factory()
|
|
return rv
|
|
|
|
def sync(self):
|
|
with self._lock:
|
|
for key, value in self._local.iteritems():
|
|
self._fs[key] = dumps(value, 2)
|
|
self._fs.sync()
|
|
|
|
def close(self):
|
|
try:
|
|
self.sync()
|
|
self._fs.close()
|
|
except:
|
|
pass
|