pykickstart/commands/multipath.py

   1 #
   2 # Chris Lumens <clumens@redhat.com>
   3 # Peter Jones <pjones@redhat.com>
   4 #
   5 # Copyright 2006, 2007 Red Hat, Inc.
   6 #
   7 # This copyrighted material is made available to anyone wishing to use, modify,
   8 # copy, or redistribute it subject to the terms and conditions of the GNU
   9 # General Public License v.2.  This program is distributed in the hope that it
  10 # will be useful, but WITHOUT ANY WARRANTY expressed or implied, including the
  11 # implied warranties of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  12 # See the GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License along with
  15 # this program; if not, write to the Free Software Foundation, Inc., 51
  16 # Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  Any Red Hat
  17 # trademarks that are incorporated in the source code or documentation are not
  18 # subject to the GNU General Public License and may only be used or replicated
  19 # with the express permission of Red Hat, Inc.
  20 #
  21 from pykickstart.base import BaseData, KickstartCommand
  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 FC6_MpPathData(BaseData):
  29     removedKeywords = BaseData.removedKeywords
  30     removedAttrs = BaseData.removedAttrs
  31
  32     def __init__(self, *args, **kwargs):
  33         BaseData.__init__(self, *args, **kwargs)
  34         self.mpdev = kwargs.get("mpdev", "")
  35         self.device = kwargs.get("device", "")
  36         self.rule = kwargs.get("rule", "")
  37
  38     def __str__(self):
  39         return " --device=%s --rule=\"%s\"" % (self.device, self.rule)
  40
  41 class FC6_MultiPathData(BaseData):
  42     removedKeywords = BaseData.removedKeywords
  43     removedAttrs = BaseData.removedAttrs
  44
  45     def __init__(self, *args, **kwargs):
  46         BaseData.__init__(self, *args, **kwargs)
  47         self.name = kwargs.get("name", "")
  48         self.paths = kwargs.get("paths", [])
  49
  50     def __str__(self):
  51         retval = BaseData.__str__(self)
  52
  53         for path in self.paths:
  54             retval += "multipath --mpdev=%s %s\n" % (self.name, path.__str__())
  55
  56         return retval
  57
  58 class FC6_MultiPath(KickstartCommand):
  59     removedKeywords = KickstartCommand.removedKeywords
  60     removedAttrs = KickstartCommand.removedAttrs
  61
  62     def __init__(self, writePriority=50, *args, **kwargs):
  63         KickstartCommand.__init__(self, writePriority, *args, **kwargs)
  64         self.op = self._getParser()
  65
  66         self.mpaths = kwargs.get("mpaths", [])
  67
  68     def __str__(self):
  69         retval = ""
  70         for mpath in self.mpaths:
  71             retval += mpath.__str__()
  72
  73         return retval
  74
  75     def _getParser(self):
  76         op = KSOptionParser()
  77         op.add_option("--name", dest="name", action="store", type="string",
  78                       required=1)
  79         op.add_option("--device", dest="device", action="store", type="string",
  80                       required=1)
  81         op.add_option("--rule", dest="rule", action="store", type="string",
  82                       required=1)
  83         return op
  84
  85     def parse(self, args):
  86         (opts, _extra) = self.op.parse_args(args=args, lineno=self.lineno)
  87         dd = FC6_MpPathData()
  88         self._setToObj(self.op, opts, dd)
  89         dd.lineno = self.lineno
  90         dd.mpdev = dd.mpdev.split('/')[-1]
  91
  92         parent = None
  93         for x in range(0, len(self.mpaths)):
  94             mpath = self.mpaths[x]
  95             for path in mpath.paths:
  96                 if path.device == dd.device:
  97                     mapping = {"device": path.device, "multipathdev": path.mpdev}
  98                     raise KickstartValueError(formatErrorMsg(self.lineno, msg=_("Device '%(device)s' is already used in multipath '%(multipathdev)s'") % mapping))
  99             if mpath.name == dd.mpdev:
 100                 parent = x
 101
 102         if parent is None:
 103             mpath = FC6_MultiPathData()
 104             return mpath
 105         else:
 106             mpath = self.mpaths[parent]
 107
 108         return dd
 109
 110     def dataList(self):
 111         return self.mpaths