[PATCH] Teach driverdisk.py command to reject extra partitions

["Paul W. Frields" <stickster gmail com>]

---
 pykickstart/commands/driverdisk.py |    3 +++
 1 files changed, 3 insertions(+), 0 deletions(-)

diff --git a/pykickstart/commands/driverdisk.py b/pykickstart/commands/driverdisk.py
index 2423d93..656ec06 100644
--- a/pykickstart/commands/driverdisk.py
+++ b/pykickstart/commands/driverdisk.py
@@ -74,6 +74,9 @@ class FC3_DriverDisk(KickstartCommand):
     def parse(self, args):
         (opts, extra) = self.op.parse_args(args=args)
 
+        if len(extra) > 1:
+            raise KickstartValueError, formatErrorMsg(self.lineno, msg=_("Only one partition may be specified for driverdisk command."))
+
         if len(extra) == 1 and opts.source:
             raise KickstartValueError, formatErrorMsg(self.lineno, msg=_("Only one of --source and partition may be specified for driverdisk command."))
 
-- 
1.6.0.6