X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=m_librarian%2Fglst.py;h=2de36c3a92180b7abaf1b7b76079e51938e93c06;hb=f29d3e8f0981bcc8df87b8e661a5bf019a152fae;hp=44ec6121cbeaf6038730f3a5f9c17abf0a99a0d4;hpb=d0a1abea3d89062e8f9b71dd6df45db5d8b84997;p=m_librarian.git diff --git a/m_librarian/glst.py b/m_librarian/glst.py index 44ec612..2de36c3 100755 --- a/m_librarian/glst.py +++ b/m_librarian/glst.py @@ -10,7 +10,6 @@ from .db import Genre def parse_glst_file(glst_filename): glst_file = codecs.open(glst_filename, 'r', 'utf-8') - genre_list = [] try: for line in glst_file: line = line.strip() @@ -23,10 +22,9 @@ def parse_glst_file(glst_filename): name, title = parts[1].split(';', 1) except (IndexError, ValueError): continue - genre_list.append((name, title)) + yield name, title finally: glst_file.close() - return genre_list def import_glst_file(glst_filename): @@ -48,7 +46,9 @@ def _import_glst(): os.path.join(ml_dir, 'glst', 'genres_fb2.glst')) old_nonfb2, new_nonfb2 = import_glst_file( os.path.join(ml_dir, 'glst', 'genres_nonfb2.glst')) - sqlhub.processConnection.query("VACUUM %s" % Genre.sqlmeta.table) + connection = sqlhub.processConnection + if connection.dbName in ('postgres', 'sqlite'): + connection.query("VACUUM %s" % Genre.sqlmeta.table) return old_fb2 + old_nonfb2, new_fb2 + new_nonfb2