Fix(glst): Return counter after VACUUM

[m_librarian.git] / m_librarian / glst.py

diff --git a/m_librarian/glst.py b/m_librarian/glst.py
index f639c1b3b6d4cb58bb6c1e58e3262336376845e4..6b2e0904157870d2de8c52298e31a64b736c3a0a 100755 (executable)
--- a/m_librarian/glst.py
+++ b/m_librarian/glst.py
@@ -48,18 +48,23 @@ def _import_glst():
     old_nonfb2, new_nonfb2 = import_glst_file(
         os.path.join(ml_dir, 'glst', 'genres_nonfb2.glst'))
     connection = sqlhub.processConnection
-    if connection.dbName in ('postgres', 'sqlite'):
+    if connection.dbName == 'postgres':
         connection.query("VACUUM %s" % Genre.sqlmeta.table)
     return old_fb2 + old_nonfb2, new_fb2 + new_nonfb2
 
 
 def import_glst():
-    return sqlhub.doInTransaction(_import_glst)
+    count_old, count_new = sqlhub.doInTransaction(_import_glst)
+    connection = sqlhub.processConnection
+    if connection.dbName == 'sqlite':
+        connection.query("VACUUM")
+    return count_old, count_new
 
 
 def test():
     ml_dir = os.path.dirname(__file__)
     print(parse_glst_file(os.path.join(ml_dir, 'glst', 'genres_fb2.glst')))
 
+
 if __name__ == '__main__':
     test()