Source Code for Module dak.dakdb.update99

 1  """ 
 2  Add component ordering 
 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  statements = [ 
31      """ 
32  ALTER TABLE component 
33  ADD COLUMN ordering INTEGER UNIQUE 
34  """, 
35      """ 
36  CREATE SEQUENCE component_ordering_seq 
37  INCREMENT BY 10 
38  START WITH 100 
39  OWNED BY component.ordering 
40  """, 
41  ] 
42   
43  ################################################################################ 
44   
45   
46 -def do_update(self): 
47      print(__doc__) 
48      try: 
49          c = self.db.cursor() 
50   
51          for stmt in statements: 
52              c.execute(stmt) 
53   
54          for component in ("main", "contrib", "non-free-firmware", "non-free"): 
55              c.execute( 
56                  "UPDATE component SET ordering = nextval('component_ordering_seq') WHERE name = '{0}'".format( 
57                      component 
58                  ) 
59              ) 
60          c.execute( 
61              "UPDATE component SET ordering = nextval('component_ordering_seq') WHERE ordering IS NULL" 
62          ) 
63          c.execute("""ALTER TABLE component ALTER COLUMN ordering SET NOT NULL""") 
64          c.execute( 
65              """ALTER TABLE component ALTER COLUMN ordering SET DEFAULT nextval('component_ordering_seq')""" 
66          ) 
67   
68          c.execute("UPDATE config SET value = '99' WHERE name = 'db_revision'") 
69   
70          self.db.commit() 
71   
72      except psycopg2.ProgrammingError as msg: 
73          self.db.rollback() 
74          raise DBUpdateError( 
75              "Unable to apply sick update 99, rollback issued. Error message: {0}".format( 
76                  msg 
77              ) 
78          ) 
79