Better management of rights for categories not associated to an area (fixes #268)

1 files changed, 36 insertions, 7 deletions

diff --git a/chimere/scripts/upgrade.py b/chimere/scripts/upgrade.py
index 0649a52..4c1cdc0 100755
--- a/chimere/scripts/upgrade.py
+++ b/chimere/scripts/upgrade.py
@@ -62,17 +62,46 @@ UNIQUE"
     transaction.commit_unless_managed()
     print " * urn field has now the constraint NOT NULL"
 
+from django.contrib.auth.models import Permission, ContentType
+
+# check if permission have been correctly set for each areas
+areas = Area.objects.all()
+for area in areas:
+    content_type = ContentType.objects.get(app_label="main",
+            model="area")
+    mnemo = 'change_area_' + area.urn
+    perm = Permission.objects.filter(codename=mnemo)
+    if not perm:
+        lbl = "Can change " + area.name
+        perm = Permission(name=lbl, content_type_id=content_type.id,
+                          codename=mnemo)
+        perm.save()
+        print ' * permission "' + lbl + '" has been created'
+        print " WARNING: don't forget to update administrator's rights with \
+this new permission"
+
 # early versions before 0.1: subcategory_areas table doesn't exist
+# version 1.0 subcategory_areas renamed to main_subcategory_areas
 
-query = """SELECT c.relname FROM pg_class c
-WHERE c.relname = 'subcategory_areas';"""
+query = QUERY_CHECK_TABLE % 'main_subcategory_areas'
 cursor.execute(query)
 transaction.commit_unless_managed()
 
 row = cursor.fetchone()
 if not row:
-    query_create = """
-CREATE TABLE "subcategory_areas" (
+    query = QUERY_CHECK_TABLE % 'subcategory_areas'
+    cursor.execute(query)
+    transaction.commit_unless_managed()
+
+    row = cursor.fetchone()
+    if row:
+        query_rename = """ALTER TABLE subcategory_areas RENAME TO main_subcategory_areas;"""
+        cursor.execute(query_rename)
+        transaction.commit_unless_managed()
+        print " * subcategory_areas renamed to main_subcategory_areas"
+    else:
+        query_create = """
+CREATE TABLE "main_subcategory_areas" (
     "id" serial NOT NULL PRIMARY KEY,
     "subcategory_id" integer NOT NULL REFERENCES "main_subcategory" ("id")
                                                   DEFERRABLE INITIALLY DEFERRED,
@@ -80,9 +109,9 @@ CREATE TABLE "subcategory_areas" (
                                                   DEFERRABLE INITIALLY DEFERRED,
     UNIQUE ("subcategory_id", "area_id"));
 """
-    cursor.execute(query_create)
-    transaction.commit_unless_managed()
-    print " * subcategory_areas created"
+        cursor.execute(query_create)
+        transaction.commit_unless_managed()
+        print " * main_subcategory_areas created"
 
 # early versions before 0.1: main_tinyurl table doesn't exist