3 # Copyright (c) 2011 Intel, Inc.
5 # This program is free software; you can redistribute it and/or modify it
6 # under the terms of the GNU General Public License as published by the Free
7 # Software Foundation; version 2 of the License
9 # This program is distributed in the hope that it will be useful, but
10 # WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11 # or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 # You should have received a copy of the GNU General Public License along
15 # with this program; if not, write to the Free Software Foundation, Inc., 59
16 # Temple Place - Suite 330, Boston, MA 02111-1307, USA.
23 from .utils import misc, runner, errors
25 DEFAULT_GSITECONF = '/etc/mic/mic.conf'
27 class ConfigMgr(object):
28 DEFAULTS = {'common': {},
30 "tmpdir": '/var/tmp/mic',
31 "cachedir": '/var/tmp/mic/cache',
32 "outdir": './mic-output',
33 "arch": None, # None means auto-detect
39 "local_pkgs_path": None,
43 "compress_disk_image": None,
44 "distro_name": "Default Distribution",
51 # make the manager class as singleton
53 def __new__(cls, *args, **kwargs):
55 cls._instance = super(ConfigMgr, cls).__new__(cls, *args, **kwargs)
59 def __init__(self, ksconf=None, siteconf=None):
60 # reset config options
64 # initial options from siteconf
65 self._siteconf = DEFAULT_GSITECONF
69 self.__siteconf = None
71 # initialize the values with defaults
72 for sec, vals in self.DEFAULTS.iteritems():
73 setattr(self, sec, vals)
75 def __set_siteconf(self, siteconf):
77 self.__siteconf = siteconf
78 self._parse_siteconf(siteconf)
79 except ConfigParser.Error, error:
80 raise errors.ConfigError("%s" % error)
81 def __get_siteconf(self):
82 return self.__siteconf
83 _siteconf = property(__get_siteconf, __set_siteconf)
85 def __set_ksconf(self, ksconf):
86 if not os.path.isfile(ksconf):
87 msger.error('Cannot find ks file: %s' % ksconf)
89 self.__ksconf = ksconf
90 self._parse_kickstart(ksconf)
91 def __get_ksconf(self):
93 _ksconf = property(__get_ksconf, __set_ksconf)
95 def _parse_siteconf(self, siteconf):
99 if not os.path.exists(siteconf):
100 raise errors.ConfigError("Failed to find config file: %s" % siteconf)
102 parser = ConfigParser.SafeConfigParser()
103 parser.read(siteconf)
105 for section in parser.sections():
106 if section in self.DEFAULTS.keys():
107 getattr(self, section).update(dict(parser.items(section)))
109 def _selinux_check(self, arch, ks):
110 """ If a user needs to use btrfs or creates ARM image, selinux must be disabled at start """
112 for path in ["/usr/sbin/getenforce",
113 "/usr/bin/getenforce",
116 "/usr/local/sbin/getenforce",
117 "/usr/locla/bin/getenforce"
119 if os.path.exists(path):
120 selinux_status = runner.outs([path])
121 if arch and arch.startswith("arm") and selinux_status == "Enforcing":
122 raise errors.ConfigError("Can't create arm image if selinux is enabled, please disable it and try again")
125 parts = ks.handler.partition.partitions
126 for part in ks.handler.partition.partitions:
127 if part.fstype == "btrfs":
131 if use_btrfs and selinux_status == "Enforcing":
132 raise errors.ConfigError("Can't create image useing btrfs filesystem if selinux is enabled, please disable it and try again")
136 def _parse_kickstart(self, ksconf=None):
140 ks = kickstart.read_kickstart(ksconf)
142 self.create['ks'] = ks
143 self.create['name'] = os.path.splitext(os.path.basename(ksconf))[0]
145 if self.create['name_prefix']:
146 self.create['name'] = "%s-%s" % (self.create['name_prefix'], self.create['name'])
148 self._selinux_check (self.create['arch'], ks)
150 msger.info("Retrieving repo metadata:")
151 ksrepos = misc.get_repostrs_from_ks(ks)
152 self.create['repomd'] = misc.get_metadata_from_repos(ksrepos, self.create['cachedir'])
155 target_archlist, archlist = misc.get_arch(self.create['repomd'])
156 if self.create['arch']:
157 if self.create['arch'] not in archlist:
158 raise errors.ConfigError("Invalid arch %s for repository. Valid arches: %s"\
159 % (self.create['arch'], ', '.join(archlist)))
161 if len(target_archlist) == 1:
162 self.create['arch'] = str(target_archlist[0])
163 msger.info("\nUse detected arch %s." % target_archlist[0])
165 raise errors.ConfigError("Please specify a valid arch, "\
166 "your choise can be: %s" % ', '.join(archlist))
168 kickstart.resolve_groups(self.create, self.create['repomd'])
170 configmgr = ConfigMgr()