def restore(self):
"""restores a griffith compressed backup"""
filename = gutils.file_chooser(_("Restore Griffith backup"), \
action=gtk.FILE_CHOOSER_ACTION_OPEN, buttons= \
(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, \
gtk.STOCK_OPEN, gtk.RESPONSE_OK))
if filename[0]:
try:
zip = zipfile.ZipFile(filename[0], 'r')
except:
gutils.error(self, _("Can't read backup file"), self.widgets['window'])
return False
mypath = os.path.join(self.locations['posters'])
for each in zip.namelist():
file_to_restore = os.path.split(each)
if not os.path.isdir(file_to_restore[1]):
if file_to_restore[1] == '':
continue
if file_to_restore[1].endswith('.jpg'):
myfile = os.path.join(mypath,file_to_restore[1])
else:
myfile = os.path.join(self.locations['home'],file_to_restore[1])
outfile = open(myfile, 'wb')
outfile.write(zip.read(each))
outfile.flush()
outfile.close()
zip.close()
# restore config file
self.config = config.Config(file=os.path.join(self.locations['home'],'griffith.conf'))
filename = os.path.join(self.locations['home'], self.config["default_db"])
self.db.metadata.engine.dispose() # close DB
from sqlalchemy.orm import clear_mappers
clear_mappers()
# check if file needs conversion
if self.config['default_db'].lower().endswith('.gri'):
self.debug.show('Old database format detected. Converting...')
from dbupgrade import convert_from_old_db
from initialize import location_posters
if convert_from_old_db(self, filename, os.path.join(self.locations['home'], 'griffith.db')):
self.config.save()
location_posters(self.locations, self.config)
else:
print 'Cant convert old database, exiting.'
import sys
sys.exit(4)
self.db = sql.GriffithSQL(self.config, self.debug, self.locations['home'])
from initialize import dictionaries, people_treeview
dictionaries(self)
people_treeview(self)
# let's refresh the treeview
self.clear_details()
self.populate_treeview()
self.go_last()
self.treeview_clicked()
self.count_statusbar()
gutils.info(self, _("Backup restored"), self.widgets['window'])
def restore(self, merge=False):
"""
Merge database from:
* compressed backup (*.zip)
* SQLite2 *.gri file
* SQLite3 *.db file
"""
# let user select a backup file
filename = gutils.file_chooser(_("Restore Griffith backup"), \
action=gtk.FILE_CHOOSER_ACTION_OPEN, backup=True, \
buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_OPEN, gtk.RESPONSE_OK))
if not filename:
log.debug('no file selected')
return False
try:
tmp_db = None
tmp_dir = mkdtemp()
os.mkdir(os.path.join(tmp_dir, 'posters'))
print filename
if filename.lower().endswith('.zip'):
try:
zip_file = zipfile.ZipFile(filename, 'r')
except:
gutils.error(_("Can't read backup file"), self.widgets['window'])
return False
old_config_file = False
# unpack files to temporary directory
for file_path in zip_file.namelist():
file_name = os.path.split(file_path)[-1]
if not os.path.isdir(file_name):
if not file_name:
log.debug('skipping %s', file_path)
continue
if 'posters' in file_path:
new_file = os.path.join(tmp_dir, 'posters', file_name)
else:
new_file = os.path.join(tmp_dir, file_name)
if file_name.endswith('.conf'):
old_config_file = new_file
outfile = open(new_file, 'wb')
outfile.write(zip_file.read(file_path))
outfile.close()
zip_file.close()
# restore config file (new one will be created if old config format is detected)
tmp_config = config.Config(file=os.path.join(tmp_dir, 'griffith.cfg'))
if old_config_file:
log.info('Old config file detected. Please note that it will not be used.')
f = open(old_config_file, 'r')
old_config_raw_data = f.read()
f.close()
if old_config_raw_data.find('griffith.gri') >= -1:
tmp_config.set('file', 'griffith.gri', section='database')
# update filename var. to point to the unpacked database
filename = os.path.join(tmp_dir, tmp_config.get('name', 'griffith', section='database') + '.db')
else: # not a zip file? prepare a fake config file then
tmp_config = config.Config(file=os.path.join(tmp_dir, 'griffith.cfg'))
tmp_config.set('type', 'sqlite', section='database')
tmp_config.set('file', 'griffith.db', section='database')
# prepare temporary GriffithSQL instance
locations = {'home': tmp_dir}
# check if file needs conversion
if filename.lower().endswith('.gri'):
from dbupgrade import convert_from_old_db
tmp_db = convert_from_old_db(tmp_config, filename, os.path.join(tmp_dir, 'griffith.db'), locations)
if not tmp_db:
log.info("MERGE: Can't convert database, aborting.")
return False
else:
tmp_db = sql.GriffithSQL(tmp_config, tmp_dir, fallback=False)
if merge:
merge_db(tmp_db, self.db)
else:
self.db.session.rollback() # cancel all pending operations
copy_db(tmp_db.session.bind, self.db.session.bind)
# update old database section with current config values
# (important while restoring to external databases)
for key in ('name', 'passwd', 'host', 'user', 'file', 'type', 'port'):
tmp_config.set(key, self.config.get(key, section='database'), section='database')
tmp_config._file = self.config._file
self.config = tmp_config
self.config.save()
dictionaries(self)
people_treeview(self)
# let's refresh the treeview
self.clear_details()
self.populate_treeview()
#gutils.info(_("Databases merged!\n\nProcessed movies: %s\nMerged movies: %s"%(movies, merged)), self.widgets['window'])
gutils.info(_("Backup restored"), self.widgets['window'])
except:
log.exception('')
raise
finally:
# disposing the temporary db connection before rmtree and in finally block to avoid locked db file
if tmp_db:
tmp_db.dispose()
log.debug('temporary directory no logger needed, removing %s', tmp_dir)
rmtree(tmp_dir)
def merge(self): # FIXME
"""
Merge database from:
* compressed backup
* SQLite2 *.gri file
* SQLite3 *.db file
"""
filename = gutils.file_chooser(_("Restore Griffith backup"), \
action=gtk.FILE_CHOOSER_ACTION_OPEN, buttons= \
(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, \
gtk.STOCK_OPEN, gtk.RESPONSE_OK))[0]
if filename:
from tempfile import mkdtemp
from shutil import rmtree, move
#tmp_config={}
#tmp_config.get('type', 'sqlite', section='database')
if filename.lower().endswith('.zip'):
tmp_dir = mkdtemp()
try:
zip = zipfile.ZipFile(filename, 'r')
except:
gutils.error(self, _("Can't read backup file"), self.widgets['window'])
return False
for each in zip.namelist():
file_to_restore = os.path.split(each)
if not os.path.isdir(file_to_restore[1]):
myfile = os.path.join(tmp_dir, file_to_restore[1])
outfile = open(myfile, 'wb')
outfile.write(zip.read(each))
outfile.flush()
outfile.close()
# load stored database filename
tmp_config = config.Config(file=os.path.join(tmp_dir,'griffith.conf'))
filename = os.path.join(tmp_dir, tmp_config('file', 'griffith.db', section='database'))
zip.close()
# check if file needs conversion
if filename.lower().endswith(".gri"):
if os.path.isfile(filename) and open(filename).readline()[:47] == "** This file contains an SQLite 2.1 database **":
self.debug.show("MERGE: SQLite2 database format detected. Converting...")
if not self.db.convert_from_sqlite2(filename, os.path.join(tmp_dir, self.config.get('file', 'griffith.db', section='database'))):
self.debug.show("MERGE: Can't convert database, aborting.")
return False
tmp_dir, tmp_file = os.path.split(filename)
self.config.get('file', tmp_file, section='database')
tmp_db = sql.GriffithSQL(tmp_config, self.debug, tmp_dir)
merged=0
movies = tmp_db.Movie.count()
for movie in tmp_db.Movie.select():
if self.db.Movie.get_by(o_title=movie.o_title) is not None:
continue
t_movies = {}
for column in movie.mapper.c.keys():
t_movies[column] = eval("movie.%s"%column)
# replace number with new one
t_movies["number"] = gutils.find_next_available(self.db)
# don't restore volume/collection/tag/language/loan data (it's dangerous)
t_movies.pop('movie_id')
t_movies.pop('loaned')
t_movies.pop('volume_id')
t_movies.pop('collection_id')
if self.db.add_movie(t_movies):
print t_movies
if movie.image is not None:
dest_file = os.path.join(self.locations['posters'], movie.image+'.jpg')
if not os.path.isfile(dest_file):
src_file = os.path.join(tmp_dir, movie.image+'.jpg')
if os.path.isfile(src_file):
move(src_file, dest_file)
merged+=1
rmtree(tmp_dir)
from initialize import dictionaries, people_treeview
dictionaries(self)
people_treeview(self)
# let's refresh the treeview
self.clear_details()
self.populate_treeview(self.db.Movie.select())
self.total = self.db.Movie.count()
self.count_statusbar()
def save_preferences(self):
w = self.widgets['preferences']
c = self.config
global spell_support
was_false = notes_was_false = plot_was_false = 1
if c.get('gtkspell', False, section='spell') == True:
was_false = 0
if c.get('notes', False, section='spell') == True:
notes_was_false = 0
if c.get('plot', False, section='spell') == True:
plot_was_false = 0
# number
if w['view_number'].get_active():
c.set('number', 'True', section='mainlist')
else:
c.set('number', 'False', section='mainlist')
# image
if w['view_image'].get_active():
c.set('image', 'True', section='mainlist')
else:
c.set('image', 'False', section='mainlist')
# original title
if w['view_o_title'].get_active():
c.set('otitle', 'True', section='mainlist')
else:
c.set('otitle', 'False', section='mainlist')
# title
if w['view_title'].get_active():
c.set('title', 'True', section='mainlist')
else:
c.set('title', 'False', section='mainlist')
# director
if w['view_director'].get_active():
c.set('director', 'True', section='mainlist')
else:
c.set('director', 'False', section='mainlist')
# genre
if w['view_genre'].get_active():
c.set('genre', 'True', section='mainlist')
else:
c.set('genre', 'False', section='mainlist')
# seen
if w['view_seen'].get_active():
c.set('seen', 'True', section='mainlist')
else:
c.set('seen', 'False', section='mainlist')
# year
if w['view_year'].get_active():
c.set('year', 'True', section='mainlist')
else:
c.set('year', 'False', section='mainlist')
# runtime
if w['view_runtime'].get_active():
c.set('runtime', 'True', section='mainlist')
else:
c.set('runtime', 'False', section='mainlist')
# rating
if w['view_rating'].get_active():
c.set('rating', 'True', section='mainlist')
else:
c.set('rating', 'False', section='mainlist')
# sortby
if w['sortby'].get_active():
field = self.sort_criteria[w['sortby'].get_active()]
if field:
c.set('sortby', field, section='mainlist')
else:
c.set('sortby', 'number', section='mainlist')
c.set('sortby_reverse', w['sortby_reverse'].get_active(), section='mainlist')
c.set('limit', str(int(w['s_limit'].get_value())), section='mainlist')
# pdf font
if w['font'].get_filename():
c['font'] = w['font'].get_filename()
# spellchecker
if w['spellchecker'].get_active():
c.set('gtkspell', True, section='spell')
else:
c.set('gtkspell', False, section='spell')
if w['spell_notes'].get_active():
c.set('notes', True, section='spell')
else:
c.set('notes', False, section='spell')
if w['spell_plot'].get_active():
c.set('plot', True, section='spell')
else:
c.set('plot', False, section='spell')
# rating image
c['rating_image'] = str(w['rating_image'].get_active())
#defaults
media_id = self.media_ids[w['media'].get_active()]
if media_id is None:
media_id = 0
c.set('media', media_id, section='defaults')
vcodec_id = self.vcodecs_ids[w['vcodec'].get_active()]
if vcodec_id is None:
vcodec_id = 0
c.set('vcodec', vcodec_id, section='defaults')
c.set('condition', str(w['condition'].get_active()), section='defaults')
c.set('region', str(w['region'].get_active()), section='defaults')
c.set('layers', str(w['layers'].get_active()), section='defaults')
c.set('color', str(w['color'].get_active()), section='defaults')
# email reminder
if w['mail_use_auth'].get_active():
c.set('use_auth', True, section='mail')
else:
c.set('use_auth', False, section='mail')
if w['mail_use_tls'].get_active():
c.set('mail_use_tls', True, section='mail')
else:
c.set('mail_use_tls', False, section='mail')
c.set('smtp_server', w['mail_smtp_server'].get_text(), section='mail')
c.set('mail_smtp_port', w['mail_smtp_port'].get_text(), section='mail')
c.set('username', w['mail_username'].get_text(), section='mail')
c.set('password', w['mail_password'].get_text(), section='mail')
c.set('email', w['mail_email'].get_text(), section='mail')
# default movie plugin
if w['default_plugin'].get_active():
c['default_movie_plugin'] = \
gutils.on_combo_box_entry_changed(w['default_plugin'])
# search for:
c.set('s_classification', w['s_classification'].get_active(), section='add')
c.set('s_country', w['s_country'].get_active(), section='add')
c.set('s_director', w['s_director'].get_active(), section='add')
c.set('s_genre', w['s_genre'].get_active(), section='add')
c.set('s_image', w['s_image'].get_active(), section='add')
c.set('s_notes', w['s_notes'].get_active(), section='add')
c.set('s_o_site', w['s_o_site'].get_active(), section='add')
c.set('s_o_title', w['s_o_title'].get_active(), section='add')
c.set('s_plot', w['s_plot'].get_active(), section='add')
c.set('s_rating', w['s_rating'].get_active(), section='add')
c.set('s_runtime', w['s_runtime'].get_active(), section='add')
c.set('s_site', w['s_site'].get_active(), section='add')
c.set('s_studio', w['s_studio'].get_active(), section='add')
c.set('s_title', w['s_title'].get_active(), section='add')
c.set('s_trailer', w['s_trailer'].get_active(), section='add')
c.set('s_cast', w['s_cast'].get_active(), section='add')
c.set('s_year', w['s_year'].get_active(), section='add')
mcounter = 0
for p in self.plugins:
plugin_module = os.path.basename(p).replace('.py','')
plugin_name = plugin_module.replace('PluginMovie','')
if gutils.on_combo_box_entry_changed(w['default_plugin']) == plugin_name:
self.d_plugin = mcounter
mcounter = mcounter + 1
self.widgets['add']['source'].set_active(self.d_plugin)
if self.windows:
save_reader = ''
else:
save_reader = w['epdf_reader'].get_text()
c.set('lang', w['spell_lang'].get_text(), section='spell')
c['pdf_reader'] = save_reader
c.set('amazon_locale', w['amazon_locale'].get_active(), section='add')
if spell_support:
if c.get('gtkspell', False, section='spell') == False and not was_false:
self.notes_spell.detach()
self.plot_spell.detach()
elif c.get('gtkspell', False, section='spell') == True and was_false:
initialize.initialize_gtkspell(self)
else:
pass
if c.get('gtkspell', False, section='spell') == True:
if c.get('plot', True, section='spell') == False and not plot_was_false:
self.plot_spell.detach()
elif c.get('plot', True, section='spell') == True and plot_was_false:
self.plot_spell = gtkspell.Spell(self.widgets['add']['plot'])
self.plot_spell.set_language(c.get('lang', 'en', section='spell'))
else:
pass
if c.get('notes', True, section='spell') == False and not notes_was_false:
self.notes_spell.detach()
elif c.get('notes', True, section='spell') == True and notes_was_false:
self.notes_spell = gtkspell.Spell(self.widgets['add']['notes'])
self.notes_spell.set_language(c.get('lang', 'en', section='spell'))
else:
pass
self.pdf_reader = save_reader
# database
old = c.toDict(section='database')
c.set('host', w['db_host'].get_text(), section='database')
c.set('port', int(w['db_port'].get_value()), section='database')
c.set('name', w['db_name'].get_text(), section='database')
c.set('user', w['db_user'].get_text(), section='database')
c.set('passwd', w['db_passwd'].get_text(), section='database')
db_type = int(w['db_type'].get_active())
if db_type == 1:
c.set('type', 'postgres', section='database')
elif db_type == 2:
c.set('type', 'mysql', section='database')
elif db_type == 3:
c.set('type', 'mssql', section='database')
else:
c.set('type', 'sqlite', section='database')
if old['type'] != c.get('type', section='database') or (old['type']!='sqlite' and (\
old['host'] != c.get('host', section='database') or \
old['port'] != c.get('port', section='database') or \
old['user'] != c.get('user', section='database') or \
old['passwd'] != c.get('passwd', section='database'))) or \
old['name'] != c.get('name', section='database'):
self.debug.show('DATABASE: connecting to new db server...')
# new database connection
import sql
self.initialized = False
self.db.metadata.clear()
from sqlalchemy.orm import clear_mappers
from sqlalchemy.exceptions import InvalidRequestError
clear_mappers()
try:
self.db = sql.GriffithSQL(c, self.debug, self.locations['home'])
except InvalidRequestError, e:
self.debug.show(str(e))
c.set('type', 'sqlite', section='database')
w['db_type'].set_active(0)
self.db = sql.GriffithSQL(c, self.debug, self.locations['home'])
self.debug.show("New database Engine: %s" % self.db.metadata.engine.name)
# initialize new database
self.total = int(self.db.Movie.count())
self.count_statusbar()
from initialize import dictionaries, people_treeview, location_posters
c['posters'] = None # force update
location_posters(self.locations, self.config)
dictionaries(self)
people_treeview(self, False)
self.initialized = True
def restore(self, merge=False):
"""
Merge database from:
* compressed backup (*.zip)
* SQLite2 *.gri file
* SQLite3 *.db file
"""
# let user select a backup file
filename, path = gutils.file_chooser(_("Restore Griffith backup"), \
action=gtk.FILE_CHOOSER_ACTION_OPEN, backup=True, \
buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_OPEN, gtk.RESPONSE_OK))
if not filename:
log.debug('no file selected')
return False
try:
tmp_db = None
tmp_dir = mkdtemp()
os.mkdir(os.path.join(tmp_dir, 'posters'))
print filename
if filename.lower().endswith('.zip'):
try:
zip_file = zipfile.ZipFile(filename, 'r')
except:
gutils.error(_("Can't read backup file"), self.widgets['window'])
return False
old_config_file = False
# unpack files to temporary directory
for file_path in zip_file.namelist():
file_name = os.path.split(file_path)[-1]
if not os.path.isdir(file_name):
if not file_name:
log.debug('skipping %s', file_path)
continue
if 'posters' in file_path:
new_file = os.path.join(tmp_dir, 'posters', file_name)
else:
new_file = os.path.join(tmp_dir, file_name)
if file_name.endswith('.conf'):
old_config_file = new_file
outfile = open(new_file, 'wb')
outfile.write(zip_file.read(file_path))
outfile.close()
zip_file.close()
# restore config file (new one will be created if old config format is detected)
tmp_config = config.Config(file=os.path.join(tmp_dir, 'griffith.cfg'))
if old_config_file:
log.info('Old config file detected. Please note that it will not be used.')
f = open(old_config_file, 'r')
old_config_raw_data = f.read()
f.close()
if old_config_raw_data.find('griffith.gri') >= -1:
tmp_config.set('file', 'griffith.gri', section='database')
# update filename var. to point to the unpacked database
filename = os.path.join(tmp_dir, tmp_config.get('name', 'griffith', section='database') + '.db')
else: # not a zip file? prepare a fake config file then
tmp_config = config.Config(file=os.path.join(tmp_dir, 'griffith.cfg'))
tmp_config.set('type', 'sqlite', section='database')
tmp_config.set('file', 'griffith.db', section='database')
# prepare temporary GriffithSQL instance
locations = {'home': tmp_dir}
# check if file needs conversion
if filename.lower().endswith('.gri'):
from dbupgrade import convert_from_old_db
tmp_db = convert_from_old_db(tmp_config, filename, os.path.join(tmp_dir, 'griffith.db'), locations)
if not tmp_db:
log.info("MERGE: Can't convert database, aborting.")
return False
else:
tmp_db = sql.GriffithSQL(tmp_config, tmp_dir, fallback=False)
if merge:
merge_db(tmp_db, self.db)
else:
self.db.session.rollback() # cancel all pending operations
copy_db(tmp_db.session.bind, self.db.session.bind)
# update old database section with current config values
# (important while restoring to external databases)
for key in ('name', 'passwd', 'host', 'user', 'file', 'type', 'port'):
tmp_config.set(key, self.config.get(key, section='database'), section='database')
tmp_config._file = self.config._file
self.config = tmp_config
self.config.save()
dictionaries(self)
people_treeview(self)
# let's refresh the treeview
self.clear_details()
self.populate_treeview()
#gutils.info(_("Databases merged!\n\nProcessed movies: %s\nMerged movies: %s"%(movies, merged)), self.widgets['window'])
gutils.info(_("Backup restored"), self.widgets['window'])
except:
log.exception('')
raise
finally:
# disposing the temporary db connection before rmtree and in finally block to avoid locked db file
if tmp_db:
tmp_db.dispose()
log.debug('temporary directory no logger needed, removing %s', tmp_dir)
rmtree(tmp_dir)
def save_preferences(self):
w = self.widgets['preferences']
c = self.config
global spell_support
was_false = notes_was_false = plot_was_false = 1
if c.get('gtkspell', False, section='spell') == True:
was_false = 0
if c.get('notes', False, section='spell') == True:
notes_was_false = 0
if c.get('plot', False, section='spell') == True:
plot_was_false = 0
# number
if w['view_number'].get_active():
c.set('number', 'True', section='mainlist')
else:
c.set('number', 'False', section='mainlist')
# image
if w['view_image'].get_active():
c.set('image', 'True', section='mainlist')
else:
c.set('image', 'False', section='mainlist')
# original title
if w['view_o_title'].get_active():
c.set('otitle', 'True', section='mainlist')
else:
c.set('otitle', 'False', section='mainlist')
# title
if w['view_title'].get_active():
c.set('title', 'True', section='mainlist')
else:
c.set('title', 'False', section='mainlist')
# director
if w['view_director'].get_active():
c.set('director', 'True', section='mainlist')
else:
c.set('director', 'False', section='mainlist')
# genre
if w['view_genre'].get_active():
c.set('genre', 'True', section='mainlist')
else:
c.set('genre', 'False', section='mainlist')
# seen
if w['view_seen'].get_active():
c.set('seen', 'True', section='mainlist')
else:
c.set('seen', 'False', section='mainlist')
# year
if w['view_year'].get_active():
c.set('year', 'True', section='mainlist')
else:
c.set('year', 'False', section='mainlist')
# runtime
if w['view_runtime'].get_active():
c.set('runtime', 'True', section='mainlist')
else:
c.set('runtime', 'False', section='mainlist')
# rating
if w['view_rating'].get_active():
c.set('rating', 'True', section='mainlist')
else:
c.set('rating', 'False', section='mainlist')
# created
if w['view_created'].get_active():
c.set('created', 'True', section='mainlist')
else:
c.set('created', 'False', section='mainlist')
# updated
if w['view_updated'].get_active():
c.set('updated', 'True', section='mainlist')
else:
c.set('updated', 'False', section='mainlist')
# sortby
if w['sortby'].get_active():
field = self.sort_criteria[w['sortby'].get_active()]
if field:
c.set('sortby', field, section='mainlist')
else:
c.set('sortby', 'number', section='mainlist')
c.set('sortby_reverse', w['sortby_reverse'].get_active(), section='mainlist')
c.set('limit', str(int(w['s_limit'].get_value())), section='mainlist')
# pdf font
if w['font'].get_filename():
c['font'] = w['font'].get_filename()
c['font_size'] = int(w['font_size'].get_value())
# pdf elements
pdf_elements = ''
for child in w['pdf_elements_table']:
if child.get_active():
pdf_elements = pdf_elements + child.get_name()[4:] + ','
if pdf_elements:
c.set('pdf_elements', pdf_elements[:-1])
else:
c.set('pdf_elements', pdf_elements)
# spellchecker
if w['spellchecker'].get_active():
c.set('gtkspell', True, section='spell')
else:
c.set('gtkspell', False, section='spell')
if w['spell_notes'].get_active():
c.set('notes', True, section='spell')
else:
c.set('notes', False, section='spell')
if w['spell_plot'].get_active():
c.set('plot', True, section='spell')
else:
c.set('plot', False, section='spell')
# rating image
c['rating_image'] = str(w['rating_image'].get_active())
#defaults
media_id = self.media_ids[w['media'].get_active()]
if media_id is None:
media_id = 0
c.set('media', media_id, section='defaults')
vcodec_id = self.vcodecs_ids[w['vcodec'].get_active()]
if vcodec_id is None:
vcodec_id = 0
c.set('vcodec', vcodec_id, section='defaults')
c.set('condition', str(w['condition'].get_active()), section='defaults')
c.set('region', str(w['region'].get_active()), section='defaults')
c.set('layers', str(w['layers'].get_active()), section='defaults')
c.set('color', str(w['color'].get_active()), section='defaults')
# email reminder
if w['mail_use_auth'].get_active():
c.set('use_auth', True, section='mail')
else:
c.set('use_auth', False, section='mail')
if w['mail_use_tls'].get_active():
c.set('mail_use_tls', True, section='mail')
else:
c.set('mail_use_tls', False, section='mail')
c.set('smtp_server', w['mail_smtp_server'].get_text(), section='mail')
c.set('mail_smtp_port', w['mail_smtp_port'].get_text(), section='mail')
c.set('username', w['mail_username'].get_text(), section='mail')
c.set('password', w['mail_password'].get_text(), section='mail')
c.set('email', w['mail_email'].get_text(), section='mail')
# default movie plugin
if w['default_plugin'].get_active():
c['default_movie_plugin'] = \
gutils.on_combo_box_entry_changed(w['default_plugin'])
# search for:
c.set('s_classification', w['s_classification'].get_active(), section='add')
c.set('s_country', w['s_country'].get_active(), section='add')
c.set('s_director', w['s_director'].get_active(), section='add')
c.set('s_genre', w['s_genre'].get_active(), section='add')
c.set('s_image', w['s_image'].get_active(), section='add')
c.set('s_notes', w['s_notes'].get_active(), section='add')
c.set('s_o_site', w['s_o_site'].get_active(), section='add')
c.set('s_o_title', w['s_o_title'].get_active(), section='add')
c.set('s_plot', w['s_plot'].get_active(), section='add')
c.set('s_rating', w['s_rating'].get_active(), section='add')
c.set('s_runtime', w['s_runtime'].get_active(), section='add')
c.set('s_site', w['s_site'].get_active(), section='add')
c.set('s_studio', w['s_studio'].get_active(), section='add')
c.set('s_title', w['s_title'].get_active(), section='add')
c.set('s_trailer', w['s_trailer'].get_active(), section='add')
c.set('s_cast', w['s_cast'].get_active(), section='add')
c.set('s_year', w['s_year'].get_active(), section='add')
c.set('s_screenplay', w['s_screenplay'].get_active(), section='add')
c.set('s_cameraman', w['s_cameraman'].get_active(), section='add')
c.set('s_resolution', w['s_resolution'].get_active(), section='add')
c.set('s_barcode', w['s_barcode'].get_active(), section='add')
mcounter = 0
for p in self.plugins:
plugin_module = os.path.basename(p).replace('.py','')
plugin_name = plugin_module.replace('PluginMovie','')
if gutils.on_combo_box_entry_changed(w['default_plugin']) == plugin_name:
break
mcounter = mcounter + 1
self.widgets['add']['source'].set_active(mcounter)
save_reader = w['epdf_reader'].get_text()
c.set('lang', w['spell_lang'].get_text(), section='spell')
c['pdf_reader'] = save_reader
if spell_support:
if c.get('gtkspell', False, section='spell') == False and not was_false:
self.notes_spell.detach()
self.plot_spell.detach()
elif c.get('gtkspell', False, section='spell') == True and was_false:
initialize.spellcheck(self)
else:
pass
if c.get('gtkspell', False, section='spell') == True:
if c.get('plot', True, section='spell') == False and not plot_was_false:
self.plot_spell.detach()
elif c.get('plot', True, section='spell') == True and plot_was_false:
self.plot_spell = gtkspell.Spell(self.widgets['add']['plot'])
self.plot_spell.set_language(c.get('lang', 'en', section='spell'))
else:
pass
if c.get('notes', True, section='spell') == False and not notes_was_false:
self.notes_spell.detach()
elif c.get('notes', True, section='spell') == True and notes_was_false:
self.notes_spell = gtkspell.Spell(self.widgets['add']['notes'])
self.notes_spell.set_language(c.get('lang', 'en', section='spell'))
else:
pass
self.pdf_reader = save_reader
# extensions settings
for ext_name in plugins.extensions.by_name:
preferenceswidgets = plugins.extensions.by_name[ext_name].preferenceswidgets
for prefname in preferenceswidgets:
widget = preferenceswidgets[prefname]
if isinstance(widget, gtk.CheckButton):
value = widget.get_active()
elif isinstance(widget, gtk.Entry):
value = widget.get_text()
elif isinstance(widget, gtk.ComboBox):
iter = widget.get_active_iter()
if iter:
value = widget.get_model().get_value(iter, 1)
else:
log.error('widget type not supported %s', type(widget))
continue
c.set("%s_%s" % (ext_name, prefname), value, section='extensions')
# database
old = c.to_dict(section='database')
c.set('host', w['db_host'].get_text(), section='database')
c.set('port', int(w['db_port'].get_value()), section='database')
c.set('name', w['db_name'].get_text(), section='database')
c.set('user', w['db_user'].get_text(), section='database')
c.set('passwd', w['db_passwd'].get_text(), section='database')
db_type = int(w['db_type'].get_active())
if db_type == 1:
c.set('type', 'postgres', section='database')
elif db_type == 2:
c.set('type', 'mysql', section='database')
elif db_type == 3:
c.set('type', 'mssql', section='database')
else:
c.set('type', 'sqlite', section='database')
if old['type'] != c.get('type', section='database') or (old['type']!='sqlite' and (\
old['host'] != c.get('host', section='database') or \
old['port'] != c.get('port', section='database') or \
old['user'] != c.get('user', section='database') or \
old['passwd'] != c.get('passwd', section='database'))) or \
old['name'] != c.get('name', section='database'):
log.info('DATABASE: connecting to new db server...')
import sql
from sqlalchemy.exceptions import InvalidRequestError
from initialize import dictionaries, people_treeview
# new database connection
self.initialized = False
if c.has_key('posters'):
c['posters'] = None # force update
try:
self.db.dispose()
self.db = sql.GriffithSQL(c, self.locations['home'], fallback=True)
except InvalidRequestError, e:
log.exception('')
c.set('type', 'sqlite', section='database')
w['db_type'].set_active(0)
self.db = sql.GriffithSQL(c, self.locations['home'])
log.info("New database Engine: %s" % self.db.session.bind.engine.name)
# initialize new database
self.total = int(self.db.session.query(db.Movie).count())
self.count_statusbar()
dictionaries(self)
people_treeview(self, False)
self.initialized = True
def restore(self):
"""restores a griffith compressed backup"""
filename = gutils.file_chooser(_("Restore Griffith backup"), \
action=gtk.FILE_CHOOSER_ACTION_OPEN, buttons= \
(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, \
gtk.STOCK_OPEN, gtk.RESPONSE_OK))
if filename[0]:
try:
zip = zipfile.ZipFile(filename[0], 'r')
except:
gutils.error(self, _("Can't read backup file"), self.widgets['window'])
return False
mypath = os.path.join(self.locations['posters'])
old_config_file = False
for each in zip.namelist():
file_to_restore = os.path.split(each)
if not os.path.isdir(file_to_restore[1]):
if file_to_restore[1] == '':
continue
if file_to_restore[1].endswith('.jpg'):
myfile = os.path.join(mypath,file_to_restore[1])
else:
myfile = os.path.join(self.locations['home'],file_to_restore[1])
if file_to_restore[1].endswith('.conf'):
old_config_file = myfile
outfile = open(myfile, 'wb')
outfile.write(zip.read(each))
outfile.flush()
outfile.close()
zip.close()
# restore config file
self.config = config.Config(file=os.path.join(self.locations['home'],'griffith.cfg'))
if old_config_file:
log.info('Old config file detected. Please note that it will not be used.')
f = open(old_config_file, 'r')
old_config_raw_data = f.read()
f.close()
if old_config_raw_data.find('griffith.gri') >= -1:
self.config.set('file', 'griffith.gri', section='database')
filename = os.path.join(self.locations['home'], self.config.get('name', 'griffith', section='database') + '.db')
self.db.session.bind.engine.dispose() # close DB
# check if file needs conversion
if self.config.get('file', 'griffith.db', section='database').lower().endswith('.gri'):
log.info('Old database format detected. Converting...')
from dbupgrade import convert_from_old_db
if convert_from_old_db(self, filename, os.path.join(self.locations['home'], 'griffith.db')):
self.config.save()
else:
log.error('Cant convert old database, exiting.')
import sys
sys.exit(4)
self.db = sql.GriffithSQL(self.config, self.locations['home'], self.locations)
from initialize import dictionaries, people_treeview
dictionaries(self)
people_treeview(self)
# let's refresh the treeview
self.clear_details()
self.populate_treeview()
gutils.info(_("Backup restored"), self.widgets['window'])
def save_preferences(self):
w = self.widgets['preferences']
c = self.config
global spell_support
was_false = notes_was_false = plot_was_false = 1
if c.get('use_gtkspell', 'False') == 'True':
was_false = 0
if c.get('spell_notes', 'False') == 'True':
notes_was_false = 0
if c.get('spell_plot', 'False') == 'True':
plot_was_false = 0
# image
if w['view_image'].get_active():
c['view_image'] = 'True'
else:
c['view_image'] = 'False'
# original title
if w['view_o_title'].get_active():
c['view_otitle'] = 'True'
else:
c['view_otitle'] = 'False'
# title
if w['view_title'].get_active():
c['view_title'] = 'True'
else:
c['view_title'] = 'False'
# director
if w['view_director'].get_active():
c['view_director'] = 'True'
else:
c['view_director'] = 'False'
# sortby
if w['sortby'].get_active():
field = self.sort_criteria[w['sortby'].get_active()]
if field:
c['sortby'] = field
else:
c['sortby'] = 'number'
c['sortby_reverse'] = w['sortby_reverse'].get_active()
# pdf font
if w['font'].get_filename():
c['font'] = w['font'].get_filename()
# spellchecker
if w['spellchecker'].get_active():
c['use_gtkspell'] = 'True'
else:
c['use_gtkspell'] = 'False'
if w['spell_notes'].get_active():
c['spell_notes'] = 'True'
else:
c['spell_notes'] = 'False'
if w['spell_plot'].get_active():
c['spell_plot'] = 'True'
else:
c['spell_plot'] = 'False'
# rating image
c['rating_image'] = str(w['rating_image'].get_active())
#defaults
c['media'] = self.media_ids[w['media'].get_active()]
c['vcodec'] = self.vcodecs_ids[w['vcodec'].get_active()]
c['condition'] = str(w['condition'].get_active())
c['region'] = str(w['region'].get_active())
c['layers'] = str(w['layers'].get_active())
c['color'] = str(w['color'].get_active())
# email reminder
if w['mail_use_auth'].get_active():
c['mail_use_auth'] = 'True'
else:
c['mail_use_auth'] = 'False'
c['mail_smtp_server'] = w['mail_smtp_server'].get_text()
c['mail_username'] = w['mail_username'].get_text()
c['mail_password'] = w['mail_password'].get_text()
c['mail_email'] = w['mail_email'].get_text()
# default movie plugin
if w['default_plugin'].get_active():
c['default_movie_plugin'] = \
gutils.on_combo_box_entry_changed(w['default_plugin'])
# search for:
c['s_classification'] = w['s_classification'].get_active()
c['s_country'] = w['s_country'].get_active()
c['s_director'] = w['s_director'].get_active()
c['s_genre'] = w['s_genre'].get_active()
c['s_image'] = w['s_image'].get_active()
c['s_notes'] = w['s_notes'].get_active()
c['s_o_site'] = w['s_o_site'].get_active()
c['s_o_title'] = w['s_o_title'].get_active()
c['s_plot'] = w['s_plot'].get_active()
c['s_rating'] = w['s_rating'].get_active()
c['s_runtime'] = w['s_runtime'].get_active()
c['s_site'] = w['s_site'].get_active()
c['s_studio'] = w['s_studio'].get_active()
c['s_title'] = w['s_title'].get_active()
c['s_trailer'] = w['s_trailer'].get_active()
c['s_cast'] = w['s_cast'].get_active()
c['s_year'] = w['s_year'].get_active()
mcounter = 0
for p in self.plugins:
plugin_module = os.path.basename(p).replace('.py','')
plugin_name = plugin_module.replace('PluginMovie','')
if gutils.on_combo_box_entry_changed(w['default_plugin']) == plugin_name:
self.d_plugin = mcounter
mcounter = mcounter + 1
self.widgets['add']['source'].set_active(self.d_plugin)
if self.windows:
save_reader = ''
else:
save_reader = w['epdf_reader'].get_text()
c['spell_lang'] = w['spell_lang'].get_text()
c['pdf_reader'] = save_reader
c['amazon_locale'] = w['amazon_locale'].get_active()
if spell_support:
if c.get('use_gtkspell', 'False') == 'False' and not was_false:
self.notes_spell.detach()
self.plot_spell.detach()
elif c.get('use_gtkspell', 'False') == 'True' and was_false:
initialize.initialize_gtkspell(self)
else:
pass
if c.get('use_gtkspell', 'False') == 'True':
if c.get('spell_plot', 'True') == 'False' and not plot_was_false:
self.plot_spell.detach()
elif c.get('spell_plot', 'True') == 'True' and plot_was_false:
self.plot_spell = gtkspell.Spell(self.widgets['add']['plot'])
self.plot_spell.set_language(c.get('spell_lang', 'en'))
else:
pass
if c.get('spell_notes', 'True') == 'False' and not notes_was_false:
self.notes_spell.detach()
elif c.get('spell_notes', 'True') == 'True' and notes_was_false:
self.notes_spell = gtkspell.Spell(self.widgets['add']['notes'])
self.notes_spell.set_language(c.get('spell_lang', 'en'))
else:
pass
self.pdf_reader = save_reader
# database
old = {
'db_type': c['db_type'],
'db_host': c['db_host'],
'db_port': c['db_port'],
'db_name': c['db_name'],
'db_user': c['db_user'],
'db_passwd': c['db_passwd'],
}
c['db_host'] = w['db_host'].get_text()
c['db_port'] = int(w['db_port'].get_value())
c['db_name'] = w['db_name'].get_text()
c['db_user'] = w['db_user'].get_text()
c['db_passwd'] = w['db_passwd'].get_text()
db_type = int(w['db_type'].get_active())
if db_type == 1:
c['db_type'] = 'postgres'
elif db_type == 2:
c['db_type'] = 'mysql'
else:
c['db_type'] = 'sqlite'
if old['db_type'] != c['db_type'] or (old['db_type']!='sqlite' and (\
old['db_host'] != c['db_host'] or \
old['db_port'] != c['db_port'] or \
old['db_name'] != c['db_name'] or \
old['db_user'] != c['db_user'] or \
old['db_passwd'] != c['db_passwd'])):
self.debug.show('DATABASE: connecting to new db server...')
# new database connection
import sql
self.initialized = False
self.db.metadata.clear()
from sqlalchemy.orm import clear_mappers
clear_mappers()
self.db = sql.GriffithSQL(c, self.debug, self.locations['home'])
self.debug.show("New database Engine: %s" % self.db.metadata.engine.name)
# initialize new database
self.total = int(self.db.Movie.count())
self.count_statusbar()
from initialize import dictionaries, people_treeview, location_posters
c['posters'] = None # force update
location_posters(self.locations, self.config)
dictionaries(self)
people_treeview(self, False)
self.initialized = True
self.clear_details()
self.populate_treeview()
self.go_last()
c.save()
def save_preferences(self):
w = self.widgets['preferences']
c = self.config
global spell_support
was_false = notes_was_false = plot_was_false = 1
if c.get('gtkspell', False, section='spell') == True:
was_false = 0
if c.get('notes', False, section='spell') == True:
notes_was_false = 0
if c.get('plot', False, section='spell') == True:
plot_was_false = 0
# number
if w['view_number'].get_active():
c.set('number', 'True', section='mainlist')
else:
c.set('number', 'False', section='mainlist')
# image
if w['view_image'].get_active():
c.set('image', 'True', section='mainlist')
else:
c.set('image', 'False', section='mainlist')
# original title
if w['view_o_title'].get_active():
c.set('otitle', 'True', section='mainlist')
else:
c.set('otitle', 'False', section='mainlist')
# title
if w['view_title'].get_active():
c.set('title', 'True', section='mainlist')
else:
c.set('title', 'False', section='mainlist')
# director
if w['view_director'].get_active():
c.set('director', 'True', section='mainlist')
else:
c.set('director', 'False', section='mainlist')
# genre
if w['view_genre'].get_active():
c.set('genre', 'True', section='mainlist')
else:
c.set('genre', 'False', section='mainlist')
# seen
if w['view_seen'].get_active():
c.set('seen', 'True', section='mainlist')
else:
c.set('seen', 'False', section='mainlist')
# year
if w['view_year'].get_active():
c.set('year', 'True', section='mainlist')
else:
c.set('year', 'False', section='mainlist')
# runtime
if w['view_runtime'].get_active():
c.set('runtime', 'True', section='mainlist')
else:
c.set('runtime', 'False', section='mainlist')
# rating
if w['view_rating'].get_active():
c.set('rating', 'True', section='mainlist')
else:
c.set('rating', 'False', section='mainlist')
# created
if w['view_created'].get_active():
c.set('created', 'True', section='mainlist')
else:
c.set('created', 'False', section='mainlist')
# updated
if w['view_updated'].get_active():
c.set('updated', 'True', section='mainlist')
else:
c.set('updated', 'False', section='mainlist')
# sortby
if w['sortby'].get_active():
field = self.sort_criteria[w['sortby'].get_active()]
if field:
c.set('sortby', field, section='mainlist')
else:
c.set('sortby', 'number', section='mainlist')
c.set('sortby_reverse',
w['sortby_reverse'].get_active(),
section='mainlist')
c.set('limit', str(int(w['s_limit'].get_value())), section='mainlist')
# pdf font
if w['font'].get_filename():
c['font'] = w['font'].get_filename()
c['font_size'] = int(w['font_size'].get_value())
# pdf elements
pdf_elements = ''
for child in w['pdf_elements_table']:
if child.get_active():
pdf_elements = pdf_elements + child.get_name()[4:] + ','
if pdf_elements:
c.set('pdf_elements', pdf_elements[:-1])
else:
c.set('pdf_elements', pdf_elements)
# spellchecker
if w['spellchecker'].get_active():
c.set('gtkspell', True, section='spell')
else:
c.set('gtkspell', False, section='spell')
if w['spell_notes'].get_active():
c.set('notes', True, section='spell')
else:
c.set('notes', False, section='spell')
if w['spell_plot'].get_active():
c.set('plot', True, section='spell')
else:
c.set('plot', False, section='spell')
# rating image
c['rating_image'] = str(w['rating_image'].get_active())
#defaults
media_id = self.media_ids[w['media'].get_active()]
if media_id is None:
media_id = 0
c.set('media', media_id, section='defaults')
vcodec_id = self.vcodecs_ids[w['vcodec'].get_active()]
if vcodec_id is None:
vcodec_id = 0
c.set('vcodec', vcodec_id, section='defaults')
c.set('condition', str(w['condition'].get_active()), section='defaults')
c.set('region', str(w['region'].get_active()), section='defaults')
c.set('layers', str(w['layers'].get_active()), section='defaults')
c.set('color', str(w['color'].get_active()), section='defaults')
c.set('seen', str(w['seen'].get_active()), section='defaults')
# email reminder
if w['mail_use_auth'].get_active():
c.set('use_auth', True, section='mail')
else:
c.set('use_auth', False, section='mail')
if w['mail_use_tls'].get_active():
c.set('mail_use_tls', True, section='mail')
else:
c.set('mail_use_tls', False, section='mail')
c.set('smtp_server', w['mail_smtp_server'].get_text(), section='mail')
c.set('mail_smtp_port', w['mail_smtp_port'].get_text(), section='mail')
c.set('username', w['mail_username'].get_text(), section='mail')
c.set('password', w['mail_password'].get_text(), section='mail')
c.set('email', w['mail_email'].get_text(), section='mail')
# default movie plugin
if w['default_plugin'].get_active():
c['default_movie_plugin'] = \
gutils.on_combo_box_entry_changed(w['default_plugin'])
# search for:
c.set('s_classification',
w['s_classification'].get_active(),
section='add')
c.set('s_country', w['s_country'].get_active(), section='add')
c.set('s_director', w['s_director'].get_active(), section='add')
c.set('s_genre', w['s_genre'].get_active(), section='add')
c.set('s_image', w['s_image'].get_active(), section='add')
c.set('s_notes', w['s_notes'].get_active(), section='add')
c.set('s_o_site', w['s_o_site'].get_active(), section='add')
c.set('s_o_title', w['s_o_title'].get_active(), section='add')
c.set('s_plot', w['s_plot'].get_active(), section='add')
c.set('s_rating', w['s_rating'].get_active(), section='add')
c.set('s_runtime', w['s_runtime'].get_active(), section='add')
c.set('s_site', w['s_site'].get_active(), section='add')
c.set('s_studio', w['s_studio'].get_active(), section='add')
c.set('s_title', w['s_title'].get_active(), section='add')
c.set('s_trailer', w['s_trailer'].get_active(), section='add')
c.set('s_cast', w['s_cast'].get_active(), section='add')
c.set('s_year', w['s_year'].get_active(), section='add')
c.set('s_screenplay', w['s_screenplay'].get_active(), section='add')
c.set('s_cameraman', w['s_cameraman'].get_active(), section='add')
c.set('s_resolution', w['s_resolution'].get_active(), section='add')
c.set('s_barcode', w['s_barcode'].get_active(), section='add')
mcounter = 0
for p in self.plugins:
plugin_module = os.path.basename(p).replace('.py', '')
plugin_name = plugin_module.replace('PluginMovie', '')
if gutils.on_combo_box_entry_changed(
w['default_plugin']) == plugin_name:
break
mcounter = mcounter + 1
self.widgets['add']['source'].set_active(mcounter)
save_reader = w['epdf_reader'].get_text()
c.set('lang', w['spell_lang'].get_text(), section='spell')
c['pdf_reader'] = save_reader
if spell_support:
if c.get('gtkspell', False,
section='spell') == False and not was_false:
self.notes_spell.detach()
self.plot_spell.detach()
elif c.get('gtkspell', False, section='spell') == True and was_false:
initialize.spellcheck(self)
else:
pass
if c.get('gtkspell', False, section='spell') == True:
if c.get('plot', True,
section='spell') == False and not plot_was_false:
self.plot_spell.detach()
elif c.get('plot', True,
section='spell') == True and plot_was_false:
self.plot_spell = gtkspell.Spell(self.widgets['add']['plot'])
self.plot_spell.set_language(
c.get('lang', 'en', section='spell'))
else:
pass
if c.get('notes', True,
section='spell') == False and not notes_was_false:
self.notes_spell.detach()
elif c.get('notes', True,
section='spell') == True and notes_was_false:
self.notes_spell = gtkspell.Spell(self.widgets['add']['notes'])
self.notes_spell.set_language(
c.get('lang', 'en', section='spell'))
else:
pass
self.pdf_reader = save_reader
# extensions settings
for ext_name in plugins.extensions.by_name:
preferenceswidgets = plugins.extensions.by_name[
ext_name].preferenceswidgets
for prefname in preferenceswidgets:
widget = preferenceswidgets[prefname]
if isinstance(widget, gtk.CheckButton):
value = widget.get_active()
elif isinstance(widget, gtk.Entry):
value = widget.get_text()
elif isinstance(widget, gtk.ComboBox):
iter = widget.get_active_iter()
if iter:
value = widget.get_model().get_value(iter, 1)
else:
log.error('widget type not supported %s', type(widget))
continue
c.set("%s_%s" % (ext_name, prefname), value, section='extensions')
# database
old = c.to_dict(section='database')
c.set('host', w['db_host'].get_text(), section='database')
c.set('port', int(w['db_port'].get_value()), section='database')
c.set('name', w['db_name'].get_text(), section='database')
c.set('user', w['db_user'].get_text(), section='database')
c.set('passwd', w['db_passwd'].get_text(), section='database')
db_type = int(w['db_type'].get_active())
if db_type == 1:
c.set('type', 'postgres', section='database')
elif db_type == 2:
c.set('type', 'mysql', section='database')
elif db_type == 3:
c.set('type', 'mssql', section='database')
else:
c.set('type', 'sqlite', section='database')
if old['type'] != c.get('type', section='database') or (old['type']!='sqlite' and (\
old['host'] != c.get('host', section='database') or \
old['port'] != c.get('port', section='database') or \
old['user'] != c.get('user', section='database') or \
old['passwd'] != c.get('passwd', section='database'))) or \
old['name'] != c.get('name', section='database'):
log.info('DATABASE: connecting to new db server...')
import sql
from sqlalchemy.exc import InvalidRequestError
from initialize import dictionaries, people_treeview
# new database connection
self.initialized = False
if c.has_key('posters'):
c['posters'] = None # force update
try:
self.db.dispose()
self.db = sql.GriffithSQL(c, self.locations['home'], fallback=True)
except InvalidRequestError, e:
log.exception('')
c.set('type', 'sqlite', section='database')
w['db_type'].set_active(0)
self.db = sql.GriffithSQL(c, self.locations['home'])
log.info("New database Engine: %s" % self.db.session.bind.engine.name)
# initialize new database
self.total = int(self.db.session.query(db.Movie).count())
self.count_statusbar()
dictionaries(self)
people_treeview(self, False)
self.initialized = True
