forked from openlp/openlp
311 lines
12 KiB
Python
311 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vim: autoindent shiftwidth=4 expandtab textwidth=80 tabstop=4 softtabstop=4
|
|
|
|
###############################################################################
|
|
# OpenLP - Open Source Lyrics Projection #
|
|
# --------------------------------------------------------------------------- #
|
|
# Copyright (c) 2008-2010 Raoul Snyman #
|
|
# Portions copyright (c) 2008-2010 Tim Bentley, Jonathan Corwin, Michael #
|
|
# Gorven, Scott Guerrieri, Christian Richter, Maikel Stuivenberg, Martin #
|
|
# Thompson, Jon Tibble, Carsten Tinggaard #
|
|
# --------------------------------------------------------------------------- #
|
|
# 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 os
|
|
import logging
|
|
import chardet
|
|
import re
|
|
|
|
from sqlalchemy import or_
|
|
from PyQt4 import QtCore
|
|
|
|
from openlp.plugins.bibles.lib.models import *
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
class BibleDB(QtCore.QObject):
|
|
"""
|
|
This class represents a database-bound Bible. It is used as a base class
|
|
for all the custom importers, so that the can implement their own import
|
|
methods, but benefit from the database methods in here via inheritance,
|
|
rather than depending on yet another object.
|
|
"""
|
|
|
|
def __init__(self, parent, **kwargs):
|
|
"""
|
|
The constructor loads up the database and creates and initialises the
|
|
tables if the database doesn't exist.
|
|
|
|
**Required keyword arguments:**
|
|
|
|
``path``
|
|
The path to the bible database file.
|
|
|
|
``name``
|
|
The name of the database. This is also used as the file name for
|
|
SQLite databases.
|
|
|
|
``config``
|
|
The configuration object, passed in from the plugin.
|
|
"""
|
|
log.info(u'BibleDB loaded')
|
|
QtCore.QObject.__init__(self)
|
|
if u'path' not in kwargs:
|
|
raise KeyError(u'Missing keyword argument "path".')
|
|
if u'config' not in kwargs:
|
|
raise KeyError(u'Missing keyword argument "config".')
|
|
if u'name' not in kwargs and u'file' not in kwargs:
|
|
raise KeyError(u'Missing keyword argument "name" or "file".')
|
|
self.stop_import_flag = False
|
|
self.config = kwargs[u'config']
|
|
if u'name' in kwargs:
|
|
self.name = kwargs[u'name']
|
|
if not isinstance(self.name, unicode):
|
|
self.name = unicode(self.name, u'utf-8')
|
|
self.file = self.clean_filename(self.name)
|
|
if u'file' in kwargs:
|
|
self.file = kwargs[u'file']
|
|
self.db_file = os.path.join(kwargs[u'path'], self.file)
|
|
log.debug(u'Load bible %s on path %s', self.file, self.db_file)
|
|
db_type = self.config.get_config(u'db type', u'sqlite')
|
|
db_url = u''
|
|
if db_type == u'sqlite':
|
|
db_url = u'sqlite:///' + self.db_file
|
|
else:
|
|
db_url = u'%s://%s:%s@%s/%s' % \
|
|
(db_type, self.config.get_config(u'db username'),
|
|
self.config.get_config(u'db password'),
|
|
self.config.get_config(u'db hostname'),
|
|
self.config.get_config(u'db database'))
|
|
self.metadata, self.session = init_models(db_url)
|
|
self.metadata.create_all(checkfirst=True)
|
|
if u'file' in kwargs:
|
|
self.get_name()
|
|
|
|
def get_name(self):
|
|
version_name = self.get_meta(u'Version')
|
|
if version_name:
|
|
self.name = version_name.value
|
|
else:
|
|
self.name = None
|
|
return self.name
|
|
|
|
def clean_filename(self, old_filename):
|
|
if not isinstance(old_filename, unicode):
|
|
old_filename = unicode(old_filename, u'utf-8')
|
|
old_filename = re.sub(r'[^\w]+', u'_', old_filename).strip(u'_')
|
|
return old_filename + u'.sqlite'
|
|
|
|
def delete(self):
|
|
try:
|
|
os.remove(self.db_file)
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
def register(self, wizard):
|
|
"""
|
|
This method basically just initialialises the database. It is called
|
|
from the Bible Manager when a Bible is imported. Descendant classes
|
|
may want to override this method to suVersionpply their own custom
|
|
initialisation as well.
|
|
"""
|
|
self.wizard = wizard
|
|
self.create_tables()
|
|
return self.name
|
|
|
|
def commit(self):
|
|
log.debug('Committing...')
|
|
self.session.commit()
|
|
|
|
def create_tables(self):
|
|
log.debug(u'createTables')
|
|
self.create_meta(u'dbversion', u'2')
|
|
self.create_testament(u'Old Testament')
|
|
self.create_testament(u'New Testament')
|
|
self.create_testament(u'Apocrypha')
|
|
|
|
def create_testament(self, testament):
|
|
log.debug(u'BibleDB.create_testament("%s")', testament)
|
|
self.session.add(Testament.populate(name=testament))
|
|
self.commit()
|
|
|
|
def create_book(self, name, abbrev, testament=1):
|
|
log.debug(u'create_book %s,%s', name, abbrev)
|
|
book = Book.populate(name=name, abbreviation=abbrev,
|
|
testament_id=testament)
|
|
self.session.add(book)
|
|
self.commit()
|
|
return book
|
|
|
|
def create_chapter(self, book_id, chapter, textlist):
|
|
log.debug(u'create_chapter %s,%s', book_id, chapter)
|
|
#text list has book and chapter as first two elements of the array
|
|
for verse_number, verse_text in textlist.iteritems():
|
|
verse = Verse.populate(
|
|
book_id = book_id,
|
|
chapter = chapter,
|
|
verse = verse_number,
|
|
text = verse_text
|
|
)
|
|
self.session.add(verse)
|
|
self.commit()
|
|
|
|
def create_verse(self, book_id, chapter, verse, text):
|
|
if not isinstance(text, unicode):
|
|
details = chardet.detect(text)
|
|
text = unicode(text, details[u'encoding'])
|
|
verse = Verse.populate(
|
|
book_id=book_id,
|
|
chapter=chapter,
|
|
verse=verse,
|
|
text=text
|
|
)
|
|
self.session.add(verse)
|
|
return verse
|
|
|
|
def create_meta(self, key, value):
|
|
log.debug(u'save_meta %s/%s', key, value)
|
|
self.session.add(BibleMeta.populate(key=key, value=value))
|
|
self.commit()
|
|
|
|
def get_books(self):
|
|
log.debug(u'BibleDB.get_books()')
|
|
return self.session.query(Book).order_by(Book.id).all()
|
|
|
|
def get_book(self, book):
|
|
log.debug(u'BibleDb.get_book("%s")', book)
|
|
db_book = self.session.query(Book)\
|
|
.filter(Book.name.like(book + u'%'))\
|
|
.first()
|
|
if db_book is None:
|
|
db_book = self.session.query(Book)\
|
|
.filter(Book.abbreviation.like(book + u'%'))\
|
|
.first()
|
|
return db_book
|
|
|
|
def get_chapter(self, id, chapter):
|
|
log.debug(u'BibleDB.get_chapter("%s", %s)', id, chapter)
|
|
return self.session.query(Verse)\
|
|
.filter_by(chapter=chapter)\
|
|
.filter_by(book_id=id)\
|
|
.first()
|
|
|
|
def get_verses(self, reference_list):
|
|
"""
|
|
This is probably the most used function. It retrieves the list of
|
|
verses based on the user's query.
|
|
|
|
``reference_list``
|
|
This is the list of references the media manager item wants. It is
|
|
a list of tuples, with the following format::
|
|
|
|
(book, chapter, start_verse, end_verse)
|
|
|
|
Therefore, when you are looking for multiple items, simply break
|
|
them up into references like this, bundle them into a list. This
|
|
function then runs through the list, and returns an amalgamated
|
|
list of ``Verse`` objects. For example::
|
|
|
|
[(u'Genesis', 1, 1, 1), (u'Genesis', 2, 2, 3)]
|
|
"""
|
|
log.debug(u'BibleDB.get_verses: %s', reference_list)
|
|
verse_list = []
|
|
for book, chapter, start_verse, end_verse in reference_list:
|
|
db_book = self.get_book(book)
|
|
if end_verse == -1:
|
|
end_verse = self.get_verse_count(book, chapter)
|
|
if db_book:
|
|
book = db_book.name
|
|
log.debug(u'Book name corrected to "%s"', book)
|
|
verses = self.session.query(Verse)\
|
|
.filter_by(book_id=db_book.id)\
|
|
.filter_by(chapter=chapter)\
|
|
.filter(Verse.verse >= start_verse)\
|
|
.filter(Verse.verse <= end_verse)\
|
|
.order_by(Verse.verse)\
|
|
.all()
|
|
verse_list.extend(verses)
|
|
return verse_list
|
|
|
|
def verse_search(self, text):
|
|
"""
|
|
Search for verses containing text ``text``.
|
|
|
|
``text``
|
|
The text to search for. If the text contains commas, it will be
|
|
split apart and OR'd on the list of values. If the text just
|
|
contains spaces, it will split apart and AND'd on the list of
|
|
values.
|
|
"""
|
|
log.debug(u'BibleDB.verse_search("%s")', text)
|
|
verses = self.session.query(Verse)
|
|
if text.find(u',') > -1:
|
|
or_clause = []
|
|
keywords = [u'%%%s%%' % keyword.strip() for keyword in text.split(u',')]
|
|
for keyword in keywords:
|
|
or_clause.append(Verse.text.like(keyword))
|
|
verses = verses.filter(or_(*or_clause))
|
|
else:
|
|
keywords = [u'%%%s%%' % keyword.strip() for keyword in text.split(u' ')]
|
|
for keyword in keywords:
|
|
verses = verses.filter(Verse.text.like(keyword))
|
|
verses = verses.all()
|
|
return verses
|
|
|
|
def get_chapter_count(self, book):
|
|
log.debug(u'BibleDB.get_chapter_count("%s")', book)
|
|
count = self.session.query(Verse.chapter).join(Book)\
|
|
.filter(Book.name==book)\
|
|
.distinct().count()
|
|
if not count:
|
|
return 0
|
|
else:
|
|
return count
|
|
|
|
def get_verse_count(self, book, chapter):
|
|
log.debug(u'BibleDB.get_verse_count("%s", %s)', book, chapter)
|
|
count = self.session.query(Verse).join(Book)\
|
|
.filter(Book.name==book)\
|
|
.filter(Verse.chapter==chapter)\
|
|
.count()
|
|
if not count:
|
|
return 0
|
|
else:
|
|
return count
|
|
|
|
def get_meta(self, key):
|
|
log.debug(u'get meta %s', key)
|
|
return self.session.query(BibleMeta).get(key)
|
|
|
|
def delete_meta(self, metakey):
|
|
biblemeta = self.get_meta(metakey)
|
|
try:
|
|
self.session.delete(biblemeta)
|
|
self.commit()
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
def dump_bible(self):
|
|
log.debug(u'.........Dumping Bible Database')
|
|
log.debug('...............................Books ')
|
|
books = self.session.query(Book).all()
|
|
log.debug(books)
|
|
log.debug(u'...............................Verses ')
|
|
verses = self.session.query(Verse).all()
|
|
log.debug(verses)
|
|
|