[dak/master] add basic dak admin command to test out sqla work
Signed-off-by: Mark Hymers <mhy@debian.org>
---
dak/admin.py | 273 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
dak/dak.py | 2 +
2 files changed, 275 insertions(+), 0 deletions(-)
create mode 100755 dak/admin.py
diff --git a/dak/admin.py b/dak/admin.py
new file mode 100755
index 0000000..2bf6161
--- /dev/null
+++ b/dak/admin.py
@@ -0,0 +1,273 @@
+#!/usr/bin/env python
+
+"""Configure dak parameters in the database"""
+# Copyright (C) 2009 Mark Hymers <mhy@debian.org>
+
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+
+################################################################################
+
+import sys
+
+import apt_pkg
+
+from daklib import utils
+from daklib.dbconn import *
+from daklib.config import Config
+
+################################################################################
+
+dispatch = {}
+dryrun = False
+
+################################################################################
+def warn(msg):
+ print >> sys.stderr, msg
+
+def die(msg, exit_code=1):
+ print >> sys.stderr, msg
+ sys.exit(exit_code)
+
+def die_arglen(args, args_needed, msg):
+ if len(args) < args_needed:
+ die(msg)
+
+def usage(exit_code=0):
+ """Perform administrative work on the dak database."""
+
+ print """Usage: dak admin COMMAND
+Perform administrative work on the dak database.
+
+ -h, --help show this help and exit.
+ -n, --dry-run don't do anything, just show what would have been done
+ (only applies to add or rm operations).
+
+ Commands can use a long or abbreviated form:
+
+ architecture / a:
+ a list show a list of architectures
+ a rm ARCH remove an architecture (will only work if
+ no longer linked to any suites)
+ a add ARCH DESCRIPTION [SUITELIST]
+ add architecture ARCH with DESCRIPTION.
+ If SUITELIST is given, add to each of the
+ suites at the same time
+
+ suite-architecture / s-a:
+ s-a list-suite ARCH show the suites an ARCH is in
+ s-a list-arch SUITE show the architectures in a SUITE
+ s-a add SUITE ARCH add ARCH to suite
+ s-a rm SUITE ARCH remove ARCH from suite (will only work if
+ no packages remain for the arch in the suite)
+"""
+ sys.exit(exit_code)
+
+################################################################################
+
+def __architecture_list(d, args):
+ q = d.session().query(Architecture).order_by('arch_string')
+ for j in q.all():
+ # HACK: We should get rid of source from the arch table
+ if j.arch_string == 'source': continue
+ print j.arch_string
+ sys.exit(0)
+
+def __architecture_add(d, args):
+ die_arglen(args, 3, "E: adding an architecture requires a name and a description")
+ print "Adding architecture %s" % args[2]
+ suites = [str(x) for x in args[4:]]
+ print suites
+ if not dryrun:
+ try:
+ s = d.session()
+ s.begin()
+ a = Architecture()
+ a.arch_string = str(args[2]).lower()
+ a.description = str(args[3])
+ s.add(a)
+ for sn in suites:
+ su = get_suite(sn ,s)
+ if su is not None:
+ archsu = SuiteArchitecture()
+ archsu.arch_id = a.arch_id
+ archsu.suite_id = su.suite_id
+ s.add(archsu)
+ else:
+ warn("W: Cannot find suite %s" % su)
+ s.commit()
+ except IntegrityError, e:
+ die("E: Integrity error adding architecture %s (it probably already exists)" % args[2])
+ except SQLAlchemyError, e:
+ die("E: Error adding architecture %s (%s)" % (args[2], e))
+ print "Architecture %s added" % (args[2])
+
+def __architecture_rm(d, args):
+ die_arglen(args, 3, "E: removing an architecture requires at least a name")
+ print "Removing architecture %s" % args[2]
+ if not dryrun:
+ try:
+ s = d.session()
+ s.begin()
+ a = get_architecture(args[2].lower(), s)
+ if a is None:
+ die("E: Cannot find architecture %s" % args[2])
+ s.delete(a)
+ s.commit()
+ except IntegrityError, e:
+ die("E: Integrity error removing architecture %s (suite-arch entries probably still exist)" % args[2])
+ except SQLAlchemyError, e:
+ die("E: Error removing architecture %s (%s)" % (args[2], e))
+ print "Architecture %s removed" % args[2]
+
+def architecture(command):
+ args = [str(x) for x in command]
+ Cnf = utils.get_conf()
+ d = DBConn()
+
+ die_arglen(args, 2, "E: architecture needs at least a command")
+
+ mode = args[1].lower()
+ if mode == 'list':
+ __architecture_list(d, args)
+ elif mode == 'add':
+ __architecture_add(d, args)
+ elif mode == 'rm':
+ __architecture_rm(d, args)
+ else:
+ die("E: architecture command unknown")
+
+dispatch['architecture'] = architecture
+dispatch['a'] = architecture
+
+################################################################################
+
+def __suite_architecture_listarch(d, args):
+ die_arglen(args, 3, "E: suite-architecture list-arch requires a suite")
+ a = get_suite_architectures(args[2].lower())
+ for j in a:
+ # HACK: We should get rid of source from the arch table
+ if j.arch_string != 'source':
+ print j.arch_string
+
+
+def __suite_architecture_listsuite(d, args):
+ die_arglen(args, 3, "E: suite-architecture list-suite requires an arch")
+ for j in get_architecture_suites(args[2].lower()):
+ print j.suite_name
+
+
+def __suite_architecture_add(d, args):
+ if len(args) < 3:
+ die("E: adding a suite-architecture entry requires a suite and arch")
+
+ s = d.session()
+
+ suite = get_suite(args[2].lower(), s)
+ if suite is None: die("E: Can't find suite %s" % args[2].lower())
+
+ arch = get_architecture(args[3].lower(), s)
+ if arch is None: die("E: Can't find architecture %s" % args[3].lower())
+
+ if not dryrun:
+ try:
+ s.begin()
+ sa = SuiteArchitecture()
+ sa.arch_id = arch.arch_id
+ sa.suite_id = suite.suite_id
+ s.add(sa)
+ s.commit()
+ except IntegrityError, e:
+ die("E: Can't add suite-architecture entry (%s, %s) - probably already exists" % (args[2].lower(), args[3].lower()))
+ except SQLAlchemyError, e:
+ die("E: Can't add suite-architecture entry (%s, %s) - %s" % (args[2].lower(), args[3].lower(), e))
+
+ print "Added suite-architecture entry for %s, %s" % (args[2].lower(), args[3].lower())
+
+
+def __suite_architecture_rm(d, args):
+ if len(args) < 3:
+ die("E: removing an suite-architecture entry requires a suite and arch")
+
+ s = d.session()
+ if not dryrun:
+ try:
+ s.begin()
+ sa = get_suite_architecture(args[2].lower(), args[3].lower(), s)
+ if sa is None:
+ die("E: can't find suite-architecture entry for %s, %s" % (args[2].lower(), args[3].lower()))
+ s.delete(sa)
+ s.commit()
+ except IntegrityError, e:
+ die("E: Can't remove suite-architecture entry (%s, %s) - it's probably referenced" % (args[2].lower(), args[3].lower()))
+ except SQLAlchemyError, e:
+ die("E: Can't remove suite-architecture entry (%s, %s) - %s" % (args[2].lower(), args[3].lower(), e))
+
+ print "Removed suite-architecture entry for %s, %s" % (args[2].lower(), args[3].lower())
+
+
+def suite_architecture(command):
+ args = [str(x) for x in command]
+ Cnf = utils.get_conf()
+ d = DBConn()
+
+ die_arglen(args, 2, "E: suite-architecture needs at least a command")
+
+ mode = args[1].lower()
+
+ if mode == 'list-arch':
+ __suite_architecture_listarch(d, args)
+ elif mode == 'list-suite':
+ __suite_architecture_listsuite(d, args)
+ elif mode == 'add':
+ __suite_architecture_add(d, args)
+ elif mode == 'rm':
+ __suite_architecture_rm(d, args)
+ else:
+ die("E: suite-architecture command unknown")
+
+dispatch['suite-architecture'] = suite_architecture
+dispatch['s-a'] = suite_architecture
+
+################################################################################
+
+def main():
+ """Perform administrative work on the dak database"""
+ global dryrun
+ Cnf = utils.get_conf()
+ arguments = [('h', "help", "Admin::Options::Help"),
+ ('n', "dry-run", "Admin::Options::Dry-Run")]
+ for i in [ "help", "dry-run" ]:
+ if not Cnf.has_key("Admin::Options::%s" % (i)):
+ Cnf["Admin::Options::%s" % (i)] = ""
+
+ arguments = apt_pkg.ParseCommandLine(Cnf, arguments, sys.argv)
+
+ options = Cnf.SubTree("Admin::Options")
+ if options["Help"] or len(arguments) < 1:
+ usage()
+ if options["Dry-Run"]:
+ dryrun = True
+
+ subcommand = str(arguments[0])
+
+ if subcommand in dispatch.keys():
+ dispatch[subcommand](arguments)
+ else:
+ die("E: Unknown command")
+
+################################################################################
+
+if __name__ == '__main__':
+ main()
diff --git a/dak/dak.py b/dak/dak.py
index 6174949..c8a6a19 100755
--- a/dak/dak.py
+++ b/dak/dak.py
@@ -153,6 +153,8 @@ def init():
"Syncs fingerprint and uid tables with Debian LDAP db"),
("import-users-from-passwd",
"Sync PostgreSQL users with passwd file"),
+ ("admin",
+ "Perform administration on the dak database"),
("init-db",
"Update the database to match the conf file"),
("update-db",
--
1.5.6.5
Reply to: