]> git.decadent.org.uk Git - dak.git/blob - dak/dakdb/update28.py
modify new-security-install to allow multiple packages accepted at once
[dak.git] / dak / dakdb / update28.py
1 #!/usr/bin/env python
2 # coding=utf8
3
4 """
5 keep contents of binary packages in tables so we can generate contents.gz files from dak
6
7 @contact: Debian FTP Master <ftpmaster@debian.org>
8 @copyright: 2009  Mike O'Connor <stew@debian.org>
9 @license: GNU General Public License version 2 or later
10 """
11
12 # This program is free software; you can redistribute it and/or modify
13 # it under the terms of the GNU General Public License as published by
14 # the Free Software Foundation; either version 2 of the License, or
15 # (at your option) any later version.
16
17 # This program is distributed in the hope that it will be useful,
18 # but WITHOUT ANY WARRANTY; without even the implied warranty of
19 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20 # GNU General Public License for more details.
21
22 # You should have received a copy of the GNU General Public License
23 # along with this program; if not, write to the Free Software
24 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
25
26 ################################################################################
27
28
29 ################################################################################
30
31 import psycopg2
32 import time
33 from daklib.dak_exceptions import DBUpdateError
34 from daklib.config import Config
35 import string
36
37 ################################################################################
38
39 def _suites():
40     """
41     return a list of suites to operate on
42     """
43     suites = Config().SubTree("Suite").List()
44     return suites
45
46 def arches(cursor, suite):
47     """
48     return a list of archs to operate on
49     """
50     arch_list = []
51     cursor.execute("""SELECT s.architecture, a.arch_string
52     FROM suite_architectures s
53     JOIN architecture a ON (s.architecture=a.id)
54     WHERE suite = '%s'""" % suite)
55
56     while True:
57         r = cursor.fetchone()
58         if not r:
59             break
60
61         if r[1] != "source" and r[1] != "all":
62             arch_list.append((r[0], r[1]))
63
64     return arch_list
65
66 def do_update(self):
67     """
68     Adding contents table as first step to maybe, finally getting rid
69     of apt-ftparchive
70     """
71
72     print __doc__
73
74     try:
75         c = self.db.cursor()
76
77         c.execute("""CREATE TABLE pending_bin_contents (
78         id serial NOT NULL,
79         package text NOT NULL,
80         version debversion NOT NULL,
81         arch int NOT NULL,
82         filename text NOT NULL,
83         type int NOT NULL,
84         PRIMARY KEY(id))""" );
85
86         c.execute("""CREATE TABLE deb_contents (
87         filename text,
88         section text,
89         package text,
90         binary_id integer,
91         arch integer,
92         suite integer)""" )
93
94         c.execute("""CREATE TABLE udeb_contents (
95         filename text,
96         section text,
97         package text,
98         binary_id integer,
99         suite integer,
100         arch integer)""" )
101
102         c.execute("""ALTER TABLE ONLY deb_contents
103         ADD CONSTRAINT deb_contents_arch_fkey
104         FOREIGN KEY (arch) REFERENCES architecture(id)
105         ON DELETE CASCADE;""")
106
107         c.execute("""ALTER TABLE ONLY udeb_contents
108         ADD CONSTRAINT udeb_contents_arch_fkey
109         FOREIGN KEY (arch) REFERENCES architecture(id)
110         ON DELETE CASCADE;""")
111
112         c.execute("""ALTER TABLE ONLY deb_contents
113         ADD CONSTRAINT deb_contents_pkey
114         PRIMARY KEY (filename,package,arch,suite);""")
115
116         c.execute("""ALTER TABLE ONLY udeb_contents
117         ADD CONSTRAINT udeb_contents_pkey
118         PRIMARY KEY (filename,package,arch,suite);""")
119
120         c.execute("""ALTER TABLE ONLY deb_contents
121         ADD CONSTRAINT deb_contents_suite_fkey
122         FOREIGN KEY (suite) REFERENCES suite(id)
123         ON DELETE CASCADE;""")
124
125         c.execute("""ALTER TABLE ONLY udeb_contents
126         ADD CONSTRAINT udeb_contents_suite_fkey
127         FOREIGN KEY (suite) REFERENCES suite(id)
128         ON DELETE CASCADE;""")
129
130         c.execute("""ALTER TABLE ONLY deb_contents
131         ADD CONSTRAINT deb_contents_binary_fkey
132         FOREIGN KEY (binary_id) REFERENCES binaries(id)
133         ON DELETE CASCADE;""")
134
135         c.execute("""ALTER TABLE ONLY udeb_contents
136         ADD CONSTRAINT udeb_contents_binary_fkey
137         FOREIGN KEY (binary_id) REFERENCES binaries(id)
138         ON DELETE CASCADE;""")
139
140         c.execute("""CREATE INDEX ind_deb_contents_binary ON deb_contents(binary_id);""" )
141
142         suites = _suites()
143
144         for suite in [i.lower() for i in suites]:
145
146             c.execute("SELECT id FROM suite WHERE suite_name ='%s'" % suite )
147             suiterow = c.fetchone()
148             suite_id = suiterow[0]
149             arch_list = arches(c, suite_id)
150             arch_list = arches(c, suite_id)
151             suitestr=string.replace(suite,'-','_');
152
153             for (arch_id,arch_str) in arch_list:
154                 arch_str = string.replace(arch_str,"-", "_")
155                 c.execute( "CREATE INDEX ind_deb_contents_%s_%s ON deb_contents (arch,suite) WHERE (arch=2 OR arch=%s) AND suite='%s'"%(arch_str,suitestr,arch_id,suite_id) )
156
157             for section, sname in [("debian-installer","main"),
158                                   ("non-free/debian-installer", "nonfree")]:
159                 c.execute( "CREATE INDEX ind_udeb_contents_%s_%s ON udeb_contents (section,suite) WHERE section='%s' AND suite='%s'"%(sname,suitestr,section,suite_id) )
160
161
162         c.execute( """CREATE OR REPLACE FUNCTION update_contents_for_bin_a() RETURNS trigger AS  $$
163     event = TD["event"]
164     if event == "DELETE" or event == "UPDATE":
165
166         plpy.execute(plpy.prepare("DELETE FROM deb_contents WHERE binary_id=$1 and suite=$2",
167                                   ["int","int"]),
168                                   [TD["old"]["bin"], TD["old"]["suite"]])
169
170     if event == "INSERT" or event == "UPDATE":
171
172        content_data = plpy.execute(plpy.prepare(
173             \"\"\"SELECT s.section, b.package, b.architecture, ot.type
174             FROM override o
175             JOIN override_type ot on o.type=ot.id
176             JOIN binaries b on b.package=o.package
177             JOIN files f on b.file=f.id
178             JOIN location l on l.id=f.location
179             JOIN section s on s.id=o.section
180             WHERE b.id=$1
181             AND o.suite=$2
182             \"\"\",
183             ["int", "int"]),
184             [TD["new"]["bin"], TD["new"]["suite"]])[0]
185
186        tablename="%s_contents" % content_data['type']
187
188        plpy.execute(plpy.prepare(\"\"\"DELETE FROM %s
189                    WHERE package=$1 and arch=$2 and suite=$3\"\"\" % tablename,
190                    ['text','int','int']),
191                    [content_data['package'],
192                    content_data['architecture'],
193                    TD["new"]["suite"]])
194
195        filenames = plpy.execute(plpy.prepare(
196            "SELECT bc.file FROM bin_contents bc where bc.binary_id=$1",
197            ["int"]),
198            [TD["new"]["bin"]])
199
200        for filename in filenames:
201            plpy.execute(plpy.prepare(
202                \"\"\"INSERT INTO %s
203                    (filename,section,package,binary_id,arch,suite)
204                    VALUES($1,$2,$3,$4,$5,$6)\"\"\" % tablename,
205                ["text","text","text","int","int","int"]),
206                [filename["file"],
207                 content_data["section"],
208                 content_data["package"],
209                 TD["new"]["bin"],
210                 content_data["architecture"],
211                 TD["new"]["suite"]] )
212 $$ LANGUAGE plpythonu VOLATILE SECURITY DEFINER;
213 """)
214
215
216         c.execute( """CREATE OR REPLACE FUNCTION update_contents_for_override() RETURNS trigger AS  $$
217     event = TD["event"]
218     if event == "UPDATE":
219
220         otype = plpy.execute(plpy.prepare("SELECT type from override_type where id=$1",["int"]),[TD["new"]["type"]] )[0];
221         if otype["type"].endswith("deb"):
222             section = plpy.execute(plpy.prepare("SELECT section from section where id=$1",["int"]),[TD["new"]["section"]] )[0];
223
224             table_name = "%s_contents" % otype["type"]
225             plpy.execute(plpy.prepare("UPDATE %s set section=$1 where package=$2 and suite=$3" % table_name,
226                                       ["text","text","int"]),
227                                       [section["section"],
228                                       TD["new"]["package"],
229                                       TD["new"]["suite"]])
230
231 $$ LANGUAGE plpythonu VOLATILE SECURITY DEFINER;
232 """)
233
234         c.execute("""CREATE OR REPLACE FUNCTION update_contents_for_override()
235                       RETURNS trigger AS  $$
236     event = TD["event"]
237     if event == "UPDATE" or event == "INSERT":
238         row = TD["new"]
239         r = plpy.execute(plpy.prepare( \"\"\"SELECT 1 from suite_architectures sa
240                   JOIN binaries b ON b.architecture = sa.architecture
241                   WHERE b.id = $1 and sa.suite = $2\"\"\",
242                 ["int", "int"]),
243                 [row["bin"], row["suite"]])
244         if not len(r):
245             plpy.error("Illegal architecture for this suite")
246
247 $$ LANGUAGE plpythonu VOLATILE;""")
248
249         c.execute( """CREATE TRIGGER illegal_suite_arch_bin_associations_trigger
250                       BEFORE INSERT OR UPDATE ON bin_associations
251                       FOR EACH ROW EXECUTE PROCEDURE update_contents_for_override();""")
252
253         c.execute( """CREATE TRIGGER bin_associations_contents_trigger
254                       AFTER INSERT OR UPDATE OR DELETE ON bin_associations
255                       FOR EACH ROW EXECUTE PROCEDURE update_contents_for_bin_a();""")
256         c.execute("""CREATE TRIGGER override_contents_trigger
257                       AFTER UPDATE ON override
258                       FOR EACH ROW EXECUTE PROCEDURE update_contents_for_override();""")
259
260
261         c.execute( "CREATE INDEX ind_deb_contents_name ON deb_contents(package);");
262         c.execute( "CREATE INDEX ind_udeb_contents_name ON udeb_contents(package);");
263
264         c.execute("UPDATE config SET value = '28' WHERE name = 'db_revision'")
265
266         self.db.commit()
267
268     except psycopg2.ProgrammingError, msg:
269         self.db.rollback()
270         raise DBUpdateError, "Unable to apply process-new update 28, rollback issued. Error message : %s" % (str(msg))
271