From 4782ec70a4f70b7735bb9cb3ef0207448229c02a Mon Sep 17 00:00:00 2001 From: Duda Dornelles Date: Tue, 21 Oct 2014 14:59:45 +0200 Subject: Adding sync_info controller for #117 --- service/pixelated/config/app_factory.py | 9 +++-- service/pixelated/controllers/__init__.py | 1 + .../pixelated/controllers/sync_info_controller.py | 42 ++++++++++++++++++++++ 3 files changed, 50 insertions(+), 2 deletions(-) create mode 100644 service/pixelated/controllers/sync_info_controller.py (limited to 'service/pixelated') diff --git a/service/pixelated/config/app_factory.py b/service/pixelated/config/app_factory.py index 1d006506..9ea00bb9 100644 --- a/service/pixelated/config/app_factory.py +++ b/service/pixelated/config/app_factory.py @@ -30,6 +30,7 @@ from leap.common.events import ( register, events_pb2 as proto ) +from pixelated.controllers import sync_info_controller def init_index_and_delete_duplicate_inboxes(querier, search_engine, mail_service): @@ -40,7 +41,7 @@ def init_index_and_delete_duplicate_inboxes(querier, search_engine, mail_service return wrapper -def _setup_routes(app, home_controller, mails_controller, tags_controller, features_controller): +def _setup_routes(app, home_controller, mails_controller, tags_controller, features_controller, sync_info_controller): # home app.add_url_rule('/', methods=['GET'], view_func=home_controller.home) # mails @@ -59,6 +60,8 @@ def _setup_routes(app, home_controller, mails_controller, tags_controller, featu app.add_url_rule('/tags', methods=['GET'], view_func=tags_controller.tags) # features app.add_url_rule('/features', methods=['GET'], view_func=features_controller.features) + # sync info + app.add_url_rule('/sync_info', methods=['GET'], view_func=sync_info_controller.sync_info) def create_app(debug_enabled, app): @@ -85,13 +88,15 @@ def create_app(debug_enabled, app): draft_service=draft_service, search_engine=search_engine) tags_controller = TagsController(search_engine=search_engine) + sync_info_controller = SyncInfoController() + register(signal=proto.SOLEDAD_SYNC_RECEIVE_STATUS, callback=sync_info_controller.set_sync_info) register(signal=proto.SOLEDAD_DONE_DATA_SYNC, callback=init_index_and_delete_duplicate_inboxes(querier=soledad_querier, search_engine=search_engine, mail_service=mail_service)) - _setup_routes(app, home_controller, mails_controller, tags_controller, features_controller) + _setup_routes(app, home_controller, mails_controller, tags_controller, features_controller, sync_info_controller) app.run(host=app.config['HOST'], debug=debug_enabled, port=app.config['PORT'], use_reloader=False) diff --git a/service/pixelated/controllers/__init__.py b/service/pixelated/controllers/__init__.py index 969e8e6f..2976b8cb 100644 --- a/service/pixelated/controllers/__init__.py +++ b/service/pixelated/controllers/__init__.py @@ -29,3 +29,4 @@ from home_controller import HomeController from mails_controller import MailsController from tags_controller import TagsController from features_controller import FeaturesController +from sync_info_controller import SyncInfoController \ No newline at end of file diff --git a/service/pixelated/controllers/sync_info_controller.py b/service/pixelated/controllers/sync_info_controller.py new file mode 100644 index 00000000..5adfcd48 --- /dev/null +++ b/service/pixelated/controllers/sync_info_controller.py @@ -0,0 +1,42 @@ +# +# Copyright (c) 2014 ThoughtWorks, Inc. +# +# Pixelated 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. +# +# Pixelated 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 Pixelated. If not, see . +from pixelated.controllers import respond_json + + +class SyncInfoController: + + def __init__(self): + self.current = 0 + self.total = 0 + + def _get_progress(self): + if self.total == 0: + return 0 + return self.current/float(self.total) + + def set_sync_info(self, soledad_sync_status): + self.current, self.total = map(int, soledad_sync_status.content.split('/')) + + def sync_info(self): + _sync_info = { + 'is_syncing': self.current != self.total, + 'count': { + 'current': self.current, + 'total': self.total, + 'progress': self._get_progress() + } + } + return respond_json(_sync_info) \ No newline at end of file -- cgit v1.2.3