Source Code for Module dak.dakdb.update69

 1  """ 
 2  Add support for Description-md5 
 3   
 4  @contact: Debian FTP Master <ftpmaster@debian.org> 
 5  @copyright: 2011 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.dak_exceptions import DBUpdateError 
28   
29  ################################################################################ 
30   
31   
32 -def do_update(self): 
33      """ 
34      Add support for Description-md5 
35      """ 
36      print(__doc__) 
37      try: 
38          c = self.db.cursor() 
39   
40          c.execute( 
41              """CREATE OR REPLACE FUNCTION public.add_missing_description_md5() 
42    RETURNS VOID 
43    VOLATILE 
44    LANGUAGE plpgsql 
45  AS $function$ 
46  DECLARE 
47    description_key_id metadata_keys.key_id%TYPE; 
48    description_md5_key_id metadata_keys.key_id%TYPE; 
49    BEGIN 
50      SELECT key_id INTO STRICT description_key_id FROM metadata_keys WHERE key='Description'; 
51      SELECT key_id INTO description_md5_key_id FROM metadata_keys WHERE key='Description-md5'; 
52      IF NOT FOUND THEN 
53        INSERT INTO metadata_keys (key) VALUES ('Description-md5') RETURNING key_id INTO description_md5_key_id; 
54      END IF; 
55   
56      INSERT INTO binaries_metadata 
57        (bin_id, key_id, value) 
58      SELECT 
59        bm.bin_id AS bin_id, 
60        description_md5_key_id AS key_id, 
61        MD5(bm.value || E'\n') AS value 
62      FROM binaries_metadata AS bm 
63      WHERE 
64        bm.key_id = description_key_id 
65        AND 
66        NOT EXISTS (SELECT 1 FROM binaries_metadata AS bm2 WHERE bm.bin_id = bm2.bin_id AND bm2.key_id = description_md5_key_id); 
67  END; 
68  $function$""" 
69          ) 
70   
71          c.execute( 
72              "ALTER TABLE suite ADD COLUMN include_long_description BOOLEAN NOT NULL DEFAULT 't'" 
73          ) 
74   
75          c.execute("UPDATE config SET value = '69' WHERE name = 'db_revision'") 
76          self.db.commit() 
77   
78      except psycopg2.ProgrammingError as msg: 
79          self.db.rollback() 
80          raise DBUpdateError( 
81              "Unable to apply sick update 69, rollback issued. Error message : %s" 
82              % (str(msg)) 
83          ) 
84