Source Code for Module dak.dakdb.update70

 1  """ 
 2  Add suite options for overrides and control-suite to DB 
 3   
 4  @contact: Debian FTP Master <ftpmaster@debian.org> 
 5  @copyright: 2011 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  ################################################################################ 
30   
31   
32 -def do_update(self): 
33      """ 
34      Add suite options for overrides and control-suite to DB 
35      """ 
36      print(__doc__) 
37      try: 
38          cnf = Config() 
39   
40          c = self.db.cursor() 
41   
42          c.execute("ALTER TABLE suite ADD COLUMN overrideprocess BOOLEAN NOT NULL DEFAULT FALSE") 
43          c.execute("COMMENT ON COLUMN suite.overrideprocess IS %s", ['If true, check-overrides will process the suite by default']) 
44          c.execute("ALTER TABLE suite ADD COLUMN overrideorigin TEXT DEFAULT NULL") 
45          c.execute("COMMENT ON COLUMN suite.overrideprocess IS %s", ['If NOT NULL, check-overrides will take missing overrides from the named suite']) 
46   
47          # Migrate config file values into database 
48          if "Check-Overrides::OverrideSuites" in cnf: 
49              for suitename in cnf.subtree("Check-Overrides::OverrideSuites").list(): 
50                  if cnf.get("Check-Overrides::OverrideSuites::%s::Process" % suitename, "0") == "1": 
51                      print("Marking %s to have overrides processed automatically" % suitename.lower()) 
52                      c.execute("UPDATE suite SET overrideprocess = TRUE WHERE suite_name = %s", [suitename.lower()]) 
53   
54                  originsuite = cnf.get("Check-Overrides::OverrideSuites::%s::OriginSuite" % suitename, '') 
55                  if originsuite != '': 
56                      print("Setting %s to use %s as origin for overrides" % (suitename.lower(), originsuite.lower())) 
57                      c.execute("UPDATE suite SET overrideorigin = %s WHERE suite_name = %s", [originsuite.lower(), suitename.lower()]) 
58   
59          c.execute("ALTER TABLE suite ADD COLUMN allowcsset BOOLEAN NOT NULL DEFAULT FALSE") 
60          c.execute("COMMENT ON COLUMN suite.allowcsset IS %s", ['Allow control-suite to be used with the --set option without forcing']) 
61   
62          # Import historical hard-coded values 
63          c.execute("UPDATE suite SET allowcsset = TRUE WHERE suite_name IN ('testing', 'squeeze-updates')") 
64   
65          c.execute("UPDATE config SET value = '70' WHERE name = 'db_revision'") 
66          self.db.commit() 
67   
68      except psycopg2.ProgrammingError as msg: 
69          self.db.rollback() 
70          raise DBUpdateError('Unable to apply sick update 70, rollback issued. Error message : %s' % (str(msg))) 
71