VACUUM all tables after mass-insert

[m_librarian.git] / m_librarian / glst.py

diff --git a/m_librarian/glst.py b/m_librarian/glst.py
index 867d4136590e2a0f4ccee0c429e0e7e270f065bf..b94381377e35072e98817665b18042cc2ac9c55a 100755 (executable)
--- a/m_librarian/glst.py
+++ b/m_librarian/glst.py
@@ -1,6 +1,6 @@
 #! /usr/bin/env python
 
-__all__ = ['import_data']
+__all__ = ['import_glst']
 
 import codecs
 import os
@@ -42,23 +42,25 @@ def import_glst_file(glst_filename):
     return old, new
 
 
-def _import_data():
+def _import_glst():
     ml_dir = os.path.dirname(__file__)
     old_fb2, new_fb2 = import_glst_file(
-        os.path.join(ml_dir, 'data', 'genres_fb2.glst'))
+        os.path.join(ml_dir, 'glst', 'genres_fb2.glst'))
     old_nonfb2, new_nonfb2 = import_glst_file(
-        os.path.join(ml_dir, 'data', 'genres_nonfb2.glst'))
-    sqlhub.processConnection.query("VACUUM %s" % Genre.sqlmeta.table)
+        os.path.join(ml_dir, 'glst', 'genres_nonfb2.glst'))
+    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
 
 
-def import_data():
-    return sqlhub.doInTransaction(_import_data)
+def import_glst():
+    return sqlhub.doInTransaction(_import_glst)
 
 
 def test():
     ml_dir = os.path.dirname(__file__)
-    print parse_glst_file(os.path.join(ml_dir, 'data', 'genres_fb2.glst'))
+    print parse_glst_file(os.path.join(ml_dir, 'glst', 'genres_fb2.glst'))
 
 if __name__ == '__main__':
     test()