Revert "Merge commit 'stew/content_generation' into merge"

[dak.git] / dak / dakdb / update4.py

diff --git a/dak/dakdb/update4.py b/dak/dakdb/update4.py
index 487b4792d94d0b709e5a6944177c42612032fa95..8c55d09b219066407a0fba9d9255f9d73aca0aa5 100755 (executable)
--- a/dak/dakdb/update4.py
+++ b/dak/dakdb/update4.py
@@ -27,17 +27,18 @@ Database Update Script - Get suite_architectures table use sane values
 
 import psycopg2
 from daklib.dak_exceptions import DBUpdateError
+from daklib.utils import get_conf
 
 ################################################################################
 
 suites = {}  #: Cache of existing suites
 archs = {}   #: Cache of existing architectures
 
-
 def do_update(self):
     """ Execute the DB update """
 
     print "Lets make suite_architecture table use sane values"
+    Cnf = get_conf()
 
     query = "INSERT into suite_architectures (suite, architecture) VALUES (%s, %s)"  #: Update query
     try:
@@ -59,7 +60,9 @@ def do_update(self):
             architectures = Cnf.SubTree("Suite::" + suite).ValueList("Architectures")
             suite = suite.lower()
             for arch in architectures:
-                c.execute(query, suites[suite], archs[arch])
+                c.execute(query, [suites[suite], archs[arch]])
+
+        c.execute("UPDATE config SET value = '4' WHERE name = 'db_revision'")
 
         self.db.commit()