openlp/openlp/core/ui/firsttimeform.py

# -*- coding: utf-8 -*-
# vim: autoindent shiftwidth=4 expandtab textwidth=80 tabstop=4 softtabstop=4

###############################################################################
# OpenLP - Open Source Lyrics Projection                                      #
# --------------------------------------------------------------------------- #
# Copyright (c) 2008-2011 Raoul Snyman                                        #
# Portions copyright (c) 2008-2011 Tim Bentley, Jonathan Corwin, Michael      #
# Gorven, Scott Guerrieri, Meinert Jordan, Armin Köhler, Andreas Preikschat,  #
# Christian Richter, Philip Ridout, Maikel Stuivenberg, Martin Thompson, Jon  #
# Tibble, Carsten Tinggaard, Frode Woldsund                                   #
# --------------------------------------------------------------------------- #
# This program 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; version 2 of the License.                              #
#                                                                             #
# This program 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 this program; if not, write to the Free Software Foundation, Inc., 59  #
# Temple Place, Suite 330, Boston, MA 02111-1307 USA                          #
###############################################################################

import ConfigParser
import io
import logging
import os
import urllib

from PyQt4 import QtCore, QtGui

from firsttimewizard import Ui_FirstTimeWizard

from openlp.core.lib import translate, PluginStatus, check_directory_exists,  \
    Receiver
from openlp.core.utils import get_web_page, AppLocation

log = logging.getLogger(__name__)

class FirstTimeForm(QtGui.QWizard, Ui_FirstTimeWizard):
    """
    This is the Theme Import Wizard, which allows easy creation and editing of
    OpenLP themes.
    """
    log.info(u'ThemeWizardForm loaded')

    def __init__(self, screens, parent=None):
        QtGui.QWizard.__init__(self, parent)
        # check to see if we have web access
        self.web = u'http://openlp.org/files/frw/'
        self.config = ConfigParser.ConfigParser()
        self.webAccess = get_web_page(u'%s%s' % (self.web, u'download.cfg'))
        if self.webAccess:
            files = self.webAccess.read()
            self.config.readfp(io.BytesIO(files))
        self.setupUi(self)
        for screen in screens.get_screen_list():
            self.displaySelectionComboBox.addItem(screen)
        self.songsText = translate('OpenLP.FirstTimeWizard', 'Songs')
        self.biblesText = translate('OpenLP.FirstTimeWizard', 'Bibles')
        self.themesText = translate('OpenLP.FirstTimeWizard', 'Themes')
        self.startUpdates = translate('OpenLP.FirstTimeWizard',
            'Starting Updates')
        self.downloading = unicode(translate('OpenLP.FirstTimeWizard',
            'Downloading %s'))
        QtCore.QObject.connect(self,
            QtCore.SIGNAL(u'currentIdChanged(int)'),
            self.onCurrentIdChanged)

    def exec_(self, edit=False):
        """
        Run the wizard.
        """
        self.setDefaults()
        return QtGui.QWizard.exec_(self)

    def setDefaults(self):
        """
        Set up display at start of theme edit.
        """
        self.restart()
        # Sort out internet access for downloads
        if self.webAccess:
            self.internetGroupBox.setVisible(True)
            self.noInternetLabel.setVisible(False)
            # If songs database exists do not allow a copy
            songs = os.path.join(AppLocation.get_section_data_path(u'songs'),
                u'songs.sqlite')
            if not os.path.exists(songs):
                treewidgetitem = QtGui.QTreeWidgetItem(self.selectionTreeWidget)
                treewidgetitem.setText(0, self.songsText)
                self._loadChild(treewidgetitem, u'songs', u'languages', u'songs')
            treewidgetitem = QtGui.QTreeWidgetItem(self.selectionTreeWidget)
            treewidgetitem.setText(0, self.biblesText)
            self._loadChild(treewidgetitem, u'bibles', u'translations',
                u'bible')
            treewidgetitem = QtGui.QTreeWidgetItem(self.selectionTreeWidget)
            treewidgetitem.setText(0, self.themesText)
            self._loadChild(treewidgetitem, u'themes', u'files', 'theme')
        else:
            self.internetGroupBox.setVisible(False)
            self.noInternetLabel.setVisible(True)

    def _loadChild(self, tree, list, tag, root):
        files = self.config.get(list, tag)
        files = files.split(u',')
        for file in files:
            if file:
                child = QtGui.QTreeWidgetItem(tree)
                child.setText(0, self.config.get(u'%s_%s'
                    % (root, file), u'title'))
                child.setData(0, QtCore.Qt.UserRole,
                    QtCore.QVariant(self.config.get(u'%s_%s'
                    % (root, file), u'filename')))
                child.setCheckState(0, QtCore.Qt.Unchecked)
                child.setFlags(QtCore.Qt.ItemIsUserCheckable |
                    QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)

    def onCurrentIdChanged(self, pageId):
        """
        Detects Page changes and updates as approprate.
        """
        if self.page(pageId) == self.DefaultsPage:
            self.themeSelectionComboBox.clear()
            listIterator = QtGui.QTreeWidgetItemIterator(
                self.selectionTreeWidget)
            while listIterator.value():
                parent = listIterator.value().parent()
                if parent and listIterator.value().checkState(0) \
                    == QtCore.Qt.Checked:
                    if unicode(parent.text(0)) == self.themesText:
                        self.themeSelectionComboBox.addItem(
                            listIterator.value().text(0))
                listIterator += 1

    def accept(self):
        Receiver.send_message(u'cursor_busy')
        self._updateMessage(self.startUpdates)
        # Set up the Plugin status's
        self._pluginStatus(self.songsCheckBox, u'songs/status')
        self._pluginStatus(self.bibleCheckBox, u'bibles/status')
        self._pluginStatus(self.presentationCheckBox, u'presentations/status')
        self._pluginStatus(self.imageCheckBox, u'images/status')
        self._pluginStatus(self.mediaCheckBox, u'media/status')
        self._pluginStatus(self.remoteCheckBox, u'remotes/status')
        self._pluginStatus(self.customCheckBox, u'custom/status')
        self._pluginStatus(self.songUsageCheckBox, u'songusage/status')
        self._pluginStatus(self.alertCheckBox, u'alerts/status')
        # Build directories for downloads
        songsDestination = AppLocation.get_section_data_path(u'songs')
        check_directory_exists(songsDestination)
        bibleDestination = AppLocation.get_section_data_path(u'bibles')
        check_directory_exists(bibleDestination)
        themeDestination = AppLocation.get_section_data_path(u'themes')
        check_directory_exists(themeDestination)
        # Install Selected Items looping through them
        listIterator = QtGui.QTreeWidgetItemIterator(self.selectionTreeWidget)
        while listIterator.value():
            type = listIterator.value().parent()
            if listIterator.value().parent():
                if listIterator.value().checkState(0) == QtCore.Qt.Checked:
                    # Install items as theu have been selected
                    item = unicode(listIterator.value().text(0))
                    # Download Song database if selected
                    if unicode(type.text(0)) == self.songsText:
                        songs = unicode(listIterator.value().data(0,
                            QtCore.Qt.UserRole).toString())
                        message = self.downloading % item
                        self._updateMessage(message)
                        # Song database is a fixed file name
                        urllib.urlretrieve(u'%s%s' % (self.web, songs),
                            os.path.join(songsDestination, u'songs.sqlite'))
                    # Download and selected Bibles
                    if unicode(type.text(0)) == self.biblesText:
                        bible = unicode(listIterator.value().data(0,
                            QtCore.Qt.UserRole).toString())
                        message = self.downloading % item
                        self._updateMessage(message)
                        urllib.urlretrieve(u'%s%s' % (self.web, bible),
                            os.path.join(bibleDestination, bible))
                    # Download any themes
                    if unicode(type.text(0)) == self.themesText:
                        theme = unicode(listIterator.value().data(0,
                            QtCore.Qt.UserRole).toString())
                        message = self.downloading % item
                        self._updateMessage(message)
                        urllib.urlretrieve(u'%s%s' % (self.web, theme),
                            os.path.join(themeDestination, theme))
            listIterator += 1
        # Set Default Display
        if self.displaySelectionComboBox.currentIndex() != -1:
            QtCore.QSettings().setValue(u'General/monitor',
                QtCore.QVariant(self.displaySelectionComboBox.
                currentIndex()))
        # Set Global Theme
        if self.themeSelectionComboBox.currentIndex() != -1:
            QtCore.QSettings().setValue(u'themes/global theme',
                QtCore.QVariant(self.themeSelectionComboBox.currentText()))
        QtCore.QSettings().setValue(u'general/first time',
            QtCore.QVariant(False))
        Receiver.send_message(u'cursor_normal')
        return QtGui.QWizard.accept(self)

    def _pluginStatus(self, field, tag):
        status = PluginStatus.Active if field.checkState() \
            == QtCore.Qt.Checked else PluginStatus.Inactive
        QtCore.QSettings().setValue(tag, QtCore.QVariant(status))

    def _updateMessage(self, text):
        """
        Keep screen up to date
        """
        self.updateLabel.setText(text)
        Receiver.send_message(u'openlp_process_events')