chiark - git - mdw - stgit/blob - setup.py

   1 #!/usr/bin/env python
   2
   3 import sys, glob, os
   4 from distutils.core import setup
   5
   6 from stgit import version
   7 from stgit import commands, completion
   8
   9 def __version_to_list(version):
  10     """Convert a version string to a list of numbers or strings
  11     """
  12     ver_list = []
  13     for p in version.split('.'):
  14         try:
  15             n = int(p)
  16         except ValueError:
  17             n = p
  18         ver_list.append(n)
  19     return ver_list
  20
  21 def __check_min_version(min_ver, ver):
  22     """Check whether ver is greater or equal to min_ver
  23     """
  24     min_ver_list = __version_to_list(min_ver)
  25     ver_list = __version_to_list(ver)
  26     return min_ver_list <= ver_list
  27
  28 def __check_python_version():
  29     """Check the minimum Python version
  30     """
  31     pyver = '.'.join(map(lambda x: str(x), sys.version_info))
  32     if not __check_min_version(version.python_min_ver, pyver):
  33         print >> sys.stderr, 'Python version %s or newer required. Found %s' \
  34               % (version.python_min_ver, pyver)
  35         sys.exit(1)
  36
  37 def __check_git_version():
  38     """Check the minimum GIT version
  39     """
  40     from stgit.run import Run
  41     gitver = Run('git', '--version').output_one_line().split()[2]
  42     if not __check_min_version(version.git_min_ver, gitver):
  43         print >> sys.stderr, 'GIT version %s or newer required. Found %s' \
  44               % (version.git_min_ver, gitver)
  45         sys.exit(1)
  46
  47 def __run_setup():
  48     setup(name = 'stgit',
  49           version = version.version,
  50           license = 'GPLv2',
  51           author = 'Catalin Marinas',
  52           author_email = 'catalin.marinas@gmail.com',
  53           url = 'http://www.procode.org/stgit/',
  54           description = 'Stacked GIT',
  55           long_description = 'Push/pop utility on top of GIT',
  56           scripts = ['stg'],
  57           packages = ['stgit', 'stgit.commands', 'stgit.lib'],
  58           data_files = [
  59             ('share/stgit/templates', glob.glob('templates/*.tmpl')),
  60             ('share/stgit/examples', glob.glob('examples/*.tmpl')),
  61             ('share/stgit/examples', ['examples/gitconfig']),
  62             ('share/stgit/contrib', ['contrib/stgbashprompt.sh']),
  63             ('share/stgit/completion', ['stgit-completion.bash'])
  64             ])
  65
  66 # Check the minimum versions required
  67 __check_python_version()
  68 __check_git_version()
  69
  70 # ensure readable template files
  71 old_mask = os.umask(0022)
  72
  73 version.write_builtin_version()
  74
  75 # generate the python command list
  76 f = file('stgit/commands/cmdlist.py', 'w')
  77 commands.py_commands(commands.get_commands(allow_cached = False), f)
  78 f.close()
  79
  80 # generate the bash completion script
  81 f = file('stgit-completion.bash', 'w')
  82 completion.write_completion(f)
  83 f.close()
  84
  85 __run_setup()
  86
  87 # restore the old mask
  88 os.umask(old_mask)