From 7ed15b809fd22f3325aed4e4a1ecac4387417437 Mon Sep 17 00:00:00 2001 From: Tollef Fog Heen Date: Wed, 9 Jan 2013 13:46:26 +0100 Subject: [PATCH] dak/dakdb/update100.py: Add component - suite mapping table This table will be used to only expose certain components in certain suites. Signed-off-by: Tollef Fog Heen --- dak/dakdb/update100.py | 56 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 56 insertions(+) create mode 100644 dak/dakdb/update100.py diff --git a/dak/dakdb/update100.py b/dak/dakdb/update100.py new file mode 100644 index 00000000..85733cb6 --- /dev/null +++ b/dak/dakdb/update100.py @@ -0,0 +1,56 @@ +#!/usr/bin/env python +# coding=utf8 + +""" +Add a component - suite mapping to only expose certain components in certain suites + +@contact: Debian FTP Master +@copyright: 2012 Varnish Software AS +@author: Tollef Fog Heen +@license: GNU General Public License version 2 or later +""" + +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. + +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. + +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + +################################################################################ + +import psycopg2 +from daklib.dak_exceptions import DBUpdateError +from daklib.config import Config + +################################################################################ +def do_update(self): + print __doc__ + try: + cnf = Config() + + c = self.db.cursor() + + c.execute(""" + CREATE TABLE component_suite ( + component_id INTEGER NOT NULL REFERENCES component(id) ON DELETE CASCADE, + suite_id INTEGER NOT NULL REFERENCES suite(id) ON DELETE CASCADE, + PRIMARY KEY (component_id, suite_id) + ) + """) + # Set up default mappings for all components to all suites + c.execute("INSERT INTO component_suite(component_id, suite_id) SELECT component.id,suite.id from suite, component") + + c.execute("UPDATE config SET value = '100' WHERE name = 'db_revision'") + self.db.commit() + + except psycopg2.ProgrammingError as msg: + self.db.rollback() + raise DBUpdateError('Unable to apply sick update 100, rollback issued. Error message: {0}'.format(msg)) -- 2.39.2