setup.py

   1 #!/usr/bin/env python
   2 import re
   3 import os
   4 import sys
   5 import stat
   6 from glob import glob
   7
   8 from distutils.core import setup
   9 from distutils.command import build_scripts
  10 build_scripts.first_line_re = re.compile('^should not match$')
  11
  12 from cola import version
  13 from cola import utils
  14 from cola import resources
  15 from cola import core
  16
  17 def main():
  18     # ensure readable files
  19     old_mask = os.umask(0022)
  20     git_version = version.git_version()
  21     if sys.argv[1] in ('install', 'build'):
  22         _setup_environment()
  23         _check_python_version()
  24         _check_git_version(git_version)
  25         _check_pyqt_version()
  26         _build_translations()      # msgfmt: .po -> .qm
  27     try:
  28         if os.path.exists('.git'):
  29             version.write_builtin_version()
  30         _run_setup(git_version)
  31     finally:
  32         if os.path.exists('.git'):
  33             version.delete_builtin_version()
  34     # restore the old mask
  35     os.umask(old_mask)
  36
  37 def _setup_environment():
  38     """Adds win32/ to our path for windows only"""
  39     if sys.platform != 'win32':
  40         return
  41     path = os.environ['PATH']
  42     win32 = os.path.join(os.path.dirname(__file__), 'win32')
  43     os.environ['PATH'] = win32 + os.pathsep + path
  44
  45 def _run_setup(git_version):
  46     """Runs distutils.setup()"""
  47
  48     scripts = ['bin/git-cola']
  49
  50     # git-difftool first moved out of git.git's contrib area in git 1.6.3
  51     if (os.environ.get('INSTALL_GIT_DIFFTOOL', '') or
  52             not version.check('difftool-builtin', git_version)):
  53         scripts.append('bin/difftool/git-difftool')
  54         scripts.append('bin/difftool/git-difftool--helper')
  55
  56     if sys.platform == 'win32':
  57         scripts.append('win32/cola')
  58         scripts.append('win32/dirname')
  59         scripts.append('win32/py2exe-setup.py')
  60         scripts.append('win32/py2exe-setup.cmd')
  61
  62     setup(name = 'git-cola',
  63           version = version.version(),
  64           license = 'GPLv2',
  65           author = 'David Aguilar and contributors',
  66           author_email = 'davvid@gmail.com',
  67           url = 'http://cola.tuxfamily.org/',
  68           description = 'git-cola',
  69           long_description = 'A highly caffeinated git gui',
  70           scripts = scripts,
  71           packages = [],
  72           data_files = cola_data_files())
  73
  74
  75 def cola_data_files():
  76     data = [_app_path('share/git-cola/qm', '*.qm'),
  77             _app_path('share/git-cola/icons', '*.png'),
  78             _app_path('share/git-cola/icons', '*.svg'),
  79             _app_path('share/git-cola/styles', '*.qss'),
  80             _app_path('share/git-cola/styles/images', '*.png'),
  81             _app_path('share/applications', '*.desktop'),
  82             _app_path('share/doc/git-cola', '*.txt'),
  83             _lib_path('cola/*.py'),
  84             _lib_path('cola/models/*.py'),
  85             _lib_path('cola/controllers/*.py'),
  86             _lib_path('cola/views/*.py'),
  87             _lib_path('jsonpickle/*.py'),
  88             _lib_path('simplejson/*.py')]
  89
  90     if sys.platform == 'darwin':
  91         data.append(_app_path('libexec/git-cola', 'ssh-askpass-darwin'))
  92     else:
  93         data.append(_app_path('libexec/git-cola', 'ssh-askpass'))
  94     return data
  95
  96 def _lib_path(entry):
  97     dirname = os.path.dirname(entry)
  98     app_dir = os.path.join('share/git-cola/lib', dirname)
  99     return (app_dir, glob(entry))
 100
 101
 102 def _app_path(dirname, entry):
 103     return (dirname, glob(os.path.join(dirname, entry)))
 104
 105
 106 def _check_python_version():
 107     """Check the minimum Python version
 108     """
 109     pyver = '.'.join(map(lambda x: str(x), sys.version_info))
 110     if not version.check('python', pyver):
 111         print >> sys.stderr, ('Python version %s or newer required.  '
 112                               'Found %s' % (version.get('python'), pyver))
 113         sys.exit(1)
 114
 115
 116 def _check_git_version(git_ver):
 117     """Check the minimum GIT version
 118     """
 119     if not version.check('git', git_ver):
 120         print >> sys.stderr, ('GIT version %s or newer required.  '
 121                               'Found %s' % (version.get('git'), git_ver))
 122         sys.exit(1)
 123
 124 def _check_pyqt_version():
 125     """Check the minimum PyQt version
 126     """
 127     has_pyqt = False
 128     pyqtver = 'None'
 129     try:
 130         from PyQt4 import QtCore
 131         pyqtver = QtCore.PYQT_VERSION_STR
 132         if version.check('pyqt', pyqtver):
 133             return
 134     except ImportError:
 135         pass
 136     print >> sys.stderr, ('PyQt4 version %s or newer required.  '
 137                           'Found %s' % (version.get('pyqt'), pyqtver))
 138     sys.exit(1)
 139
 140
 141 def _dirty(src, dst):
 142     if not os.path.exists(dst):
 143         return True
 144     srcstat = os.stat(src)
 145     dststat = os.stat(dst)
 146     return srcstat[stat.ST_MTIME] > dststat[stat.ST_MTIME]
 147
 148
 149 def _build_translations():
 150     print 'running build_translations'
 151     sources = glob(resources.share('po', '*.po'))
 152     sources = glob('share/git-cola/po/*.po')
 153     for src in sources:
 154         dst = resources.qm(os.path.basename(src)[:-3])
 155         if _dirty(src, dst):
 156             print '\tmsgfmt --qt %s -o %s' % (src, dst)
 157             utils.run_cmd(['msgfmt', '--qt', src, '-o', dst])
 158
 159 def _run_cmd(cmd):
 160     """Runs a command and returns its output."""
 161     pipe = os.popen(cmd)
 162     contents = core.read_nointr(pipe).strip()
 163     pipe.close()
 164     return contents
 165
 166
 167 if __name__ == '__main__':
 168     main()