1 files changed, 70 insertions, 4 deletions
diff --git a/chimere/scripts/upgrade.py b/chimere/scripts/upgrade.py
index f260a07..4c1cdc0 100755
--- a/chimere/scripts/upgrade.py
+++ b/chimere/scripts/upgrade.py
@@ -13,7 +13,7 @@ from django.db import connection, transaction
 
 cursor = connection.cursor()
 
-from main.models import Area, Route
+from main.models import Area, Marker, Route, Icon, SubCategory
 from django.contrib.gis.geos import LineString
 
 # early versions before 0.1: urn field doesn't exist for area
@@ -32,12 +32,14 @@ def slugfy(text, separator):
     ret = re.sub(" +", separator, ret)
     return ret.strip()
 
+QUERY_CHECK_FIELD = """SELECT a.attname AS field FROM pg_class c, pg_attribute a
+    WHERE c.relname = '%s' AND a.attnum > 0 AND a.attrelid = c.oid
+          AND a.attname='%s';"""
 QUERY_CHECK_TABLE = """SELECT c.relname FROM pg_class c
 WHERE c.relname = '%s';"""
 
-query = """SELECT a.attname AS field FROM pg_class c, pg_attribute a
-    WHERE c.relname = 'main_area' AND a.attnum > 0 AND a.attrelid = c.oid
-          AND a.attname='urn';"""
+
+query = QUERY_CHECK_FIELD % ('main_area', 'urn')
 cursor.execute(query)
 transaction.commit_unless_managed()
 
@@ -159,6 +161,69 @@ for area in areas:
 if changed:
     print " * projections of areas corrected"
 
+# changement from version 1.0 to 1.1: version of django 1.2
+# create specific height and width for image fields
+
+for cls, attr in ((Icon, "image"), (Marker, "picture"),
+                  (Route, "picture")):
+    table = cls._meta.db_table
+    query = QUERY_CHECK_FIELD % (table, 'width')
+    cursor.execute(query)
+    transaction.commit_unless_managed()
+
+    row = cursor.fetchone()
+    if not row:
+        query_update = "ALTER TABLE "+table+" ADD COLUMN width integer"
+        cursor.execute(query_update)
+        transaction.commit_unless_managed()
+        query_update = "ALTER TABLE "+table+" ADD COLUMN height integer"
+        cursor.execute(query_update)
+        transaction.commit_unless_managed()
+        for obj in cls.objects.all():
+            image = getattr(obj, attr)
+            if not image:
+                continue
+            obj.width = image.width
+            obj.height = image.height
+            obj.save()
+        print " * height and width of " + table + " corrected"
+
+# changement from version 1.0 to 1.1: multiple selection of categories
+
+for cls in (Marker, Route):
+    table = cls._meta.db_table[len("main_"):]
+    query = QUERY_CHECK_TABLE % ('main_' + table + '_categories')
+    cursor.execute(query)
+    transaction.commit_unless_managed()
+
+    row = cursor.fetchone()
+    if row:
+        continue
+    query_create = """
+CREATE TABLE "main_%s_categories" (
+  "id" serial NOT NULL PRIMARY KEY,
+  "%s_id" integer NOT NULL REFERENCES "main_%s" ("id") DEFERRABLE INITIALLY DEFERRED,
+  "subcategory_id" integer NOT NULL REFERENCES "main_subcategory" ("id") DEFERRABLE INITIALLY DEFERRED,
+  UNIQUE ("%s_id", "subcategory_id"));
+""" % (table, table, table, table)
+    cursor.execute(query_create)
+    transaction.commit_unless_managed()
+    for obj in cls.objects.all():
+        query = "select subcategory_id from main_%s where id=%d" % (table,
+                                                                    obj.id)
+        cursor.execute(query)
+        transaction.commit_unless_managed()
+
+        row = cursor.fetchone()
+        if row:
+            obj.categories.add(SubCategory.objects.get(id=row[0]))
+            obj.save()
+    query = "ALTER TABLE main_%s DROP COLUMN subcategory_id;" % table
+    cursor.execute(query)
+    transaction.commit_unless_managed()
+    print " * main_%s_categories created" % table
+
+
 # early versions before 0.1: save route with wrong SRID
 # only errors with default SRID is managed adapt the script for your SRID
 
@@ -177,3 +242,4 @@ for route in routes:
         route.save()
 if changed:
     print " * projections of routes corrected"
+