pykickstart/commands/autopart.py

   1 #
   2 # Chris Lumens <clumens@redhat.com>
   3 #
   4 # Copyright 2005, 2006, 2007, 2008 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 imgcreate.pykickstart.base import *
  21 from imgcreate.pykickstart.errors import *
  22 from imgcreate.pykickstart.options import *
  23
  24 import gettext
  25 _ = lambda x: gettext.ldgettext("pykickstart", x)
  26
  27 class FC3_AutoPart(KickstartCommand):
  28     def __init__(self, writePriority=100, autopart=False):
  29         KickstartCommand.__init__(self, writePriority)
  30         self.autopart = autopart
  31
  32     def __str__(self):
  33         if self.autopart:
  34             return "autopart\n"
  35         else:
  36             return ""
  37
  38     def parse(self, args):
  39         if len(args) > 0:
  40             raise KickstartValueError, formatErrorMsg(self.lineno, msg=_("Kickstart command %s does not take any arguments") % "autopart")
  41
  42         self.autopart = True
  43
  44 class F9_AutoPart(FC3_AutoPart):
  45     def __init__(self, writePriority=100, autopart=False, encrypted=False,
  46                  passphrase=""):
  47         FC3_AutoPart.__init__(self, writePriority=writePriority, autopart=autopart)
  48         self.op = self._getParser()
  49
  50         self.encrypted = encrypted
  51         self.passphrase = passphrase
  52
  53     def __str__(self):
  54         if self.autopart:
  55             retval = "autopart"
  56         else:
  57             return ""
  58
  59         if self.encrypted:
  60             retval += " --encrypted"
  61
  62             if self.passphrase != "":
  63                 retval += " --passphrase=\"%s\""% self.passphrase
  64
  65         return retval + "\n"
  66
  67     def _getParser(self):
  68         op = KSOptionParser(lineno=self.lineno)
  69         op.add_option("--encrypted", action="store_true", default=False)
  70         op.add_option("--passphrase")
  71         return op
  72
  73     def parse(self, args):
  74         (opts, extra) = self.op.parse_args(args=args)
  75         self._setToSelf(self.op, opts)
  76         self.autopart = True