setup.py

   1 #!/usr/bin/env python
   2
   3 import os, sys
   4 import glob
   5 from distutils.core import setup
   6 try:
   7     import setuptools
   8     # enable "setup.py develop", optional
   9 except ImportError:
  10     pass
  11
  12 MOD_NAME = 'mic'
  13
  14 version_path = 'VERSION'
  15 if not os.path.isfile(version_path):
  16     print 'No VERSION file in topdir, abort'
  17     sys.exit(1)
  18
  19 try:
  20     # first line should be the version number
  21     version = open(version_path).readline().strip()
  22     if not version:
  23         print 'VERSION file is invalid, abort'
  24         sys.exit(1)
  25
  26     ver_file = open('%s/__version__.py' % MOD_NAME, 'w')
  27     ver_file.write("VERSION = \"%s\"\n" % version)
  28     ver_file.close()
  29 except IOError:
  30     print 'WARNING: Cannot write version number file'
  31
  32 # --install-layout is recognized after 2.5
  33 if sys.version_info[:2] > (2, 5):
  34     if len(sys.argv) > 1 and 'install' in sys.argv:
  35         try:
  36             import platform
  37             (dist, ver, id) = platform.linux_distribution()
  38
  39             # for debian-like distros, mods will be installed to
  40             # ${PYTHONLIB}/dist-packages
  41             if dist in ('debian', 'Ubuntu'):
  42                 sys.argv.append('--install-layout=deb')
  43         except:
  44             pass
  45
  46 PACKAGES = [MOD_NAME,
  47             MOD_NAME + '/utils',
  48             MOD_NAME + '/imager',
  49             MOD_NAME + '/kickstart',
  50             MOD_NAME + '/kickstart/custom_commands',
  51             MOD_NAME + '/3rdparty/pykickstart',
  52             MOD_NAME + '/3rdparty/pykickstart/commands',
  53             MOD_NAME + '/3rdparty/pykickstart/handlers',
  54             MOD_NAME + '/3rdparty/pykickstart/urlgrabber',
  55            ]
  56
  57 IMAGER_PLUGINS = glob.glob(os.path.join("plugins", "imager", "*.py"))
  58 BACKEND_PLUGINS = glob.glob(os.path.join("plugins", "backend", "*.py"))
  59
  60 # the following code to do a simple parse for '--prefix' opts
  61 prefix = sys.prefix
  62 is_next = False
  63 for arg in sys.argv:
  64     if is_next:
  65         prefix = arg
  66         break
  67     if '--prefix=' in arg:
  68         prefix = arg[9:]
  69         break
  70     elif '--prefix' == arg:
  71         is_next = True
  72
  73 # get the installation path of mic.conf
  74 prefix = os.path.abspath(os.path.expanduser(prefix)).rstrip('/')
  75 if prefix.lstrip('/') == 'usr':
  76     etc_prefix = '/etc'
  77 else:
  78     etc_prefix = os.path.join(prefix, 'etc')
  79
  80 conffile = 'distfiles/mic.conf'
  81 if os.path.isfile('%s/mic/mic.conf' % etc_prefix):
  82     conffile += '.new'
  83
  84 # apply prefix to mic.conf.in to generate actual mic.conf
  85 conf_str = file('distfiles/mic.conf.in').read()
  86 conf_str = conf_str.replace('@PREFIX@', prefix)
  87 with file(conffile, 'w') as wf:
  88     wf.write(conf_str)
  89
  90 try:
  91     os.environ['PREFIX'] = prefix
  92     setup(name=MOD_NAME,
  93           version = version,
  94           description = 'Image Creator for Linux Distributions',
  95           author='Jian-feng Ding, Qiang Zhang, Gui Chen',
  96           author_email='jian-feng.ding@intel.com, qiang.z.zhang@intel.com, gui.chen@intel.com',
  97           url='https://github.com/jfding/mic',
  98           scripts=[
  99               'tools/mic',
 100               ],
 101           packages = PACKAGES,
 102           data_files = [("%s/lib/mic/plugins/imager" % prefix, IMAGER_PLUGINS),
 103                         ("%s/lib/mic/plugins/backend" % prefix, BACKEND_PLUGINS),
 104                         ("%s/mic" % etc_prefix, [conffile])]
 105     )
 106 finally:
 107     # remove dynamic file distfiles/mic.conf
 108     os.unlink(conffile)
 109