* Convert from tar.bz2 database (db + images) format to pure (or

bzipped) sqlite db.

katalog_convert_1.0rc_to_1.0.py

@@ -0,0 +1,93 @@
+#!/usr/bin/python
+# This Python file uses the following encoding: utf-8
+#
+#  Author: Roman 'gryf' Dobosz  gryf@elysium.pl
+#
+#  Copyright (C) 2007 by Roman 'gryf' Dobosz
+#
+#  This file is part of pyGTKtalog.
+#
+#  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; either version 2 of the License, or
+#  (at your option) any later version.
+#
+#  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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+
+#  -------------------------------------------------------------------------
+import sys
+import os
+
+def setup_path():
+    """Sets up the python include paths to include needed directories"""
+    import os.path
+
+    from src.utils.globals import TOPDIR
+    sys.path = [os.path.join(TOPDIR, "src")] + sys.path
+    return
+
+if __name__ == "__main__":
+    """run the stuff"""
+    if len(sys.argv) != 3:
+        print "Usage: %s old_katalog_filename new_katalog_filename" % \
+              sys.argv[0]
+        print "All available pictures will be exported aswell, however",
+        print "thumbnails without"
+        print "images will be lost."
+        sys.exit()
+
+    # Directory from where pygtkatalog was invoced. We need it for calculate
+    # path for argument (catalog file)
+    execution_dir = os.path.abspath(os.path.curdir)
+
+    # Directory, where this files lies. We need it to setup private source
+    # paths
+    libraries_dir = os.path.dirname(__file__)
+    os.chdir(libraries_dir)
+
+    setup_path()
+
+    from shutil import copy
+
+    from utils.img import Img
+    from models.m_main10rc1 import MainModel
+    from models.m_main import MainModel as NewModel
+
+    model = MainModel()
+    new_model = NewModel()
+    if not model.open(os.path.join(execution_dir, sys.argv[1])):
+        print "cannot open katalog in 1.0RC1 format"
+        sys.exit()
+
+    model.db_cursor.execute("""create table
+                               images2(id INTEGER PRIMARY KEY AUTOINCREMENT,
+                                      file_id INTEGER,
+                                      filename TEXT);""")
+
+    result = model.db_cursor.execute("select file_id, filename from images")
+    # (id, filename)
+    # (4921, u'images/13/39.jpg')
+    for row in result.fetchall():
+        if row[1] and os.path.exists(os.path.join(model.internal_dirname,
+                                                  row[1])):
+            im = Img(os.path.join(model.internal_dirname, row[1]),
+                     new_model.image_path)
+            image = im.save()
+            sql = "insert into images2(file_id, filename) values (?, ?)"
+            model.db_cursor.execute(sql, (row[0], image))
+
+    model.db_connection.commit()
+    model.db_cursor.execute("drop table images")
+    model.db_cursor.execute("alter table images2 rename to images")
+
+    copy(os.path.join(model.internal_dirname, 'db.sqlite'),
+         os.path.join(execution_dir, sys.argv[2]))
+    # remove stuff
+    model.cleanup()