Source Code for Module dak.dakdb.update105

 1  """ 
 2  Add a new release_suite name which we use in generate_releases 
 3   
 4  @contact: Debian FTP Master <ftpmaster@debian.org> 
 5  @copyright: 2014, Mark Hymers <mhy@debian.org> 
 6  @license: GNU General Public License version 2 or later 
 7  """ 
 8   
 9  # This program is free software; you can redistribute it and/or modify 
10  # it under the terms of the GNU General Public License as published by 
11  # the Free Software Foundation; either version 2 of the License, or 
12  # (at your option) any later version. 
13   
14  # This program is distributed in the hope that it will be useful, 
15  # but WITHOUT ANY WARRANTY; without even the implied warranty of 
16  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
17  # GNU General Public License for more details. 
18   
19  # You should have received a copy of the GNU General Public License 
20  # along with this program; if not, write to the Free Software 
21  # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
22   
23  ################################################################################ 
24   
25  import psycopg2 
26  from daklib.dak_exceptions import DBUpdateError 
27  from daklib.config import Config 
28   
29  # This includes some updates to the tables which are Debian specific but 
30  # shouldn't affect anyone else 
31  statements = [ 
32  "ALTER TABLE suite ADD COLUMN release_suite TEXT DEFAULT NULL", 
33  "UPDATE suite SET release_suite = 'oldstable-updates' WHERE suite_name = 'squeeze-updates'", 
34  "UPDATE suite SET release_suite = 'stable-updates' WHERE suite_name = 'wheezy-updates'", 
35  "UPDATE suite SET release_suite = 'testing-updates' WHERE suite_name = 'jessie-updates'", 
36  ] 
37   
38  ################################################################################ 
39   
40   
41 -def do_update(self): 
42      print(__doc__) 
43      try: 
44          cnf = Config() 
45   
46          c = self.db.cursor() 
47   
48          for stmt in statements: 
49              c.execute(stmt) 
50   
51          c.execute("UPDATE config SET value = '105' WHERE name = 'db_revision'") 
52          self.db.commit() 
53   
54      except psycopg2.ProgrammingError as msg: 
55          self.db.rollback() 
56          raise DBUpdateError('Unable to apply sick update 105, rollback issued. Error message: {0}'.format(msg)) 
57