Upgrade script to create table main_tiny_url if not available

1 files changed, 18 insertions, 1 deletions

diff --git a/chimere/scripts/upgrade.py b/chimere/scripts/upgrade.py
index b8cbec8..463834d 100755
--- a/chimere/scripts/upgrade.py
+++ b/chimere/scripts/upgrade.py
@@ -74,12 +74,29 @@ CREATE TABLE "subcategory_areas" (
     "area_id" integer NOT NULL REFERENCES "main_area" ("id")
                                                   DEFERRABLE INITIALLY DEFERRED,
     UNIQUE ("subcategory_id", "area_id"));
-
 """
     cursor.execute(query_create)
     transaction.commit_unless_managed()
     print " * subcategory_areas created"
 
+# early versions before 0.1: main_tinyurl table doesn't exist
+
+query = """SELECT c.relname FROM pg_class c
+WHERE c.relname = 'main_tinyurl';"""
+cursor.execute(query)
+transaction.commit_unless_managed()
+
+row = cursor.fetchone()
+if not row:
+    query_create = """
+CREATE TABLE "main_tinyurl" (
+    "id" serial NOT NULL PRIMARY KEY,
+    "parameters" varchar(500) NOT NULL);
+"""
+    cursor.execute(query_create)
+    transaction.commit_unless_managed()
+    print " * main_tinyurl created"
+
 
 # early versions before 0.1: save area with wrong SRID
 # only errors with default SRID is managed adapt the script for your SRID