From c849a67b986c826f128b045181d8b880744de104 Mon Sep 17 00:00:00 2001 From: Kali Kaneko Date: Fri, 5 Jun 2015 11:41:32 -0400 Subject: [feature] filter documents before passing them to the plugin --- client/src/leap/soledad/client/api.py | 17 +++++++++-------- client/src/leap/soledad/client/sync.py | 8 +------- 2 files changed, 10 insertions(+), 15 deletions(-) diff --git a/client/src/leap/soledad/client/api.py b/client/src/leap/soledad/client/api.py index df69037c..01c06f20 100644 --- a/client/src/leap/soledad/client/api.py +++ b/client/src/leap/soledad/client/api.py @@ -38,6 +38,7 @@ try: import cchardet as chardet except ImportError: import chardet +from itertools import chain from StringIO import StringIO from u1db.remote import http_client @@ -679,19 +680,19 @@ class Soledad(object): defer_decryption=defer_decryption) def _sync_callback(local_gen): - self._last_received_docs = self._dbsyncer.received_docs - print "***" - print "LAST RECEIVED (API)", self._last_received_docs - print "***" - received_doc_ids = self._dbsyncer.received_docs + self._last_received_docs = docs = self._dbsyncer.received_docs # Post-Sync Hooks synced_plugin = soledad_interfaces.ISoledadPostSyncPlugin - if received_doc_ids: + if docs: suitable_plugins = collect_plugins(synced_plugin) for plugin in suitable_plugins: - # TODO filter the doc_ids here - plugin.process_received_docs(received_doc_ids) + watched = plugin.watched_doc_types + r = [filter( + lambda s: s.startswith(preffix), + docs) for preffix in watched] + filtered = list(chain(*r)) + plugin.process_received_docs(filtered) soledad_events.emit( soledad_events.SOLEDAD_DONE_DATA_SYNC, self.uuid) diff --git a/client/src/leap/soledad/client/sync.py b/client/src/leap/soledad/client/sync.py index 917c21ea..14c547cf 100644 --- a/client/src/leap/soledad/client/sync.py +++ b/client/src/leap/soledad/client/sync.py @@ -158,15 +158,9 @@ class SoledadSynchronizer(Synchronizer): _, _, changes = self.source.whats_changed(target_my_gen) changed_doc_ids = [doc_id for doc_id, _, _ in changes] - print "--------------------------" - print "SENT", ids_sent - print "CHANGED_DOC_IDS", changed_doc_ids - just_received = list(set(changed_doc_ids) - set(ids_sent)) - print "RECEIVED:", just_received - print "--------------------------" - self.received_docs = just_received + defer.returnValue(my_gen) def complete_sync(self): -- cgit v1.2.3