Source Code for Module dak.dakdb.update73

 1  """ 
 2  Reference archive table from suite and add path to archive root 
 3   
 4  @contact: Debian FTP Master <ftpmaster@debian.org> 
 5  @copyright: 2012 Ansgar Burchardt <ansgar@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   
27  from daklib.config import Config 
28  from daklib.dak_exceptions import DBUpdateError 
29   
30  ################################################################################ 
31   
32   
33 -def do_update(self): 
34      print(__doc__) 
35      try: 
36          cnf = Config() 
37   
38          c = self.db.cursor() 
39   
40          archive_root = cnf["Dir::Root"] 
41          c.execute( 
42              "ALTER TABLE archive ADD COLUMN path TEXT NOT NULL DEFAULT %s", 
43              (archive_root,), 
44          ) 
45          c.execute("ALTER TABLE archive ALTER COLUMN path DROP DEFAULT") 
46   
47          c.execute( 
48              "ALTER TABLE archive ADD COLUMN mode CHAR(4) NOT NULL DEFAULT '0644' CHECK (mode SIMILAR TO '[0-7]{4}')" 
49          ) 
50          c.execute("ALTER TABLE archive ADD COLUMN tainted BOOLEAN NOT NULL DEFAULT 'f'") 
51          c.execute( 
52              "ALTER TABLE archive ADD COLUMN use_morgue BOOLEAN NOT NULL DEFAULT 't'" 
53          ) 
54   
55          c.execute("SELECT id FROM archive") 
56          (archive_id,) = c.fetchone() 
57   
58          if c.fetchone() is not None: 
59              raise DBUpdateError( 
60                  "Cannot automatically upgrade form installation with multiple archives." 
61              ) 
62   
63          c.execute( 
64              "ALTER TABLE suite ADD COLUMN archive_id INT REFERENCES archive(id) NOT NULL DEFAULT %s", 
65              (archive_id,), 
66          ) 
67          c.execute("ALTER TABLE suite ALTER COLUMN archive_id DROP DEFAULT") 
68   
69          c.execute("UPDATE config SET value = '73' WHERE name = 'db_revision'") 
70          self.db.commit() 
71   
72      except psycopg2.ProgrammingError as msg: 
73          self.db.rollback() 
74          raise DBUpdateError( 
75              "Unable to apply sick update 73, rollback issued. Error message : %s" 
76              % (str(msg)) 
77          ) 
78