forked from openlp/openlp
124 lines
6.6 KiB
Python
124 lines
6.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vim: autoindent shiftwidth=4 expandtab textwidth=120 tabstop=4 softtabstop=4
|
|
|
|
###############################################################################
|
|
# OpenLP - Open Source Lyrics Projection #
|
|
# --------------------------------------------------------------------------- #
|
|
# Copyright (c) 2008-2019 OpenLP Developers #
|
|
# --------------------------------------------------------------------------- #
|
|
# 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 logging
|
|
|
|
from PyQt5 import QtCore, QtWidgets
|
|
from sqlalchemy.sql import and_
|
|
|
|
from openlp.core.common.i18n import translate
|
|
from openlp.core.common.mixins import RegistryProperties
|
|
from openlp.core.common.path import create_paths
|
|
from openlp.core.common.settings import Settings
|
|
from openlp.core.lib.ui import critical_error_message_box
|
|
from openlp.plugins.songusage.lib.db import SongUsageItem
|
|
|
|
from .songusagedetaildialog import Ui_SongUsageDetailDialog
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class SongUsageDetailForm(QtWidgets.QDialog, Ui_SongUsageDetailDialog, RegistryProperties):
|
|
"""
|
|
Class documentation goes here.
|
|
"""
|
|
log.info('SongUsage Detail Form Loaded')
|
|
|
|
def __init__(self, plugin, parent):
|
|
"""
|
|
Initialise the form
|
|
"""
|
|
super(SongUsageDetailForm, self).__init__(parent, QtCore.Qt.WindowSystemMenuHint | QtCore.Qt.WindowTitleHint |
|
|
QtCore.Qt.WindowCloseButtonHint)
|
|
self.plugin = plugin
|
|
self.setup_ui(self)
|
|
|
|
def initialise(self):
|
|
"""
|
|
We need to set up the screen
|
|
"""
|
|
to_date = Settings().value(self.plugin.settings_section + '/to date')
|
|
if not (isinstance(to_date, QtCore.QDate) and to_date.isValid()):
|
|
to_date = QtCore.QDate.currentDate()
|
|
from_date = Settings().value(self.plugin.settings_section + '/from date')
|
|
if not (isinstance(from_date, QtCore.QDate) and from_date.isValid()):
|
|
from_date = to_date.addYears(-1)
|
|
self.from_date_calendar.setSelectedDate(from_date)
|
|
self.to_date_calendar.setSelectedDate(to_date)
|
|
self.report_path_edit.path = Settings().value(self.plugin.settings_section + '/last directory export')
|
|
|
|
def on_report_path_edit_path_changed(self, file_path):
|
|
"""
|
|
Handle the `pathEditChanged` signal from report_path_edit
|
|
|
|
:param openlp.core.common.path.Path file_path: The new path.
|
|
:rtype: None
|
|
"""
|
|
Settings().setValue(self.plugin.settings_section + '/last directory export', file_path)
|
|
|
|
def accept(self):
|
|
"""
|
|
Ok was triggered so lets save the data and run the report
|
|
"""
|
|
log.debug('accept')
|
|
path = self.report_path_edit.path
|
|
if not path:
|
|
self.main_window.error_message(
|
|
translate('SongUsagePlugin.SongUsageDetailForm', 'Output Path Not Selected'),
|
|
translate('SongUsagePlugin.SongUsageDetailForm', 'You have not set a valid output location for your'
|
|
' song usage report.\nPlease select an existing path on your computer.')
|
|
)
|
|
return
|
|
create_paths(path)
|
|
file_name = translate('SongUsagePlugin.SongUsageDetailForm',
|
|
'usage_detail_{old}_{new}.txt'
|
|
).format(old=self.from_date_calendar.selectedDate().toString('ddMMyyyy'),
|
|
new=self.to_date_calendar.selectedDate().toString('ddMMyyyy'))
|
|
Settings().setValue(self.plugin.settings_section + '/from date', self.from_date_calendar.selectedDate())
|
|
Settings().setValue(self.plugin.settings_section + '/to date', self.to_date_calendar.selectedDate())
|
|
usage = self.plugin.manager.get_all_objects(
|
|
SongUsageItem, and_(SongUsageItem.usagedate >= self.from_date_calendar.selectedDate().toPyDate(),
|
|
SongUsageItem.usagedate < self.to_date_calendar.selectedDate().toPyDate()),
|
|
[SongUsageItem.usagedate, SongUsageItem.usagetime])
|
|
report_file_name = path / file_name
|
|
try:
|
|
with report_file_name.open('wb') as file_handle:
|
|
for instance in usage:
|
|
record = ('\"{date}\",\"{time}\",\"{title}\",\"{copyright}\",\"{ccli}\",\"{authors}\",'
|
|
'\"{name}\",\"{source}\"\n').format(date=instance.usagedate, time=instance.usagetime,
|
|
title=instance.title, copyright=instance.copyright,
|
|
ccli=instance.ccl_number, authors=instance.authors,
|
|
name=instance.plugin_name, source=instance.source)
|
|
file_handle.write(record.encode('utf-8'))
|
|
self.main_window.information_message(
|
|
translate('SongUsagePlugin.SongUsageDetailForm', 'Report Creation'),
|
|
translate('SongUsagePlugin.SongUsageDetailForm',
|
|
'Report\n{name}\nhas been successfully created.').format(name=report_file_name)
|
|
)
|
|
except OSError as ose:
|
|
log.exception('Failed to write out song usage records')
|
|
critical_error_message_box(translate('SongUsagePlugin.SongUsageDetailForm', 'Report Creation Failed'),
|
|
translate('SongUsagePlugin.SongUsageDetailForm',
|
|
'An error occurred while creating the report: {error}'
|
|
).format(error=ose.strerror))
|
|
self.close()
|