3 """Configure dak parameters in the database"""
4 # Copyright (C) 2009 Mark Hymers <mhy@debian.org>
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.
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.
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
20 ################################################################################
26 from daklib import utils
27 from daklib.dbconn import *
28 from sqlalchemy.orm.exc import NoResultFound
30 ################################################################################
35 ################################################################################
37 print >> sys.stderr, msg
39 def die(msg, exit_code=1):
40 print >> sys.stderr, msg
43 def die_arglen(args, args_needed, msg):
44 if len(args) < args_needed:
47 def usage(exit_code=0):
48 """Perform administrative work on the dak database."""
50 print """Usage: dak admin COMMAND
51 Perform administrative work on the dak database.
53 -h, --help show this help and exit.
54 -n, --dry-run don't do anything, just show what would have been done
55 (only applies to add or rm operations).
57 Commands can use a long or abbreviated form:
61 c db-shell show db config in a usable form for psql
62 c NAME show option NAME as set in configuration table
65 k list-all list all keyrings
66 k list-binary list all keyrings with a NULL source acl
67 k list-source list all keyrings with a non NULL source acl
70 a list show a list of architectures
71 a rm ARCH remove an architecture (will only work if
72 no longer linked to any suites)
73 a add ARCH DESCRIPTION [SUITELIST]
74 add architecture ARCH with DESCRIPTION.
75 If SUITELIST is given, add to each of the
76 suites at the same time
79 s list show a list of suites
80 s show SUITE show config details for a suite
81 s add SUITE VERSION [ label=LABEL ] [ description=DESCRIPTION ]
82 [ origin=ORIGIN ] [ codename=CODENAME ]
83 [ signingkey=SIGNINGKEY ]
84 add suite SUITE, version VERSION.
85 label, description, origin, codename
86 and signingkey are optional.
88 s add-all-arches SUITE VERSION... as "s add" but adds suite-architecture
89 relationships for all architectures
91 suite-architecture / s-a:
92 s-a list show the architectures for all suites
93 s-a list-suite ARCH show the suites an ARCH is in
94 s-a list-arch SUITE show the architectures in a SUITE
95 s-a add SUITE ARCH add ARCH to suite
96 s-a rm SUITE ARCH remove ARCH from suite (will only work if
97 no packages remain for the arch in the suite)
100 v-c list show version checks for all suites
101 v-c list-suite SUITE show version checks for suite SUITE
102 v-c add SUITE CHECK REFERENCE add a version check for suite SUITE
103 v-c rm SUITE CHECK REFERENCE remove a version check
105 CHECK is one of Enhances, MustBeNewerThan, MustBeOlderThan
106 REFERENCE is another suite name
110 ################################################################################
112 def __architecture_list(d, args):
113 q = d.session().query(Architecture).order_by(Architecture.arch_string)
115 # HACK: We should get rid of source from the arch table
116 if j.arch_string == 'source': continue
120 def __architecture_add(d, args):
121 die_arglen(args, 4, "E: adding an architecture requires a name and a description")
122 print "Adding architecture %s" % args[2]
123 suites = [str(x) for x in args[4:]]
125 print "Adding to suites %s" % ", ".join(suites)
130 a.arch_string = str(args[2]).lower()
131 a.description = str(args[3])
134 su = get_suite(sn, s)
138 warn("W: Cannot find suite %s" % su)
140 except IntegrityError as e:
141 die("E: Integrity error adding architecture %s (it probably already exists)" % args[2])
142 except SQLAlchemyError as e:
143 die("E: Error adding architecture %s (%s)" % (args[2], e))
144 print "Architecture %s added" % (args[2])
146 def __architecture_rm(d, args):
147 die_arglen(args, 3, "E: removing an architecture requires at least a name")
148 print "Removing architecture %s" % args[2]
152 a = get_architecture(args[2].lower(), s)
154 die("E: Cannot find architecture %s" % args[2])
157 except IntegrityError as e:
158 die("E: Integrity error removing architecture %s (suite-arch entries probably still exist)" % args[2])
159 except SQLAlchemyError as e:
160 die("E: Error removing architecture %s (%s)" % (args[2], e))
161 print "Architecture %s removed" % args[2]
163 def architecture(command):
164 args = [str(x) for x in command]
165 Cnf = utils.get_conf()
168 die_arglen(args, 2, "E: architecture needs at least a command")
170 mode = args[1].lower()
172 __architecture_list(d, args)
174 __architecture_add(d, args)
176 __architecture_rm(d, args)
178 die("E: architecture command unknown")
180 dispatch['architecture'] = architecture
181 dispatch['a'] = architecture
183 ################################################################################
185 def __suite_list(d, args):
187 for j in s.query(Suite).order_by(Suite.suite_name).all():
190 def __suite_show(d, args):
192 die("E: showing an suite entry requires a suite")
195 su = get_suite(args[2].lower())
197 die("E: can't find suite entry for %s" % (args[2].lower()))
201 def __suite_add(d, args, addallarches=False):
202 die_arglen(args, 4, "E: adding a suite requires at least a name and a version")
203 suite_name = args[2].lower()
207 def get_field(field):
209 if varval.startswith(field + '='):
210 return varval.split('=')[1]
213 print "Adding suite %s" % suite_name
218 suite.suite_name = suite_name
219 suite.overridecodename = suite_name
220 suite.version = version
221 suite.label = get_field('label')
222 suite.description = get_field('description')
223 suite.origin = get_field('origin')
224 suite.codename = get_field('codename')
225 signingkey = get_field('signingkey')
226 if signingkey is not None:
227 suite.signingkeys = [signingkey.upper()]
228 archive_name = get_field('archive')
229 if archive_name is not None:
230 suite.archive = get_archive(archive_name, s)
232 suite.archive = s.query(Archive).filter(~Archive.archive_name.in_(['build-queues', 'new', 'policy'])).one()
233 suite.srcformats = s.query(SrcFormat).all()
236 except IntegrityError as e:
237 die("E: Integrity error adding suite %s (it probably already exists)" % suite_name)
238 except SQLAlchemyError as e:
239 die("E: Error adding suite %s (%s)" % (suite_name, e))
240 print "Suite %s added" % (suite_name)
244 q = s.query(Architecture).order_by(Architecture.arch_string)
246 suite.architectures.append(arch)
247 arches.append(arch.arch_string)
249 print "Architectures %s added to %s" % (','.join(arches), suite_name)
255 args = [str(x) for x in command]
256 Cnf = utils.get_conf()
259 die_arglen(args, 2, "E: suite needs at least a command")
261 mode = args[1].lower()
264 __suite_list(d, args)
266 __suite_show(d, args)
268 __suite_add(d, args, False)
269 elif mode == 'add-all-arches':
270 __suite_add(d, args, True)
272 die("E: suite command unknown")
274 dispatch['suite'] = suite
275 dispatch['s'] = suite
277 ################################################################################
279 def __suite_architecture_list(d, args):
281 for j in s.query(Suite).order_by(Suite.suite_name):
282 architectures = j.get_architectures(skipsrc = True, skipall = True)
283 print j.suite_name + ': ' + \
284 ', '.join([a.arch_string for a in architectures])
286 def __suite_architecture_listarch(d, args):
287 die_arglen(args, 3, "E: suite-architecture list-arch requires a suite")
288 suite = get_suite(args[2].lower(), d.session())
290 die('E: suite %s is invalid' % args[2].lower())
291 a = suite.get_architectures(skipsrc = True, skipall = True)
296 def __suite_architecture_listsuite(d, args):
297 die_arglen(args, 3, "E: suite-architecture list-suite requires an arch")
298 architecture = get_architecture(args[2].lower(), d.session())
299 if architecture is None:
300 die("E: architecture %s is invalid" % args[2].lower())
301 for j in architecture.suites:
305 def __suite_architecture_add(d, args):
307 die("E: adding a suite-architecture entry requires a suite and arch")
311 suite = get_suite(args[2].lower(), s)
312 if suite is None: die("E: Can't find suite %s" % args[2].lower())
314 arch = get_architecture(args[3].lower(), s)
315 if arch is None: die("E: Can't find architecture %s" % args[3].lower())
319 suite.architectures.append(arch)
321 except IntegrityError as e:
322 die("E: Can't add suite-architecture entry (%s, %s) - probably already exists" % (args[2].lower(), args[3].lower()))
323 except SQLAlchemyError as e:
324 die("E: Can't add suite-architecture entry (%s, %s) - %s" % (args[2].lower(), args[3].lower(), e))
326 print "Added suite-architecture entry for %s, %s" % (args[2].lower(), args[3].lower())
329 def __suite_architecture_rm(d, args):
331 die("E: removing an suite-architecture entry requires a suite and arch")
336 suite_name = args[2].lower()
337 suite = get_suite(suite_name, s)
339 die('E: no such suite %s' % suite_name)
340 arch_string = args[3].lower()
341 architecture = get_architecture(arch_string, s)
342 if architecture not in suite.architectures:
343 die("E: architecture %s not found in suite %s" % (arch_string, suite_name))
344 suite.architectures.remove(architecture)
346 except IntegrityError as e:
347 die("E: Can't remove suite-architecture entry (%s, %s) - it's probably referenced" % (args[2].lower(), args[3].lower()))
348 except SQLAlchemyError as e:
349 die("E: Can't remove suite-architecture entry (%s, %s) - %s" % (args[2].lower(), args[3].lower(), e))
351 print "Removed suite-architecture entry for %s, %s" % (args[2].lower(), args[3].lower())
354 def suite_architecture(command):
355 args = [str(x) for x in command]
356 Cnf = utils.get_conf()
359 die_arglen(args, 2, "E: suite-architecture needs at least a command")
361 mode = args[1].lower()
364 __suite_architecture_list(d, args)
365 elif mode == 'list-arch':
366 __suite_architecture_listarch(d, args)
367 elif mode == 'list-suite':
368 __suite_architecture_listsuite(d, args)
370 __suite_architecture_add(d, args)
372 __suite_architecture_rm(d, args)
374 die("E: suite-architecture command unknown")
376 dispatch['suite-architecture'] = suite_architecture
377 dispatch['s-a'] = suite_architecture
379 ################################################################################
381 def __version_check_list(d):
382 session = d.session()
383 for s in session.query(Suite).order_by(Suite.suite_name):
384 __version_check_list_suite(d, s.suite_name)
386 def __version_check_list_suite(d, suite_name):
387 vcs = get_version_checks(suite_name)
389 print "%s %s %s" % (suite_name, vc.check, vc.reference.suite_name)
391 def __version_check_add(d, suite_name, check, reference_name):
392 suite = get_suite(suite_name)
394 die("E: Could not find suite %s." % (suite_name))
395 reference = get_suite(reference_name)
397 die("E: Could not find reference suite %s." % (reference_name))
399 session = d.session()
403 vc.reference = reference
407 def __version_check_rm(d, suite_name, check, reference_name):
408 suite = get_suite(suite_name)
410 die("E: Could not find suite %s." % (suite_name))
411 reference = get_suite(reference_name)
413 die("E: Could not find reference suite %s." % (reference_name))
415 session = d.session()
417 vc = session.query(VersionCheck).filter_by(suite=suite, check=check, reference=reference).one()
420 except NoResultFound:
421 print "W: version-check not found."
423 def version_check(command):
424 args = [str(x) for x in command]
425 Cnf = utils.get_conf()
428 die_arglen(args, 2, "E: version-check needs at least a command")
429 mode = args[1].lower()
432 __version_check_list(d)
433 elif mode == 'list-suite':
435 die("E: version-check list-suite needs a single parameter")
436 __version_check_list_suite(d, args[2])
439 die("E: version-check add needs three parameters")
440 __version_check_add(d, args[2], args[3], args[4])
443 die("E: version-check rm needs three parameters")
444 __version_check_rm(d, args[2], args[3], args[4])
446 die("E: version-check command unknown")
448 dispatch['version-check'] = version_check
449 dispatch['v-c'] = version_check
451 ################################################################################
453 def show_config(command):
454 args = [str(x) for x in command]
455 cnf = utils.get_conf()
457 die_arglen(args, 2, "E: config needs at least a command")
459 mode = args[1].lower()
463 if cnf.has_key("DB::Service"):
465 connstr = "postgresql://service=%s" % cnf["DB::Service"]
466 elif cnf.has_key("DB::Host"):
468 connstr = "postgres://%s" % cnf["DB::Host"]
469 if cnf.has_key("DB::Port") and cnf["DB::Port"] != "-1":
470 connstr += ":%s" % cnf["DB::Port"]
471 connstr += "/%s" % cnf["DB::Name"]
474 connstr = "postgres:///%s" % cnf["DB::Name"]
475 if cnf["DB::Port"] and cnf["DB::Port"] != "-1":
476 connstr += "?port=%s" % cnf["DB::Port"]
478 elif mode == 'db-shell':
480 if cnf.has_key("DB::Service"):
481 e.append('PGSERVICE')
482 print "PGSERVICE=%s" % cnf["DB::Service"]
483 if cnf.has_key("DB::Name"):
484 e.append('PGDATABASE')
485 print "PGDATABASE=%s" % cnf["DB::Name"]
486 if cnf.has_key("DB::Host"):
487 print "PGHOST=%s" % cnf["DB::Host"]
489 if cnf.has_key("DB::Port") and cnf["DB::Port"] != "-1":
490 print "PGPORT=%s" % cnf["DB::Port"]
492 print "export " + " ".join(e)
494 session = DBConn().session()
496 o = session.query(DBConfig).filter_by(name = mode).one()
498 except NoResultFound:
499 print "W: option '%s' not set" % mode
501 dispatch['config'] = show_config
502 dispatch['c'] = show_config
504 ################################################################################
506 def show_keyring(command):
507 args = [str(x) for x in command]
508 cnf = utils.get_conf()
510 die_arglen(args, 2, "E: keyring needs at least a command")
512 mode = args[1].lower()
516 q = d.session().query(Keyring).filter(Keyring.active == True)
518 if mode == 'list-all':
520 elif mode == 'list-binary':
521 q = q.filter(Keyring.default_source_acl_id == None)
522 elif mode == 'list-source':
523 q = q.filter(Keyring.default_source_acl_id != None)
525 die("E: keyring command unknown")
530 dispatch['keyring'] = show_keyring
531 dispatch['k'] = show_keyring
533 ################################################################################
536 """Perform administrative work on the dak database"""
538 Cnf = utils.get_conf()
539 arguments = [('h', "help", "Admin::Options::Help"),
540 ('n', "dry-run", "Admin::Options::Dry-Run")]
541 for i in [ "help", "dry-run" ]:
542 if not Cnf.has_key("Admin::Options::%s" % (i)):
543 Cnf["Admin::Options::%s" % (i)] = ""
545 arguments = apt_pkg.parse_commandline(Cnf, arguments, sys.argv)
547 options = Cnf.subtree("Admin::Options")
548 if options["Help"] or len(arguments) < 1:
550 if options["Dry-Run"]:
553 subcommand = str(arguments[0])
555 if subcommand in dispatch.keys():
556 dispatch[subcommand](arguments)
558 die("E: Unknown command")
560 ################################################################################
562 if __name__ == '__main__':