From 047416ad2a7d39aa41459df91a6a0b49daba92a4 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Thu, 18 Nov 2010 19:10:58 +0100 Subject: Correct the upgrade script --- chimere/scripts/upgrade.py | 6 ++++-- 1 file 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() -- cgit v1.2.3