summaryrefslogtreecommitdiff
path: root/chimere/scripts/upgrade.py
diff options
context:
space:
mode:
Diffstat (limited to 'chimere/scripts/upgrade.py')
-rwxr-xr-xchimere/scripts/upgrade.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/chimere/scripts/upgrade.py b/chimere/scripts/upgrade.py
index 5b6b04a..f260a07 100755
--- a/chimere/scripts/upgrade.py
+++ b/chimere/scripts/upgrade.py
@@ -32,6 +32,9 @@ def slugfy(text, separator):
ret = re.sub(" +", separator, ret)
return ret.strip()
+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';"""
@@ -110,8 +113,7 @@ CREATE TABLE "main_subcategory_areas" (
# early versions before 0.1: main_tinyurl table doesn't exist
-query = """SELECT c.relname FROM pg_class c
-WHERE c.relname = 'main_tinyurl';"""
+query = QUERY_CHECK_TABLE % 'main_tinyurl'
cursor.execute(query)
transaction.commit_unless_managed()