]> git.decadent.org.uk Git - dak.git/blob - dak/override.py
Merge
[dak.git] / dak / override.py
1 #!/usr/bin/env python
2
3 # Microscopic modification and query tool for overrides in projectb
4 # Copyright (C) 2004, 2006  Daniel Silverstone <dsilvers@digital-scurf.org>
5
6 # This program is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
10
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 # GNU General Public License for more details.
15
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, write to the Free Software
18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19
20
21 ################################################################################
22 ## So line up your soldiers and she'll shoot them all down
23 ## Coz Alisha Rules The World
24 ## You think you found a dream, then it shatters and it seems,
25 ## That Alisha Rules The World
26 ################################################################################
27
28 import pg, sys
29 import apt_pkg
30 import daklib.logging
31 import daklib.database
32 import daklib.utils
33
34 ################################################################################
35
36 Cnf = None
37 projectB = None
38
39 ################################################################################
40
41 # Shamelessly stolen from 'dak rm'. Should probably end up in daklib.utils.py
42 def game_over():
43     answer = daklib.utils.our_raw_input("Continue (y/N)? ").lower()
44     if answer != "y":
45         print "Aborted."
46         sys.exit(1)
47
48
49 def usage (exit_code=0):
50     print """Usage: dak override [OPTIONS] package [section] [priority]
51 Make microchanges or microqueries of the binary overrides
52
53   -h, --help                 show this help and exit
54   -d, --done=BUG#            send priority/section change as closure to bug#
55   -n, --no-action            don't do anything
56   -s, --suite                specify the suite to use
57 """
58     sys.exit(exit_code)
59
60 def main ():
61     global Cnf, projectB
62
63     Cnf = daklib.utils.get_conf()
64
65     Arguments = [('h',"help","Override::Options::Help"),
66                  ('d',"done","Override::Options::Done", "HasArg"),
67                  ('n',"no-action","Override::Options::No-Action"),
68                  ('s',"suite","Override::Options::Suite", "HasArg"),
69                  ]
70     for i in ["help", "no-action"]:
71         if not Cnf.has_key("Override::Options::%s" % (i)):
72             Cnf["Override::Options::%s" % (i)] = ""
73     if not Cnf.has_key("Override::Options::Suite"):
74         Cnf["Override::Options::Suite"] = "unstable"
75
76     arguments = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv)
77     Options = Cnf.SubTree("Override::Options")
78
79     if Options["Help"]:
80         usage()
81
82     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]))
83     daklib.database.init(Cnf, projectB)
84
85     if not arguments:
86         daklib.utils.fubar("package name is a required argument.")
87
88     package = arguments.pop(0)
89     suite = Options["Suite"]
90     if arguments and len(arguments) > 2:
91         daklib.utils.fubar("Too many arguments")
92
93     if arguments and len(arguments) == 1:
94         # Determine if the argument is a priority or a section...
95         arg = arguments.pop()
96         q = projectB.query("""
97         SELECT ( SELECT COUNT(*) FROM section WHERE section=%s ) AS secs,
98                ( SELECT COUNT(*) FROM priority WHERE priority=%s ) AS prios
99                """ % ( pg._quote(arg,"str"), pg._quote(arg,"str")))
100         r = q.getresult()
101         if r[0][0] == 1:
102             arguments = (arg,".")
103         elif r[0][1] == 1:
104             arguments = (".",arg)
105         else:
106             daklib.utils.fubar("%s is not a valid section or priority" % (arg))
107
108     # Retrieve current section/priority...
109     oldsection, oldsourcesection, oldpriority = None, None, None
110     for type in ['source', 'binary']:
111         eqdsc = '!='
112         if type == 'source':
113             eqdsc = '='
114         q = projectB.query("""
115     SELECT priority.priority AS prio, section.section AS sect, override_type.type AS type
116       FROM override, priority, section, suite, override_type
117      WHERE override.priority = priority.id
118        AND override.type = override_type.id
119        AND override_type.type %s 'dsc'
120        AND override.section = section.id
121        AND override.package = %s
122        AND override.suite = suite.id
123        AND suite.suite_name = %s
124         """ % (eqdsc, pg._quote(package,"str"), pg._quote(suite,"str")))
125
126         if q.ntuples() == 0:
127             continue
128         if q.ntuples() > 1:
129             daklib.utils.fubar("%s is ambiguous. Matches %d packages" % (package,q.ntuples()))
130
131         r = q.getresult()
132         if type == 'binary':
133             oldsection = r[0][1]
134             oldpriority = r[0][0]
135         else:
136             oldsourcesection = r[0][1]
137
138     if not oldpriority and not oldsourcesection:
139         daklib.utils.fubar("Unable to find package %s" % (package))
140     if oldsection and oldsourcesection and oldsection != oldsourcesection:
141         # When setting overrides, both source & binary will become the same section
142         daklib.utils.warn("Source is in section '%s' instead of '%s'" % (oldsourcesection, oldsection))
143     if not oldsection:
144         oldsection = oldsourcesection
145
146     if not arguments:
147         if oldpriority:
148             print "%s is in section '%s' at priority '%s'" % (
149                 package,oldsection,oldpriority)
150         elif oldsourcesection:
151             # no use printing this line if also binary
152             print "%s is in section '%s'" % (
153                 package,oldsourcesection)
154         sys.exit(0)
155
156     # At this point, we have a new section and priority... check they're valid...
157     newsection, newpriority = arguments
158
159     if newsection == ".":
160         newsection = oldsection
161     if newpriority == ".":
162         newpriority = oldpriority
163
164     q = projectB.query("SELECT id FROM section WHERE section=%s" % (
165         pg._quote(newsection,"str")))
166
167     if q.ntuples() == 0:
168         daklib.utils.fubar("Supplied section %s is invalid" % (newsection))
169     newsecid = q.getresult()[0][0]
170
171     q = projectB.query("SELECT id FROM priority WHERE priority=%s" % (
172         pg._quote(newpriority,"str")))
173
174     if q.ntuples() == 0:
175         daklib.utils.fubar("Supplied priority %s is invalid" % (newpriority))
176     newprioid = q.getresult()[0][0]
177
178     if newpriority == oldpriority and newsection == oldsection:
179         print "I: Doing nothing"
180         sys.exit(0)
181
182     if newpriority and not oldpriority:
183         daklib.utils.fubar("Trying to set priority of a source-only package")
184
185     # If we're in no-action mode
186     if Options["No-Action"]:
187         if newpriority != oldpriority:
188             print "I: Would change priority from %s to %s" % (oldpriority,newpriority)
189         if newsection != oldsection:
190             print "I: Would change section from %s to %s" % (oldsection,newsection)
191         if Options.has_key("Done"):
192             print "I: Would also close bug(s): %s" % (Options["Done"])
193
194         sys.exit(0)
195
196     if newpriority != oldpriority:
197         print "I: Will change priority from %s to %s" % (oldpriority,newpriority)
198     if newsection != oldsection:
199         print "I: Will change section from %s to %s" % (oldsection,newsection)
200
201     if not Options.has_key("Done"):
202         pass
203         #daklib.utils.warn("No bugs to close have been specified. Noone will know you have done this.")
204     else:
205         print "I: Will close bug(s): %s" % (Options["Done"])
206
207     game_over()
208
209     Logger = daklib.logging.Logger(Cnf, "override")
210
211     projectB.query("BEGIN WORK")
212     # We're in "do it" mode, we have something to do... do it
213     if newpriority != oldpriority:
214         q = projectB.query("""
215         UPDATE override
216            SET priority=%d
217          WHERE package=%s
218            AND override.type != %d
219            AND suite = (SELECT id FROM suite WHERE suite_name=%s)""" % (
220             newprioid,
221             pg._quote(package,"str"), daklib.database.get_override_type_id("dsc"),
222             pg._quote(suite,"str") ))
223         Logger.log(["changed priority",package,oldpriority,newpriority])
224
225     if newsection != oldsection:
226         q = projectB.query("""
227         UPDATE override
228            SET section=%d
229          WHERE package=%s
230            AND suite = (SELECT id FROM suite WHERE suite_name=%s)""" % (
231             newsecid,
232             pg._quote(package,"str"),
233             pg._quote(suite,"str") ))
234         Logger.log(["changed section",package,oldsection,newsection])
235     projectB.query("COMMIT WORK")
236
237     if Options.has_key("Done"):
238         Subst = {}
239         Subst["__OVERRIDE_ADDRESS__"] = Cnf["Override::MyEmailAddress"]
240         Subst["__BUG_SERVER__"] = Cnf["Dinstall::BugServer"]
241         bcc = []
242         if Cnf.Find("Dinstall::Bcc") != "":
243             bcc.append(Cnf["Dinstall::Bcc"])
244         if Cnf.Find("Override::Bcc") != "":
245             bcc.append(Cnf["Override::Bcc"])
246         if bcc:
247             Subst["__BCC__"] = "Bcc: " + ", ".join(bcc)
248         else:
249             Subst["__BCC__"] = "X-Filler: 42"
250         Subst["__CC__"] = "X-DAK: dak override\nX-Katie: alicia $Revision: 1.6$"
251         Subst["__ADMIN_ADDRESS__"] = Cnf["Dinstall::MyAdminAddress"]
252         Subst["__DISTRO__"] = Cnf["Dinstall::MyDistribution"]
253         Subst["__WHOAMI__"] = daklib.utils.whoami()
254
255         summary = "Concerning package %s...\n" % (package)
256         summary += "Operating on the %s suite\n" % (suite)
257         if newpriority != oldpriority:
258             summary += "Changed priority from %s to %s\n" % (oldpriority,newpriority)
259         if newsection != oldsection:
260             summary += "Changed section from %s to %s\n" % (oldsection,newsection)
261         Subst["__SUMMARY__"] = summary
262
263         for bug in daklib.utils.split_args(Options["Done"]):
264             Subst["__BUG_NUMBER__"] = bug
265             mail_message = daklib.utils.TemplateSubst(
266                 Subst,Cnf["Dir::Templates"]+"/override.bug-close")
267             daklib.utils.send_mail(mail_message)
268             Logger.log(["closed bug",bug])
269
270     Logger.close()
271
272     print "Done"
273
274 #################################################################################
275
276 if __name__ == '__main__':
277     main()