Source code for timelinelib.wxgui.dialogs.importevents.controller

# Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018  Rickard Lindberg, Roger Lindberg
#
# This file is part of Timeline.
#
# Timeline 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 3 of the License, or
# (at your option) any later version.
#
# Timeline 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 Timeline.  If not, see <http://www.gnu.org/licenses/>.


import os.path

from timelinelib.canvas.data.exceptions import TimelineIOError
from timelinelib.db import db_open
from timelinelib.utils import ex_msg
from timelinelib.wxgui.framework import Controller


class ImportEventsDialogController(Controller):

    def on_init(self, db):
        self._db = db
        self._db_to_import = None
        self._show_preview()

    def on_file_path_changed(self, event):
        self._show_preview()

    def on_ok_clicked(self, event):
        if not self._db_to_import:
            return
        self._db.import_db(self._db_to_import)
        self.view.Close()

    def _show_preview(self):
        if self._path_exists():
            self._handle_valid_path()
        else:
            self._handle_invalid_path()

    def _path_exists(self):
        path = self.view.GetFilePath()
        return path in [":tutorial:", ":numtutorial:"] or os.path.exists(path)

    def _handle_invalid_path(self):
        self._set_error(_("File does not exist."))

    def _handle_valid_path(self):
        try:
            db_to_import = db_open(self.view.GetFilePath())
        except Exception as e:
            self._set_error(_("Unable to load events: %s.") % ex_msg(e))
        else:
            self._report_nbr_of_events_in_db(db_to_import)

    def _report_nbr_of_events_in_db(self, db_to_import):
        if self._is_same_timetype(db_to_import):
            self._set_success(db_to_import, _("%d events will be imported." % len(db_to_import.get_all_events())))
        else:
            self._set_error(_("The selected timeline has a different time type."))

    def _is_same_timetype(self, db_to_import):
        return db_to_import.get_time_type() == self._db.get_time_type()

    def _set_success(self, db_to_import, text):
        self._db_to_import = db_to_import
        self.view.SetSuccess(text)

    def _set_error(self, text):
        self._db_to_import = None
        self.view.SetError(text)