forked from openlp/openlp
Importers have replaced scripts
This commit is contained in:
Jon Tibble
parent
004a38fd7f
commit
7bd9d8a10b
@ -1,308 +0,0 @@
|
|||||||
#!/usr/bin/env 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, Meinert Jordan, 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 sys
|
|
||||||
import os
|
|
||||||
import sqlite
|
|
||||||
import sqlite3
|
|
||||||
|
|
||||||
from optparse import OptionParser
|
|
||||||
from traceback import format_tb as get_traceback
|
|
||||||
|
|
||||||
# Some global options to be used throughout the import process
|
|
||||||
verbose = False
|
|
||||||
debug = False
|
|
||||||
old_cursor = None
|
|
||||||
new_cursor = None
|
|
||||||
|
|
||||||
# SQL create statments
|
|
||||||
create_statements = [
|
|
||||||
(u'table "book"', u"""CREATE TABLE book (
|
|
||||||
id INTEGER NOT NULL,
|
|
||||||
testament_id INTEGER,
|
|
||||||
name VARCHAR(30),
|
|
||||||
abbreviation VARCHAR(5),
|
|
||||||
PRIMARY KEY (id),
|
|
||||||
FOREIGN KEY(testament_id) REFERENCES testament (id)
|
|
||||||
)"""),
|
|
||||||
(u'table "metadata"', u"""CREATE TABLE metadata (
|
|
||||||
"key" VARCHAR(255) NOT NULL,
|
|
||||||
value VARCHAR(255),
|
|
||||||
PRIMARY KEY ("key")
|
|
||||||
)"""),
|
|
||||||
(u'table "testament"', u"""CREATE TABLE testament (
|
|
||||||
id INTEGER NOT NULL,
|
|
||||||
name VARCHAR(30),
|
|
||||||
PRIMARY KEY (id)
|
|
||||||
)"""),
|
|
||||||
(u'table "verse"', u"""CREATE TABLE verse (
|
|
||||||
id INTEGER NOT NULL,
|
|
||||||
book_id INTEGER,
|
|
||||||
chapter INTEGER,
|
|
||||||
verse INTEGER,
|
|
||||||
text TEXT,
|
|
||||||
PRIMARY KEY (id),
|
|
||||||
FOREIGN KEY(book_id) REFERENCES book (id)
|
|
||||||
)"""),
|
|
||||||
(u'index "idx_abbrev"',
|
|
||||||
u"""CREATE INDEX idx_abbrev ON book (abbreviation, id)"""),
|
|
||||||
(u'index "idx_chapter_verse_book',
|
|
||||||
u"""CREATE INDEX idx_chapter_verse_book ON verse (chapter, verse, book_id, id)"""),
|
|
||||||
(u'index "idx_chapter_verse_text"',
|
|
||||||
u"""CREATE INDEX idx_chapter_verse_text ON verse (text, verse, book_id, id)"""),
|
|
||||||
(u'index "idx_name"',
|
|
||||||
u"""CREATE INDEX idx_name ON book (name, id)""")
|
|
||||||
]
|
|
||||||
|
|
||||||
def display_sql(sql, params):
|
|
||||||
prepared_params = []
|
|
||||||
for param in params:
|
|
||||||
if isinstance(param, basestring):
|
|
||||||
prepared_params.append(u'"%s"' % param)
|
|
||||||
elif isinstance(param, (int, long)):
|
|
||||||
prepared_params.append(u'%d' % param)
|
|
||||||
elif isinstance(param, (float, complex)):
|
|
||||||
prepared_params.append(u'%f' % param)
|
|
||||||
else:
|
|
||||||
prepared_params.append(u'"%s"' % str(param))
|
|
||||||
for prepared_param in prepared_params:
|
|
||||||
sql = sql.replace(u'?', prepared_param, 1)
|
|
||||||
return sql
|
|
||||||
|
|
||||||
def create_database():
|
|
||||||
global new_cursor, create_statements
|
|
||||||
if debug or verbose:
|
|
||||||
print 'Creating new database:'
|
|
||||||
else:
|
|
||||||
print 'Creating new database...',
|
|
||||||
for statement_type, sql_create in create_statements:
|
|
||||||
if debug:
|
|
||||||
print '... ', sql_create.replace('\n', ' ').replace(' ', ' ')
|
|
||||||
elif verbose:
|
|
||||||
print '... creating %s...' % statement_type,
|
|
||||||
new_cursor.execute(sql_create)
|
|
||||||
if verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
if not verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
|
|
||||||
def import_bible():
|
|
||||||
global old_cursor, new_cursor, debug, verbose
|
|
||||||
if debug or verbose:
|
|
||||||
print 'Importing metadata:'
|
|
||||||
else:
|
|
||||||
print 'Importing metadata...',
|
|
||||||
if debug:
|
|
||||||
print '... SELECT "key", "value" FROM metadata'
|
|
||||||
elif verbose:
|
|
||||||
print '... fetching metadata from old database...',
|
|
||||||
old_cursor.execute(u'SELECT "key", "value" FROM metadata')
|
|
||||||
rows = old_cursor.fetchall()
|
|
||||||
if not debug and verbose:
|
|
||||||
print 'done.'
|
|
||||||
for row in rows:
|
|
||||||
key = unicode(row[0], u'cp1252')
|
|
||||||
value = unicode(row[1], u'cp1252')
|
|
||||||
if key == u'Permission':
|
|
||||||
key = u'Permissions'
|
|
||||||
sql_insert = u'INSERT INTO metadata '\
|
|
||||||
'("key", "value") '\
|
|
||||||
'VALUES (?, ?)'
|
|
||||||
sql_params = (key, value)
|
|
||||||
if debug:
|
|
||||||
print '...', display_sql(sql_insert, sql_params)
|
|
||||||
elif verbose:
|
|
||||||
print '... importing "%s"' % key
|
|
||||||
new_cursor.execute(sql_insert, sql_params)
|
|
||||||
if not verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
if debug or verbose:
|
|
||||||
print 'Importing testaments:'
|
|
||||||
else:
|
|
||||||
print 'Importing testaments...',
|
|
||||||
if debug:
|
|
||||||
print '... SELECT id, name FROM testament'
|
|
||||||
elif verbose:
|
|
||||||
print '... fetching testaments from old database...',
|
|
||||||
old_cursor.execute(u'SELECT id, name FROM testament')
|
|
||||||
rows = old_cursor.fetchall()
|
|
||||||
if not debug and verbose:
|
|
||||||
print 'done.'
|
|
||||||
for row in rows:
|
|
||||||
id = int(row[0])
|
|
||||||
name = unicode(row[1], u'cp1252')
|
|
||||||
sql_insert = u'INSERT INTO testament '\
|
|
||||||
'(id, name) '\
|
|
||||||
'VALUES (?, ?)'
|
|
||||||
sql_params = (id, name)
|
|
||||||
if debug:
|
|
||||||
print '...', display_sql(sql_insert, sql_params)
|
|
||||||
elif verbose:
|
|
||||||
print '... importing "%s"' % name
|
|
||||||
new_cursor.execute(sql_insert, sql_params)
|
|
||||||
if not verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
if debug or verbose:
|
|
||||||
print 'Importing books:'
|
|
||||||
else:
|
|
||||||
print 'Importing books...',
|
|
||||||
if debug:
|
|
||||||
print '... SELECT id, testament_id, name, abbreviation FROM book'
|
|
||||||
elif verbose:
|
|
||||||
print '... fetching books from old database...',
|
|
||||||
old_cursor.execute(u'SELECT id, testament_id, name, abbreviation FROM book')
|
|
||||||
rows = old_cursor.fetchall()
|
|
||||||
if not debug and verbose:
|
|
||||||
print 'done.'
|
|
||||||
book_map = {}
|
|
||||||
for row in rows:
|
|
||||||
testament_id = int(row[1])
|
|
||||||
name = unicode(row[2], u'cp1252')
|
|
||||||
abbreviation = unicode(row[3], u'cp1252')
|
|
||||||
sql_insert = u'INSERT INTO book '\
|
|
||||||
'(id, testament_id, name, abbreviation) '\
|
|
||||||
'VALUES (NULL, ?, ?, ?)'
|
|
||||||
sql_params = (testament_id, name, abbreviation)
|
|
||||||
if debug:
|
|
||||||
print '...', display_sql(sql_insert, sql_params)
|
|
||||||
elif verbose:
|
|
||||||
print '... importing "%s"' % name
|
|
||||||
new_cursor.execute(sql_insert, sql_params)
|
|
||||||
book_map[row[0]] = new_cursor.lastrowid
|
|
||||||
if debug:
|
|
||||||
print ' >>> (old) books.id =', row[0], ' (new) books.id =', book_map[row[0]]
|
|
||||||
if not verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
if debug or verbose:
|
|
||||||
print 'Importing verses:'
|
|
||||||
else:
|
|
||||||
print 'Importing verses...',
|
|
||||||
if debug:
|
|
||||||
print '... SELECT id, book_id, chapter, verse, text || \'\' AS text FROM verse...',
|
|
||||||
elif verbose:
|
|
||||||
print '... fetching verses from old database...',
|
|
||||||
old_cursor.execute(u'SELECT id, book_id, chapter, verse, text || \'\' AS text FROM verse')
|
|
||||||
rows = old_cursor.fetchall()
|
|
||||||
if debug or verbose:
|
|
||||||
print 'done.'
|
|
||||||
for row in rows:
|
|
||||||
book_id = int(row[1])
|
|
||||||
chapter = int(row[2])
|
|
||||||
verse = int(row[3])
|
|
||||||
text = unicode(row[4], u'cp1252')
|
|
||||||
sql_insert = u'INSERT INTO verse '\
|
|
||||||
'(id, book_id, chapter, verse, text) '\
|
|
||||||
'VALUES (NULL, ?, ?, ?, ?)'
|
|
||||||
sql_params = (book_map[book_id], chapter, verse, text)
|
|
||||||
if debug:
|
|
||||||
print '...', display_sql(sql_insert, sql_params)
|
|
||||||
elif verbose:
|
|
||||||
print '... importing "%s..."' % text[:17]
|
|
||||||
new_cursor.execute(sql_insert, sql_params)
|
|
||||||
if not verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
|
|
||||||
def main(old_db, new_db):
|
|
||||||
global old_cursor, new_cursor, debug
|
|
||||||
old_connection = None
|
|
||||||
new_connection = None
|
|
||||||
try:
|
|
||||||
old_connection = sqlite.connect(old_db)
|
|
||||||
except:
|
|
||||||
if debug:
|
|
||||||
errormsg = '\n' + ''.join(get_traceback(sys.exc_info()[2]))\
|
|
||||||
+ str(sys.exc_info()[1])
|
|
||||||
else:
|
|
||||||
errormsg = sys.exc_info()[1]
|
|
||||||
print 'There was a problem connecting to the old database:', errormsg
|
|
||||||
return 1
|
|
||||||
try:
|
|
||||||
new_connection = sqlite3.connect(new_db)
|
|
||||||
except:
|
|
||||||
if debug:
|
|
||||||
errormsg = '\n' + ''.join(get_traceback(sys.exc_info()[2]))\
|
|
||||||
+ str(sys.exc_info()[1])
|
|
||||||
else:
|
|
||||||
errormsg = sys.exc_info()[1]
|
|
||||||
print 'There was a problem creating the new database:', errormsg
|
|
||||||
return 1
|
|
||||||
old_cursor = old_connection.cursor()
|
|
||||||
new_cursor = new_connection.cursor()
|
|
||||||
try:
|
|
||||||
create_database()
|
|
||||||
except:
|
|
||||||
if debug:
|
|
||||||
errormsg = '\n' + ''.join(get_traceback(sys.exc_info()[2]))\
|
|
||||||
+ str(sys.exc_info()[1])
|
|
||||||
else:
|
|
||||||
errormsg = sys.exc_info()[1]
|
|
||||||
print 'There was a problem creating the database:', errormsg
|
|
||||||
return 1
|
|
||||||
try:
|
|
||||||
import_bible()
|
|
||||||
new_connection.commit()
|
|
||||||
except:
|
|
||||||
new_connection.rollback()
|
|
||||||
if debug:
|
|
||||||
errormsg = '\n' + ''.join(get_traceback(sys.exc_info()[2]))\
|
|
||||||
+ str(sys.exc_info()[1])
|
|
||||||
else:
|
|
||||||
errormsg = sys.exc_info()[1]
|
|
||||||
print 'There was a problem importing songs:', errormsg
|
|
||||||
return 1
|
|
||||||
print 'Import complete.'
|
|
||||||
|
|
||||||
if __name__ == u'__main__':
|
|
||||||
option_parser = OptionParser(usage='Usage: %prog [options] OLDDATABASE NEWDATABASE')
|
|
||||||
option_parser.add_option('-o', '--overwrite', dest='overwrite', default=False,
|
|
||||||
action=u'store_true', help='Overwrite database file if it already exists.')
|
|
||||||
option_parser.add_option('-v', '--verbose', dest='verbose', default=False,
|
|
||||||
action=u'store_true', help='Outputs additional progress data.')
|
|
||||||
option_parser.add_option('-d', '--debug', dest='debug', default=False,
|
|
||||||
action=u'store_true', help='Outputs raw SQL statements (overrides verbose).')
|
|
||||||
options, arguments = option_parser.parse_args()
|
|
||||||
if len(arguments) < 2:
|
|
||||||
if len(arguments) == 0:
|
|
||||||
option_parser.error('Please specify an old database and a new database.')
|
|
||||||
else:
|
|
||||||
option_parser.error('Please specify a new database.')
|
|
||||||
old_db = os.path.abspath(arguments[0])
|
|
||||||
new_db = os.path.abspath(arguments[1])
|
|
||||||
if not os.path.isfile(old_db):
|
|
||||||
option_parser.error('Old database file ("%s") is not a file.' % old_db)
|
|
||||||
if not os.path.exists(old_db):
|
|
||||||
option_parser.error('Old database file ("%s") does not exist.' % old_db)
|
|
||||||
if os.path.exists(new_db):
|
|
||||||
if not options.overwrite:
|
|
||||||
option_parser.error('New database file ("%s") exists. If you want to overwrite it, specify the --overwrite option.' % new_db)
|
|
||||||
else:
|
|
||||||
if not os.path.isfile(new_db):
|
|
||||||
option_parser.error('New database file ("%s") is not a file.' % new_db)
|
|
||||||
os.unlink(new_db)
|
|
||||||
verbose = options.verbose
|
|
||||||
debug = options.debug
|
|
||||||
main(old_db, new_db)
|
|
||||||
@ -1,323 +0,0 @@
|
|||||||
#!/usr/bin/env 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, Meinert Jordan, 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 sys
|
|
||||||
import os
|
|
||||||
import sqlite
|
|
||||||
import sqlite3
|
|
||||||
import re
|
|
||||||
from optparse import OptionParser
|
|
||||||
from traceback import format_tb as get_traceback
|
|
||||||
|
|
||||||
# Some global options to be used throughout the import process
|
|
||||||
dirty_chars = re.compile(r'\W ', re.UNICODE)
|
|
||||||
verbose = False
|
|
||||||
debug = False
|
|
||||||
old_cursor = None
|
|
||||||
new_cursor = None
|
|
||||||
|
|
||||||
# SQL create statments
|
|
||||||
create_statements = [
|
|
||||||
(u'table "authors"', u"""CREATE TABLE authors (
|
|
||||||
id INTEGER NOT NULL,
|
|
||||||
first_name VARCHAR(128),
|
|
||||||
last_name VARCHAR(128),
|
|
||||||
display_name VARCHAR(255) NOT NULL,
|
|
||||||
PRIMARY KEY (id)
|
|
||||||
)"""),
|
|
||||||
(u'table "song_books"', u"""CREATE TABLE song_books (
|
|
||||||
id INTEGER NOT NULL,
|
|
||||||
name VARCHAR(128) NOT NULL,
|
|
||||||
publisher VARCHAR(128),
|
|
||||||
PRIMARY KEY (id)
|
|
||||||
)"""),
|
|
||||||
(u'table "songs"', u"""CREATE TABLE songs (
|
|
||||||
id INTEGER NOT NULL,
|
|
||||||
song_book_id INTEGER,
|
|
||||||
title VARCHAR(255) NOT NULL,
|
|
||||||
alternate_title VARCHAR(255),
|
|
||||||
lyrics TEXT NOT NULL,
|
|
||||||
verse_order VARCHAR(128),
|
|
||||||
copyright VARCHAR(255),
|
|
||||||
comments TEXT,
|
|
||||||
ccli_number VARCHAR(64),
|
|
||||||
song_number VARCHAR(64),
|
|
||||||
theme_name VARCHAR(128),
|
|
||||||
search_title VARCHAR(255) NOT NULL,
|
|
||||||
search_lyrics TEXT NOT NULL,
|
|
||||||
PRIMARY KEY (id),
|
|
||||||
FOREIGN KEY(song_book_id) REFERENCES song_books (id)
|
|
||||||
)"""),
|
|
||||||
(u'table "topics"', u"""CREATE TABLE topics (
|
|
||||||
id INTEGER NOT NULL,
|
|
||||||
name VARCHAR(128) NOT NULL,
|
|
||||||
PRIMARY KEY (id)
|
|
||||||
)"""),
|
|
||||||
(u'index "ix_songs_search_lyrics"',
|
|
||||||
u"""CREATE INDEX ix_songs_search_lyrics ON songs (search_lyrics)"""),
|
|
||||||
(u'index "ix_songs_search_title',
|
|
||||||
u"""CREATE INDEX ix_songs_search_title ON songs (search_title)"""),
|
|
||||||
(u'table "authors_songs"', u"""CREATE TABLE authors_songs (
|
|
||||||
author_id INTEGER NOT NULL,
|
|
||||||
song_id INTEGER NOT NULL,
|
|
||||||
PRIMARY KEY (author_id, song_id),
|
|
||||||
FOREIGN KEY(author_id) REFERENCES authors (id),
|
|
||||||
FOREIGN KEY(song_id) REFERENCES songs (id)
|
|
||||||
)"""),
|
|
||||||
(u'table "songs_topics"', u"""CREATE TABLE songs_topics (
|
|
||||||
song_id INTEGER NOT NULL,
|
|
||||||
topic_id INTEGER NOT NULL,
|
|
||||||
PRIMARY KEY (song_id, topic_id),
|
|
||||||
FOREIGN KEY(song_id) REFERENCES songs (id),
|
|
||||||
FOREIGN KEY(topic_id) REFERENCES topics (id)
|
|
||||||
)""")
|
|
||||||
]
|
|
||||||
|
|
||||||
def prepare_string(dirty):
|
|
||||||
return dirty_chars.sub(u'', dirty.replace(u'\r\n', u' ').replace(u'\n', u' '))
|
|
||||||
|
|
||||||
def display_sql(sql, params):
|
|
||||||
prepared_params = []
|
|
||||||
for param in params:
|
|
||||||
if isinstance(param, basestring):
|
|
||||||
prepared_params.append(u'"%s"' % param)
|
|
||||||
elif isinstance(param, (int, long)):
|
|
||||||
prepared_params.append(u'%d' % param)
|
|
||||||
elif isinstance(param, (float, complex)):
|
|
||||||
prepared_params.append(u'%f' % param)
|
|
||||||
else:
|
|
||||||
prepared_params.append(u'"%s"' % str(param))
|
|
||||||
for prepared_param in prepared_params:
|
|
||||||
sql = sql.replace(u'?', prepared_param, 1)
|
|
||||||
return sql
|
|
||||||
|
|
||||||
def create_database():
|
|
||||||
global new_cursor, create_statements
|
|
||||||
if debug or verbose:
|
|
||||||
print 'Creating new database:'
|
|
||||||
else:
|
|
||||||
print 'Creating new database...',
|
|
||||||
for statement_type, sql_create in create_statements:
|
|
||||||
if debug:
|
|
||||||
print '... ', sql_create.replace('\n', ' ').replace(' ', ' ')
|
|
||||||
elif verbose:
|
|
||||||
print '... creating %s...' % statement_type,
|
|
||||||
new_cursor.execute(sql_create)
|
|
||||||
if verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
if not verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
|
|
||||||
def import_songs():
|
|
||||||
global old_cursor, new_cursor, debug, verbose
|
|
||||||
if debug or verbose:
|
|
||||||
print 'Importing authors:'
|
|
||||||
else:
|
|
||||||
print 'Importing authors...',
|
|
||||||
if debug:
|
|
||||||
print '... SELECT authorid AS id, authorname AS displayname FROM authors'
|
|
||||||
elif verbose:
|
|
||||||
print '... fetching authors from old database...',
|
|
||||||
old_cursor.execute(u'SELECT authorid AS id, authorname AS displayname FROM authors')
|
|
||||||
rows = old_cursor.fetchall()
|
|
||||||
if not debug and verbose:
|
|
||||||
print 'done.'
|
|
||||||
author_map = {}
|
|
||||||
for row in rows:
|
|
||||||
display_name = unicode(row[1], u'cp1252')
|
|
||||||
names = display_name.split(u' ')
|
|
||||||
first_name = names[0]
|
|
||||||
last_name = u' '.join(names[1:])
|
|
||||||
if last_name is None:
|
|
||||||
last_name = u''
|
|
||||||
sql_insert = u'INSERT INTO authors '\
|
|
||||||
'(id, first_name, last_name, display_name) '\
|
|
||||||
'VALUES (NULL, ?, ?, ?)'
|
|
||||||
sql_params = (first_name, last_name, display_name)
|
|
||||||
if debug:
|
|
||||||
print '...', display_sql(sql_insert, sql_params)
|
|
||||||
elif verbose:
|
|
||||||
print '... importing "%s"' % display_name
|
|
||||||
new_cursor.execute(sql_insert, sql_params)
|
|
||||||
author_map[row[0]] = new_cursor.lastrowid
|
|
||||||
if debug:
|
|
||||||
print ' >>> authors.authorid =', row[0], 'authors.id =', author_map[row[0]]
|
|
||||||
if not verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
if debug or verbose:
|
|
||||||
print 'Importing songs:'
|
|
||||||
else:
|
|
||||||
print 'Importing songs...',
|
|
||||||
if debug:
|
|
||||||
print '... SELECT songid AS id, songtitle AS title, lyrics || \'\' AS lyrics, copyrightinfo AS copyright FROM songs...',
|
|
||||||
elif verbose:
|
|
||||||
print '... fetching songs from old database...',
|
|
||||||
old_cursor.execute(u'SELECT songid AS id, songtitle AS title, lyrics || \'\' AS lyrics, copyrightinfo AS copyright FROM songs')
|
|
||||||
rows = old_cursor.fetchall()
|
|
||||||
if debug or verbose:
|
|
||||||
print 'done.'
|
|
||||||
song_map = {}
|
|
||||||
xml_lyrics_template = u'<?xml version="1.0" encoding="utf-8"?><song version="1.0"><lyrics language="en">%s</lyrics></song>'
|
|
||||||
xml_verse_template = u'<verse label="%d" type="Verse"><![CDATA[%s]]></verse>'
|
|
||||||
for row in rows:
|
|
||||||
clean_title = unicode(row[1], u'cp1252')
|
|
||||||
clean_lyrics = unicode(row[2], u'cp1252').replace(u'\r\n', u'\n')
|
|
||||||
clean_copyright = unicode(row[3], u'cp1252')
|
|
||||||
verse_order = u''
|
|
||||||
text_lyrics = clean_lyrics.split(u'\n\n')
|
|
||||||
xml_verse = u''
|
|
||||||
verses = []
|
|
||||||
for line, verse in enumerate(text_lyrics):
|
|
||||||
if not verse:
|
|
||||||
continue
|
|
||||||
xml_verse += (xml_verse_template % (line + 1, verse))
|
|
||||||
verses.append(u'V%d' % (line + 1))
|
|
||||||
verse_order = u' '.join(verses)
|
|
||||||
xml_lyrics = xml_lyrics_template % xml_verse
|
|
||||||
search_title = prepare_string(clean_title)
|
|
||||||
search_lyrics = prepare_string(clean_lyrics)
|
|
||||||
sql_insert = u'INSERT INTO songs '\
|
|
||||||
'(id, song_book_id, title, lyrics, verse_order, copyright, search_title, search_lyrics) '\
|
|
||||||
'VALUES (NULL, 0, ?, ?, ?, ?, ?, ?)'
|
|
||||||
sql_params = (clean_title, xml_lyrics, verse_order, clean_copyright, search_title, search_lyrics)
|
|
||||||
if debug:
|
|
||||||
print '...', display_sql(sql_insert, (sql_params[0], u'%s...' % clean_lyrics[:7], sql_params[2], sql_params[3], sql_params[4], u'%s...' % search_lyrics[:7]))
|
|
||||||
elif verbose:
|
|
||||||
print '... importing "%s"' % clean_title
|
|
||||||
new_cursor.execute(sql_insert, sql_params)
|
|
||||||
song_map[row[0]] = new_cursor.lastrowid
|
|
||||||
if debug:
|
|
||||||
print ' >>> songs.songid =', row[0], 'songs.id =', song_map[row[0]]
|
|
||||||
if not verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
if debug or verbose:
|
|
||||||
print 'Importing song-to-author mapping:'
|
|
||||||
else:
|
|
||||||
print 'Importing song-to-author mapping...',
|
|
||||||
if debug:
|
|
||||||
print '... SELECT authorid AS author_id, songid AS song_id FROM songauthors'
|
|
||||||
elif verbose:
|
|
||||||
print '... fetching song-to-author mapping from old database...',
|
|
||||||
old_cursor.execute(u'SELECT authorid AS author_id, songid AS song_id FROM songauthors')
|
|
||||||
rows = old_cursor.fetchall()
|
|
||||||
if not debug and verbose:
|
|
||||||
print 'done.'
|
|
||||||
for row in rows:
|
|
||||||
sql_insert = u'INSERT INTO authors_songs '\
|
|
||||||
'(author_id, song_id) '\
|
|
||||||
'VALUES (?, ?)'
|
|
||||||
sql_params = (author_map[row[0]], song_map[row[1]])
|
|
||||||
if debug:
|
|
||||||
print '... ', display_sql(sql_insert, sql_params)
|
|
||||||
elif verbose:
|
|
||||||
print '... Author %d (was %d) => Song %d (was %d)'\
|
|
||||||
% (int(row[0]), author_map[row[0]],
|
|
||||||
int(row[1]), song_map[row[1]])
|
|
||||||
new_cursor.execute(sql_insert, sql_params)
|
|
||||||
if not verbose and not debug:
|
|
||||||
print 'done.'
|
|
||||||
|
|
||||||
def main(old_db, new_db):
|
|
||||||
global old_cursor, new_cursor, debug
|
|
||||||
old_connection = None
|
|
||||||
new_connection = None
|
|
||||||
try:
|
|
||||||
old_connection = sqlite.connect(old_db)
|
|
||||||
except:
|
|
||||||
if debug:
|
|
||||||
errormsg = '\n' + ''.join(get_traceback(sys.exc_info()[2]))\
|
|
||||||
+ str(sys.exc_info()[1])
|
|
||||||
else:
|
|
||||||
errormsg = sys.exc_info()[1]
|
|
||||||
print 'There was a problem connecting to the old database:', errormsg
|
|
||||||
return 1
|
|
||||||
try:
|
|
||||||
new_connection = sqlite3.connect(new_db)
|
|
||||||
except:
|
|
||||||
if debug:
|
|
||||||
errormsg = '\n' + ''.join(get_traceback(sys.exc_info()[2]))\
|
|
||||||
+ str(sys.exc_info()[1])
|
|
||||||
else:
|
|
||||||
errormsg = sys.exc_info()[1]
|
|
||||||
print 'There was a problem creating the new database:', errormsg
|
|
||||||
return 1
|
|
||||||
old_cursor = old_connection.cursor()
|
|
||||||
new_cursor = new_connection.cursor()
|
|
||||||
try:
|
|
||||||
create_database()
|
|
||||||
except:
|
|
||||||
if debug:
|
|
||||||
errormsg = '\n' + ''.join(get_traceback(sys.exc_info()[2]))\
|
|
||||||
+ str(sys.exc_info()[1])
|
|
||||||
else:
|
|
||||||
errormsg = sys.exc_info()[1]
|
|
||||||
print 'There was a problem creating the database:', errormsg
|
|
||||||
return 1
|
|
||||||
try:
|
|
||||||
import_songs()
|
|
||||||
new_connection.commit()
|
|
||||||
except:
|
|
||||||
new_connection.rollback()
|
|
||||||
if debug:
|
|
||||||
errormsg = '\n' + ''.join(get_traceback(sys.exc_info()[2]))\
|
|
||||||
+ str(sys.exc_info()[1])
|
|
||||||
else:
|
|
||||||
errormsg = sys.exc_info()[1]
|
|
||||||
print 'There was a problem importing songs:', errormsg
|
|
||||||
return 1
|
|
||||||
print 'Import complete.'
|
|
||||||
|
|
||||||
if __name__ == u'__main__':
|
|
||||||
option_parser = OptionParser(usage='Usage: %prog [options] OLDDATABASE NEWDATABASE')
|
|
||||||
option_parser.add_option('-o', '--overwrite', dest='overwrite', default=False,
|
|
||||||
action=u'store_true', help='Overwrite database file if it already exists.')
|
|
||||||
option_parser.add_option('-v', '--verbose', dest='verbose', default=False,
|
|
||||||
action=u'store_true', help='Outputs additional progress data.')
|
|
||||||
option_parser.add_option('-d', '--debug', dest='debug', default=False,
|
|
||||||
action=u'store_true', help='Outputs raw SQL statements (overrides verbose).')
|
|
||||||
options, arguments = option_parser.parse_args()
|
|
||||||
if len(arguments) < 2:
|
|
||||||
if len(arguments) == 0:
|
|
||||||
option_parser.error('Please specify an old database and a new database.')
|
|
||||||
else:
|
|
||||||
option_parser.error('Please specify a new database.')
|
|
||||||
old_db = os.path.abspath(arguments[0])
|
|
||||||
new_db = os.path.abspath(arguments[1])
|
|
||||||
if not os.path.isfile(old_db):
|
|
||||||
option_parser.error('Old database file ("%s") is not a file.' % old_db)
|
|
||||||
if not os.path.exists(old_db):
|
|
||||||
option_parser.error('Old database file ("%s") does not exist.' % old_db)
|
|
||||||
if os.path.exists(new_db):
|
|
||||||
if not options.overwrite:
|
|
||||||
option_parser.error('New database file ("%s") exists. If you want to overwrite it, specify the --overwrite option.' % new_db)
|
|
||||||
else:
|
|
||||||
if not os.path.isfile(new_db):
|
|
||||||
option_parser.error('New database file ("%s") is not a file.' % new_db)
|
|
||||||
os.unlink(new_db)
|
|
||||||
verbose = options.verbose
|
|
||||||
debug = options.debug
|
|
||||||
main(old_db, new_db)
|
|
||||||
Loading…
Reference in New Issue
Block a user