Source Code for Module dak.dakdb.update100

 1  """ 
 2  Add a component - suite mapping to only expose certain components in certain suites 
 3   
 4  @contact: Debian FTP Master <ftpmaster@debian.org> 
 5  @copyright: 2012 Varnish Software AS 
 6  @author: Tollef Fog Heen <tfheen@varnish-software.com> 
 7  @license: GNU General Public License version 2 or later 
 8  """ 
 9   
10  # This program is free software; you can redistribute it and/or modify 
11  # it under the terms of the GNU General Public License as published by 
12  # the Free Software Foundation; either version 2 of the License, or 
13  # (at your option) any later version. 
14   
15  # This program is distributed in the hope that it will be useful, 
16  # but WITHOUT ANY WARRANTY; without even the implied warranty of 
17  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
18  # GNU General Public License for more details. 
19   
20  # You should have received a copy of the GNU General Public License 
21  # along with this program; if not, write to the Free Software 
22  # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
23   
24  ################################################################################ 
25   
26  import psycopg2 
27   
28  from daklib.dak_exceptions import DBUpdateError 
29   
30  ################################################################################ 
31   
32   
33 -def do_update(self): 
34      print(__doc__) 
35      try: 
36          c = self.db.cursor() 
37   
38          c.execute( 
39              """ 
40              CREATE TABLE component_suite ( 
41                  component_id INTEGER NOT NULL REFERENCES component(id) ON DELETE CASCADE, 
42                  suite_id INTEGER NOT NULL REFERENCES suite(id) ON DELETE CASCADE, 
43                  PRIMARY KEY (component_id, suite_id) 
44              ) 
45              """ 
46          ) 
47          # Set up default mappings for all components to all suites 
48          c.execute( 
49              "INSERT INTO component_suite(component_id, suite_id) SELECT component.id,suite.id from suite, component" 
50          ) 
51   
52          c.execute("UPDATE config SET value = '100' WHERE name = 'db_revision'") 
53          self.db.commit() 
54   
55      except psycopg2.ProgrammingError as msg: 
56          self.db.rollback() 
57          raise DBUpdateError( 
58              "Unable to apply sick update 100, rollback issued. Error message: {0}".format( 
59                  msg 
60              ) 
61          ) 
62