diff options
Diffstat (limited to 'muck/mem.py')
-rw-r--r-- | muck/mem.py | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/muck/mem.py b/muck/mem.py new file mode 100644 index 0000000..437085d --- /dev/null +++ b/muck/mem.py @@ -0,0 +1,56 @@ +# Copyright (C) 2018 Lars Wirzenius +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + + +import muck + + +class MemoryStore: + + def __init__(self): + self._dict = {} + + def __len__(self): + return len(self._dict) + + def as_dict(self): + return self._dict + + def change(self, chg): + funcs = { + 'create': self._create, + 'update': self._update, + 'delete': self._delete, + } + op = chg.get_op() + func = funcs[op] + func(chg) + + def _create(self, chg): + rid = chg.get_id() + if rid in self._dict: + raise muck.Error('Resource {} already exists'.format(rid)) + self._dict[rid] = (chg.get_meta(), chg.get_res()) + + def _update(self, chg): + rid = chg.get_id() + if rid not in self._dict: + raise muck.Error('Resource {} does not exist'.format(rid)) + self._dict[rid] = (chg.get_meta(), chg.get_res()) + + def _delete(self, chg): + rid = chg.get_id() + if rid not in self._dict: + raise muck.Error('Resource {} does not exist'.format(rid)) + del self._dict[rid] |