bump up to release 0.28.4

[platform/upstream/mic.git] / setup.py

diff --git a/setup.py b/setup.py
index 22706fd..bef22ef 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -1,47 +1,64 @@
 #!/usr/bin/env python
 
+# Copyright (c) 2014 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.
+
+"""
+    Main for installing mic
+"""
+
+
 import os, sys
 import glob
 from distutils.core import setup
-try:
-    import setuptools
-    # enable "setup.py develop", optional
-except ImportError:
-    pass
+
 
 MOD_NAME = 'mic'
 
-version_path = 'VERSION'
-if not os.path.isfile(version_path):
-    print 'No VERSION file in topdir, abort'
-    sys.exit(1)
+
+def check_debian():
+    """--install-layout is recognized after 2.5"""
+    if sys.version_info[:2] > (2, 5):
+        if len(sys.argv) > 1 and 'install' in sys.argv:
+            try:
+                import platform
+                (dist, _, _) = platform.linux_distribution()
+                # for debian-like distros, mods will be installed to
+                # ${PYTHONLIB}/dist-packages
+                if dist in ('debian', 'Ubuntu'):
+                    sys.argv.append('--install-layout=deb')
+            except AttributeError:
+                pass
+
+
+def create_conf_file():
+    """Apply prefix to mic.conf.in to generate actual mic.conf"""
+    with open('etc/mic.conf.in') as source_file:
+        conf_str = source_file.read()
+        conf_str = conf_str.replace('@PREFIX@', PREFIX)
+        with open(CONF_FILE, 'w') as conf_file:
+            conf_file.write(conf_str)
+
 
 try:
-    # first line should be the version number
-    version = open(version_path).readline().strip()
-    if not version:
-        print 'VERSION file is invalid, abort'
-        sys.exit(1)
-
-    ver_file = open('%s/__version__.py' % MOD_NAME, 'w')
-    ver_file.write("VERSION = \"%s\"\n" % version)
-    ver_file.close()
-except IOError:
-    print 'WARNING: Cannot write version number file'
-
-# --install-layout is recognized after 2.5
-if sys.version_info[:2] > (2, 5):
-    if len(sys.argv) > 1 and 'install' in sys.argv:
-        try:
-            import platform
-            (dist, ver, id) = platform.linux_distribution()
-
-            # for debian-like distros, mods will be installed to
-            # ${PYTHONLIB}/dist-packages
-            if dist in ('debian', 'Ubuntu'):
-                sys.argv.append('--install-layout=deb')
-        except:
-            pass
+    import mic
+    VERSION = mic.__version__
+except (ImportError, AttributeError):
+    VERSION = "dev"
+
+check_debian()
 
 PACKAGES = [MOD_NAME,
             MOD_NAME + '/utils',
@@ -57,45 +74,25 @@ PACKAGES = [MOD_NAME,
 IMAGER_PLUGINS = glob.glob(os.path.join("plugins", "imager", "*.py"))
 BACKEND_PLUGINS = glob.glob(os.path.join("plugins", "backend", "*.py"))
 
-# the following code to do a simple parse for '--prefix' opts
-prefix = '/usr'
-is_next = False
-for arg in sys.argv:
-    if is_next:
-        prefix = arg
-        break
-    if '--prefix=' in arg:
-        prefix = arg[9:]
-    elif '--prefix' == arg:
-        is_next = True
-
-if prefix == '/usr':
-    etc_prefix = '/etc'
-else:
-    etc_prefix = os.path.join(prefix, 'etc')
-
-# apply prefix to mic.conf.in to generate actual mic.conf
-conf_str = file('distfiles/mic.conf.in').read()
-conf_str = conf_str.replace('@PREFIX@', prefix)
-with file('distfiles/mic.conf', 'w') as wf:
-    wf.write(conf_str)
+PREFIX = sys.prefix
+# if real_prefix, it must be in virtualenv, use prefix as root
+ROOT = sys.prefix if hasattr(sys, 'real_prefix') else ''
 
-try:
-    setup(name=MOD_NAME,
-          version = version,
-          description = 'Image Creator for Linux Distributions',
-          author='Jian-feng Ding, Qiang Zhang, Gui Chen',
-          author_email='jian-feng.ding@intel.com, qiang.z.zhang@intel.com, gui.chen@intel.com',
-          url='https://github.com/jfding/mic',
-          scripts=[
-              'tools/mic',
-              ],
-          packages = PACKAGES,
-          data_files = [("%s/lib/mic/plugins/imager" % prefix, IMAGER_PLUGINS),
-                        ("%s/lib/mic/plugins/backend" % prefix, BACKEND_PLUGINS),
-                        ("%s/mic" % etc_prefix, ["distfiles/mic.conf"])]
-    )
-finally:
-    # remove dynamic file distfiles/mic.conf
-    os.unlink('distfiles/mic.conf')
+CONF_FILE = 'etc/mic.conf'
+create_conf_file()
 
+setup(name=MOD_NAME,
+  version = VERSION,
+  description = 'Image Creator for Linux Distributions',
+  author='Jian-feng Ding, Qiang Zhang, Gui Chen',
+  author_email='jian-feng.ding@intel.com, qiang.z.zhang@intel.com,\
+                gui.chen@intel.com',
+  url='https://github.com/01org/mic',
+  scripts=[
+      'tools/mic',
+      ],
+  packages = PACKAGES,
+  data_files = [("%s/lib/mic/plugins/imager" % PREFIX, IMAGER_PLUGINS),
+                ("%s/lib/mic/plugins/backend" % PREFIX, BACKEND_PLUGINS),
+                ("%s/etc/mic" % ROOT, [CONF_FILE])]
+)