mic/creator.py

   1 #!/usr/bin/python -tt
   2 #
   3 # Copyright (c) 2011 Intel, Inc.
   4 #
   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
   8 #
   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
  12 # for more details.
  13 #
  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.
  17
  18 import os, sys
  19 from optparse import SUPPRESS_HELP
  20
  21 from mic import configmgr, pluginmgr, msger
  22 from mic.utils import cmdln, errors
  23
  24 class Creator(cmdln.Cmdln):
  25     """${name}: create an image
  26
  27     usage:
  28         ${name} SUBCOMMAND [OPTS] [ARGS..]
  29
  30     ${command_list}
  31     ${option_list}
  32     """
  33
  34     name = 'mic create(cr)'
  35
  36     def __init__(self, *args, **kwargs):
  37         cmdln.Cmdln.__init__(self, *args, **kwargs)
  38
  39         # load configmgr
  40         self.configmgr = configmgr.getConfigMgr()
  41
  42         # load pluginmgr
  43         self.pluginmgr = pluginmgr.PluginMgr()
  44         self.plugincmds = self.pluginmgr.get_plugins('imager')
  45
  46         # mix-in do_subcmd interface
  47         for subcmd, klass in self.plugincmds.iteritems():
  48             if not hasattr(klass, 'do_create'):
  49                 msger.warning("Unsurpport subcmd: %s" % subcmd)
  50                 continue
  51
  52             func = getattr(klass, 'do_create')
  53             setattr(self.__class__, "do_"+subcmd, func)
  54
  55     def get_optparser(self):
  56         optparser = cmdln.CmdlnOptionParser(self)
  57         optparser.add_option('-d', '--debug', action='store_true', dest='debug', help=SUPPRESS_HELP)
  58         optparser.add_option('-v', '--verbose', action='store_true', dest='verbose', help=SUPPRESS_HELP)
  59         optparser.add_option('', '--logfile', type='string', dest='logfile', default=None, help='Path of logfile')
  60         optparser.add_option('-c', '--config', type='string', dest='config', default=None, help='Specify config file for mic')
  61         optparser.add_option('-o', '--outdir', type='string', action='store', dest='outdir', default=None, help='Output directory')
  62         optparser.add_option('-A', '--arch', type='string', dest='arch', default=None, help='Specify repo architecture')
  63         optparser.add_option('', '--release', type='string', dest='release', default=None, metavar='RID', help='Generate a release of RID with all neccessary files ')
  64         optparser.add_option("", "--record-pkgs", type="string", dest="record_pkgs", default=None, help="Record the installed packages, valid values: name, content, license")
  65         optparser.add_option('', '--pkgmgr', type='string', dest='pkgmgr', default=None, help='Specify backend package manager')
  66         optparser.add_option('', '--local-pkgs-path', type='string', dest='local_pkgs_path', default=None, help='Path for local pkgs(rpms) to be installed')
  67         return optparser
  68
  69     def preoptparse(self, argv):
  70         optparser = self.get_optparser()
  71
  72         largs = []
  73         rargs = []
  74         while argv:
  75             arg = argv.pop(0)
  76
  77             if arg in ('-h', '--help'):
  78                 rargs.append(arg)
  79
  80             elif optparser.has_option(arg):
  81                 largs.append(arg)
  82
  83                 if optparser.get_option(arg).takes_value():
  84                     try:
  85                         largs.append(argv.pop(0))
  86                     except IndexError:
  87                         raise errors.Usage("%s option requires an argument" % arg)
  88
  89             else:
  90                 if arg.startswith("--"):
  91                     if "=" in arg:
  92                         opt = arg.split("=")[0]
  93                     else:
  94                         opt = None
  95                 elif arg.startswith("-") and len(arg) > 2:
  96                     opt = arg[0:2]
  97                 else:
  98                     opt = None
  99
 100                 if opt and optparser.has_option(opt):
 101                     largs.append(arg)
 102                 else:
 103                     rargs.append(arg)
 104
 105         return largs + rargs
 106
 107     def postoptparse(self):
 108         if self.options.verbose:
 109             msger.set_loglevel('verbose')
 110         if self.options.debug:
 111             msger.set_loglevel('debug')
 112
 113         if self.options.config:
 114             self.configmgr.reset()
 115             self.configmgr._siteconf = self.options.config
 116
 117         if self.options.outdir is not None:
 118             self.configmgr.create['outdir'] = self.options.outdir
 119         if self.options.local_pkgs_path is not None:
 120             self.configmgr.create['local_pkgs_path'] = self.options.local_pkgs_path
 121
 122         if self.options.release:
 123             self.configmgr.create['release'] = self.options.release
 124
 125         if self.options.record_pkgs:
 126             self.configmgr.create['record_pkgs'] = []
 127             for type in self.options.record_pkgs.split(','):
 128                 if type not in ('name', 'content', 'license'):
 129                     msger.error("Please specify what to be recorded: \"name\", \"content\", \"license\"")
 130                 self.configmgr.create['record_pkgs'].append(type)
 131
 132         if self.options.arch is not None:
 133             self.configmgr.create['arch'] = self.options.arch
 134
 135         if self.options.pkgmgr is not None:
 136             self.configmgr.create['pkgmgr'] = self.options.pkgmgr
 137
 138         if self.options.logfile:
 139             msger.set_interactive(False)
 140             msger.set_logfile(self.options.logfile)
 141
 142     def main(self, argv=None):
 143         if argv is None:
 144             argv = sys.argv
 145         else:
 146             argv = argv[:] # don't modify caller's list
 147
 148         self.optparser = self.get_optparser()
 149         if self.optparser:
 150             try:
 151                 argv = self.preoptparse(argv)
 152                 self.options, args = self.optparser.parse_args(argv)
 153
 154             except cmdln.CmdlnUserError, ex:
 155                 msg = "%s: %s\nTry '%s help' for info.\n"\
 156                       % (self.name, ex, self.name)
 157                 msger.error(msg)
 158
 159             except cmdln.StopOptionProcessing, ex:
 160                 return 0
 161         else:
 162             # optparser=None means no process for opts
 163             self.options, args = None, argv[1:]
 164
 165         self.postoptparse()
 166
 167         if not args:
 168             return self.emptyline()
 169
 170         if os.geteuid() != 0:
 171             msger.error('Root permission is required to continue, abort')
 172
 173         return self.cmd(args)
 174