From d1ee38bbd085c492c883b8b84cd305062a78e661 Mon Sep 17 00:00:00 2001 From: Adrian Moennich Date: Mon, 10 Nov 2014 15:47:15 +0100 Subject: [PATCH] Handle initial export --- livesync_debug/indico_livesync_debug/agent.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/livesync_debug/indico_livesync_debug/agent.py b/livesync_debug/indico_livesync_debug/agent.py index 65e55c1..7a2838a 100644 --- a/livesync_debug/indico_livesync_debug/agent.py +++ b/livesync_debug/indico_livesync_debug/agent.py @@ -18,6 +18,7 @@ from __future__ import unicode_literals from indico.core.db import db from indico.util.console import cformat +from indico.util.struct.iterables import grouper from indico_livesync import LiveSyncAgentBase, SimpleChange, MARCXMLGenerator, process_records from indico_livesync.util import obj_deref @@ -56,3 +57,10 @@ class LiveSyncDebugAgent(LiveSyncAgentBase): for record in records: record.processed = True db.session.commit() + + def run_initial_export(self, events): + for i, batch in enumerate(grouper(events, 10), 1): + print + print cformat('%{white!}Batch {}:%{reset}').format(i) + print MARCXMLGenerator.objects_to_xml(event for event in batch if event is not None) + print