pykickstart/commands/key.py

   1 #
   2 # Chris Lumens <clumens@redhat.com>
   3 #
   4 # Copyright 2007 Red Hat, Inc.
   5 #
   6 # This copyrighted material is made available to anyone wishing to use, modify,
   7 # copy, or redistribute it subject to the terms and conditions of the GNU
   8 # General Public License v.2.  This program is distributed in the hope that it
   9 # will be useful, but WITHOUT ANY WARRANTY expressed or implied, including the
  10 # implied warranties of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  11 # See the GNU General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU General Public License along with
  14 # this program; if not, write to the Free Software Foundation, Inc., 51
  15 # Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  Any Red Hat
  16 # trademarks that are incorporated in the source code or documentation are not
  17 # subject to the GNU General Public License and may only be used or replicated
  18 # with the express permission of Red Hat, Inc.
  19 #
  20 from pykickstart.base import KickstartCommand
  21 from pykickstart.constants import KS_INSTKEY_SKIP
  22 from pykickstart.errors import KickstartValueError, formatErrorMsg
  23 from pykickstart.options import KSOptionParser
  24
  25 import gettext
  26 _ = lambda x: gettext.ldgettext("pykickstart", x)
  27
  28 class RHEL5_Key(KickstartCommand):
  29     removedKeywords = KickstartCommand.removedKeywords
  30     removedAttrs = KickstartCommand.removedAttrs
  31
  32     def __init__(self, writePriority=0, *args, **kwargs):
  33         KickstartCommand.__init__(self, writePriority, *args, **kwargs)
  34         self.op = self._getParser()
  35         self.key = kwargs.get("key", "")
  36         self.skip = kwargs.get("skip", False)
  37
  38     def __str__(self):
  39         retval = KickstartCommand.__str__(self)
  40
  41         if self.key == KS_INSTKEY_SKIP:
  42             retval += "key --skip\n"
  43         elif self.key != "":
  44             retval += "key %s\n" % self.key
  45
  46         return retval
  47
  48     def _getParser(self):
  49         op = KSOptionParser()
  50         op.add_option("--skip", action="store_true", default=False)
  51         return op
  52
  53     def parse(self, args):
  54         (opts, extra) = self.op.parse_args(args=args, lineno=self.lineno)
  55         self._setToSelf(self.op, opts)
  56
  57         if self.skip:
  58             self.key = KS_INSTKEY_SKIP
  59         elif len(extra) != 1:
  60             raise KickstartValueError(formatErrorMsg(self.lineno, msg=_("Kickstart command %s requires one argument") % "key"))
  61         else:
  62             self.key = extra[0]
  63
  64         return self