Web   ·   Wiki   ·   Activities   ·   Blog   ·   Lists   ·   Chat   ·   Meeting   ·   Bugs   ·   Git   ·   Translate   ·   Archive   ·   People   ·   Donate
summaryrefslogtreecommitdiffstats
path: root/sharedstate.git/sharedstate/shareddict.py
blob: 1a51a12f85643326087c865ee5cc58eba36d063e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# shareddict.py, classes to aid activities in sharing a state
# Reinier Heeres, reinier@heeres.eu
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
#
# Change log:
#   2007-06-21: rwh, first version

import logging
_logger = logging.getLogger('sharinghelper')

from sharedobject import SharedObject

class SharedDict(SharedObject):
    """Shared dictionary object, generates key-based difference objects"""

    def __init__(self, name, helper, opt={}):
        SharedObject.__init__(self, name, helper, opt=opt)
        self._value = {}

    def __getitem__(self, key):
#        _logger.debug('SharedDict.__getitem__(%s)', key)
        return self.get_key(key)

    def __setitem__(self, key, val):
#        _logger.debug('SharedDict.__setitem__(%s, %s)', key, val)
        return self.set_key(key, val)

    def __delitem__(self, key):
        self.delete_key(key)

    def get_key(self, key, val):
        return self._value[key]

    def set_key(self, key, val):
        if key in self._value:
            d = {'change': {key: val}}
        else:
            d = {'add': {key: val}}
        self._value[key] = val
        self.changed(d, True)

    def delete_key(self, key):
        if key in self._value:
            del self._value[key]
            d = {'remove': key}
            self.changed(d, True)

    def _compatible_diffs(self, diffa, diffb):
        for key in diffa:
            if key in diffb:
                return False
        return True

    def diff(self, cur, old):
        ret = {'remove': [], 'add': {}, 'change': {}}
        if old is None:
            return ret

        for key in set(old.keys()).difference(cur.keys()):
            ret['remove'].append(key)

        for key in set(cur.keys()).difference(old.keys()):
            ret['add'][key] = cur[key]

        for key in set(cur.keys()).intersection(old.keys()):
            if cur[key] is not old[key]:
                ret['change'][key] = cur[key]

        return ret

    def _apply_diff_to(self, obj, diffobj):
        """Apply a diff and return an object that describes the inverse diff"""

        if diffobj is None:
            return (None, None)

        ret = {'add':{}, 'remove':[], 'change':{}}

        if 'remove' in diffobj:
            for key in diffobj['remove']:
                ret['add'][key] = obj[key]
                del self._value[key]

        if 'add' in diffobj:
            for key, val in diffobj['add'].iteritems():
                ret['remove'].append(key)
                obj[key] = val

        if 'change' in diffobj:
            for key, val in diffobj['change'].iteritems():
                ret[key] = self._value[key]
                obj[key] = val

        return (obj, ret)