sjs/vdirsyncer
1
0
Fork 0
mirror of https://github.com/samsonjs/vdirsyncer.git synced 2026-04-27 14:57:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow storages to modify sync behavior

Browse source
This commit is contained in:
Markus Unterwaditzer 2015-04-09 14:15:59 +02:00
parent 13408779cb
commit e138af8d37
2 changed files with 19 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
vdirsyncer/storage/base.py
View file

@ -3,7 +3,7 @@
import contextlib import contextlib
import functools import functools
from .. import exceptions from .. import exceptions, sync
from ..utils import uniq from ..utils import uniq
from ..utils.compat import with_metaclass from ..utils.compat import with_metaclass
from ..utils.vobject import Item # noqa from ..utils.vobject import Item # noqa
@ -48,6 +48,8 @@ class Storage(with_metaclass(StorageMeta)):
fileext = '.txt' fileext = '.txt'
syncer_class = sync.StorageSyncer
# The string used in the config to denote the type of storage. Should be # The string used in the config to denote the type of storage. Should be
# overridden by subclasses. # overridden by subclasses.
storage_name = None storage_name = None

31
vdirsyncer/sync.py
View file

@ -76,7 +76,7 @@ class BothReadOnly(SyncError):
''' '''
class StorageInfo(object): class StorageSyncer(object):
'''A wrapper class that holds prefetched items, the status and other '''A wrapper class that holds prefetched items, the status and other
things.''' things.'''
def __init__(self, storage, status): def __init__(self, storage, status):
@ -122,6 +122,10 @@ class StorageInfo(object):
if props.setdefault('etag', etag) != etag: if props.setdefault('etag', etag) != etag:
raise SyncError('Etag changed during sync.') raise SyncError('Etag changed during sync.')
def is_changed(self, ident):
_, status_etag = self.status.get(ident, (None, None))
return self.idents[ident]['etag'] != status_etag
def sync(storage_a, storage_b, status, conflict_resolution=None, def sync(storage_a, storage_b, status, conflict_resolution=None,
force_delete=False): force_delete=False):
@ -146,11 +150,11 @@ def sync(storage_a, storage_b, status, conflict_resolution=None,
if storage_a.read_only and storage_b.read_only: if storage_a.read_only and storage_b.read_only:
raise BothReadOnly() raise BothReadOnly()
a_info = StorageInfo(storage_a, dict( a_info = storage_a.syncer_class(storage_a, dict(
(ident, (href_a, etag_a)) (ident, (href_a, etag_a))
for ident, (href_a, etag_a, href_b, etag_b) in iteritems(status) for ident, (href_a, etag_a, href_b, etag_b) in iteritems(status)
)) ))
b_info = StorageInfo(storage_b, dict( b_info = storage_b.syncer_class(storage_b, dict(
(ident, (href_b, etag_b)) (ident, (href_b, etag_b))
for ident, (href_a, etag_a, href_b, etag_b) in iteritems(status) for ident, (href_a, etag_a, href_b, etag_b) in iteritems(status)
)) ))
@ -283,27 +287,24 @@ def _action_conflict_resolve(ident):
def _get_actions(a_info, b_info): def _get_actions(a_info, b_info):
for ident in uniq(itertools.chain(a_info.idents, b_info.idents, for ident in uniq(itertools.chain(a_info.idents, b_info.idents,
a_info.status)): a_info.status)):
a = a_info.idents.get(ident, None) a = ident in a_info.idents # item exists in a
b = b_info.idents.get(ident, None) b = ident in b_info.idents # item exists in b
assert not a or a['etag'] is not None
assert not b or b['etag'] is not None
_, status_etag_a = a_info.status.get(ident, (None, None))
_, status_etag_b = b_info.status.get(ident, (None, None))
if a and b: if a and b:
if a['etag'] != status_etag_a and b['etag'] != status_etag_b: a_changed = a_info.is_changed(ident)
b_changed = b_info.is_changed(ident)
if a_changed and b_changed:
# item was modified on both sides # item was modified on both sides
# OR: missing status # OR: missing status
yield _action_conflict_resolve(ident) yield _action_conflict_resolve(ident)
elif a['etag'] != status_etag_a: elif a_changed and not b_changed:
# item was only modified in a # item was only modified in a
yield _action_update(ident, a_info, b_info) yield _action_update(ident, a_info, b_info)
elif b['etag'] != status_etag_b: elif not a_changed and b_changed:
# item was only modified in b # item was only modified in b
yield _action_update(ident, b_info, a_info) yield _action_update(ident, b_info, a_info)
elif a and not b: elif a and not b:
if a['etag'] != status_etag_a: if a_info.is_changed(ident):
# was deleted from b but modified on a # was deleted from b but modified on a
# OR: new item was created in a # OR: new item was created in a
yield _action_upload(ident, a_info, b_info) yield _action_upload(ident, a_info, b_info)
@ -311,7 +312,7 @@ def _get_actions(a_info, b_info):
# was deleted from b and not modified on a # was deleted from b and not modified on a
yield _action_delete(ident, a_info) yield _action_delete(ident, a_info)
elif not a and b: elif not a and b:
if b['etag'] != status_etag_b: if b_info.is_changed(ident):
# was deleted from a but modified on b # was deleted from a but modified on b
# OR: new item was created in b # OR: new item was created in b
yield _action_upload(ident, b_info, a_info) yield _action_upload(ident, b_info, a_info)