X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=dak%2Fcontrol_suite.py;h=f648b5ece88ca2032fe2d3db2839161978f83580;hb=cd5b29ddfd8de263c085f494b9573d683913f6f3;hp=63a0386b1ed0f26f14cc5033c2055defdee09e57;hpb=1408d82381556c9e2b3eaa04412846c766a011db;p=dak.git

diff --git a/dak/control_suite.py b/dak/control_suite.py
old mode 100644
new mode 100755
index 63a0386b..f648b5ec
--- a/dak/control_suite.py
+++ b/dak/control_suite.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Manipulate suite tags
+""" Manipulate suite tags """
 # Copyright (C) 2000, 2001, 2002, 2003, 2004, 2005, 2006  James Troup <james@nocrew.org>
 
 # This program is free software; you can redistribute it and/or modify
@@ -43,9 +43,9 @@
 
 import pg, sys
 import apt_pkg
-import daklib.database as database
-import daklib.logging as logging
-import daklib.utils as utils
+from daklib import database
+from daklib import logging
+from daklib import utils
 
 #######################################################################################
 
@@ -244,7 +244,11 @@ def main ():
         if not Cnf.has_key("Control-Suite::Options::%s" % (i)):
             Cnf["Control-Suite::Options::%s" % (i)] = ""
 
-    file_list = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv)
+    try:
+        file_list = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
+    except SystemError, e:
+        print "%s\n" % e
+        usage(1)
     Options = Cnf.SubTree("Control-Suite::Options")
 
     if Options["Help"]:
@@ -271,8 +275,8 @@ def main ():
         utils.fubar("No action specified.")
 
     # Safety/Sanity check
-    if action == "set" and suite not in ["testing", "etch-m68k"]:
-        utils.fubar("Will not reset a suite other than testing.")
+    if action == "set" and suite not in ["testing", "etch-m68k", "lenny-r0"]:
+        utils.fubar("Will not reset suite %s" % (suite))
 
     if action == "list":
         get_list(suite)