mirror of
git://git.yoctoproject.org/poky.git
synced 2025-07-19 21:09:03 +02:00
8d34eea303
This is a preparation for 'include' support. Used unique counter instead of line number for partitions in .ks file. Line numbers can be equal for different .ks files, which can cause problems if one .ks file is included into another. (From OE-Core rev: cc2233b51f1d22d4e540f4a3e9ceedd7ede9ffa9) Signed-off-by: Ed Bartosh <ed.bartosh@linux.intel.com> Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
142 lines
4.8 KiB
Python
142 lines
4.8 KiB
Python
#!/usr/bin/env python -tt
|
|
# ex:ts=4:sw=4:sts=4:et
|
|
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
|
|
#
|
|
# Copyright (c) 2016 Intel, Inc.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it
|
|
# under the terms of the GNU General Public License as published by the Free
|
|
# Software Foundation; version 2 of the License
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
# or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
# for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with this program; if not, write to the Free Software Foundation, Inc., 59
|
|
# Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
#
|
|
# DESCRIPTION
|
|
# This module provides parser for kickstart format
|
|
#
|
|
# AUTHORS
|
|
# Tom Zanussi <tom.zanussi (at] linux.intel.com>
|
|
# Ed Bartosh <ed.bartosh> (at] linux.intel.com>
|
|
|
|
|
|
|
|
import shlex
|
|
from argparse import ArgumentParser, ArgumentError, ArgumentTypeError
|
|
|
|
from wic.partition import Partition
|
|
|
|
class KickStartError(Exception):
|
|
pass
|
|
|
|
class KickStartParser(ArgumentParser):
|
|
"""
|
|
This class overwrites error method to throw exception
|
|
instead of producing usage message(default argparse behavior).
|
|
"""
|
|
def error(self, message):
|
|
raise ArgumentError(None, message)
|
|
|
|
def sizetype(arg):
|
|
"""
|
|
Custom type for ArgumentParser
|
|
Converts size string in <num>[K|k|M|G] format into the integer value
|
|
"""
|
|
if arg.isdigit():
|
|
return int(arg) * 1024L
|
|
|
|
if not arg[:-1].isdigit():
|
|
raise ArgumentTypeError("Invalid size: %r" % arg)
|
|
|
|
size = int(arg[:-1])
|
|
if arg.endswith("k") or arg.endswith("K"):
|
|
return size
|
|
if arg.endswith("M"):
|
|
return size * 1024L
|
|
if arg.endswith("G"):
|
|
return size * 1024L * 1024L
|
|
|
|
raise ArgumentTypeError("Invalid size: %r" % arg)
|
|
|
|
def overheadtype(arg):
|
|
"""
|
|
Custom type for ArgumentParser
|
|
Converts overhead string to float and checks if it's bigger than 1.0
|
|
"""
|
|
try:
|
|
result = float(arg)
|
|
except ValueError:
|
|
raise ArgumentTypeError("Invalid value: %r" % arg)
|
|
|
|
if result < 1.0:
|
|
raise ArgumentTypeError("Overhead factor should be > 1.0" % arg)
|
|
|
|
return result
|
|
|
|
class KickStart(object):
|
|
def __init__(self, confpath):
|
|
|
|
self.partitions = []
|
|
self.bootloader = None
|
|
self.lineno = 0
|
|
self.partnum = 0
|
|
|
|
parser = KickStartParser()
|
|
subparsers = parser.add_subparsers()
|
|
|
|
part = subparsers.add_parser('part')
|
|
part.add_argument('mountpoint')
|
|
part.add_argument('--active', action='store_true')
|
|
part.add_argument('--align', type=int)
|
|
part.add_argument("--extra-space", type=sizetype, default=10*1024L)
|
|
part.add_argument('--fsoptions', dest='fsopts')
|
|
part.add_argument('--fstype')
|
|
part.add_argument('--label')
|
|
part.add_argument('--no-table')
|
|
part.add_argument('--ondisk', '--ondrive', dest='disk')
|
|
part.add_argument("--overhead-factor", type=overheadtype, default=1.3)
|
|
part.add_argument('--part-type')
|
|
part.add_argument('--rootfs-dir')
|
|
part.add_argument('--size', type=sizetype, default=0)
|
|
part.add_argument('--source')
|
|
part.add_argument('--sourceparams')
|
|
part.add_argument('--use-uuid', action='store_true')
|
|
part.add_argument('--uuid')
|
|
|
|
bootloader = subparsers.add_parser('bootloader')
|
|
bootloader.add_argument('--append')
|
|
bootloader.add_argument('--configfile')
|
|
bootloader.add_argument('--ptable', choices=('msdos', 'gpt'),
|
|
default='msdos')
|
|
bootloader.add_argument('--timeout', type=int)
|
|
bootloader.add_argument('--source')
|
|
|
|
self._parse(parser, confpath)
|
|
|
|
def _parse(self, parser, confpath):
|
|
with open(confpath) as conf:
|
|
lineno = 0
|
|
for line in conf:
|
|
line = line.strip()
|
|
lineno += 1
|
|
if line and line[0] != '#':
|
|
try:
|
|
parsed = parser.parse_args(shlex.split(line))
|
|
except ArgumentError as err:
|
|
raise KickStartError('%s:%d: %s' % \
|
|
(confpath, lineno, err))
|
|
if line.startswith('part'):
|
|
self.partnum += 1
|
|
self.partitions.append(Partition(parsed, self.partnum))
|
|
else:
|
|
if not self.bootloader:
|
|
self.bootloader = parsed
|
|
else:
|
|
raise KickStartError("%s:%d: more than one bootloader "\
|
|
"specified" % (confpath, lineno))
|