setup.py 5.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. try:
  4. from setuptools import setup, find_packages
  5. from setuptools.command.test import test
  6. except ImportError:
  7. raise
  8. from ez_setup import use_setuptools
  9. use_setuptools()
  10. from setuptools import setup, find_packages # noqa
  11. from setuptools.command.test import test # noqa
  12. import os
  13. import sys
  14. import codecs
  15. if sys.version_info < (2, 6):
  16. raise Exception('Celery 3.1 requires Python 2.6 or higher.')
  17. try:
  18. orig_path = sys.path[:]
  19. for path in (os.path.curdir, os.getcwd()):
  20. if path in sys.path:
  21. sys.path.remove(path)
  22. try:
  23. import celery.app
  24. import imp
  25. import shutil
  26. _, task_path, _ = imp.find_module('task', celery.app.__path__)
  27. if task_path.endswith('/task'):
  28. print('- force upgrading previous installation')
  29. print(' - removing %r package...' % task_path)
  30. try:
  31. shutil.rmtree(os.path.abspath(task_path))
  32. except Exception:
  33. sys.stderr.write('Could not remove %r: %r\n' % (
  34. task_path, sys.exc_info[1]))
  35. except ImportError:
  36. print('Upgrade: no old version found.')
  37. finally:
  38. sys.path[:] = orig_path
  39. except:
  40. pass
  41. NAME = 'celery'
  42. entrypoints = {}
  43. extra = {}
  44. # -*- Classifiers -*-
  45. classes = """
  46. Development Status :: 5 - Production/Stable
  47. License :: OSI Approved :: BSD License
  48. Topic :: System :: Distributed Computing
  49. Topic :: Software Development :: Object Brokering
  50. Programming Language :: Python
  51. Programming Language :: Python :: 2
  52. Programming Language :: Python :: 2.6
  53. Programming Language :: Python :: 2.7
  54. Programming Language :: Python :: Implementation :: CPython
  55. Programming Language :: Python :: Implementation :: PyPy
  56. Programming Language :: Python :: Implementation :: Jython
  57. Operating System :: OS Independent
  58. Operating System :: POSIX
  59. Operating System :: Microsoft :: Windows
  60. Operating System :: MacOS :: MacOS X
  61. """
  62. classifiers = [s.strip() for s in classes.split('\n') if s]
  63. # -*- Python 3 -*-
  64. is_py3k = sys.version_info[0] == 3
  65. if is_py3k:
  66. extra.update(use_2to3=True)
  67. # -*- Distribution Meta -*-
  68. import re
  69. re_meta = re.compile(r'__(\w+?)__\s*=\s*(.*)')
  70. re_vers = re.compile(r'VERSION\s*=\s*\((.*?)\)')
  71. re_doc = re.compile(r'^"""(.+?)"""')
  72. rq = lambda s: s.strip("\"'")
  73. def add_default(m):
  74. attr_name, attr_value = m.groups()
  75. return ((attr_name, rq(attr_value)), )
  76. def add_version(m):
  77. v = list(map(rq, m.groups()[0].split(', ')))
  78. return (('VERSION', '.'.join(v[0:3]) + ''.join(v[3:])), )
  79. def add_doc(m):
  80. return (('doc', m.groups()[0]), )
  81. pats = {re_meta: add_default,
  82. re_vers: add_version,
  83. re_doc: add_doc}
  84. here = os.path.abspath(os.path.dirname(__file__))
  85. meta_fh = open(os.path.join(here, 'celery/__init__.py'))
  86. try:
  87. meta = {}
  88. for line in meta_fh:
  89. if line.strip() == '# -eof meta-':
  90. break
  91. for pattern, handler in pats.items():
  92. m = pattern.match(line.strip())
  93. if m:
  94. meta.update(handler(m))
  95. finally:
  96. meta_fh.close()
  97. # -*- Custom Commands -*-
  98. class quicktest(test):
  99. extra_env = dict(SKIP_RLIMITS=1, QUICKTEST=1)
  100. def run(self, *args, **kwargs):
  101. for env_name, env_value in self.extra_env.items():
  102. os.environ[env_name] = str(env_value)
  103. test.run(self, *args, **kwargs)
  104. # -*- Installation Requires -*-
  105. py_version = sys.version_info
  106. is_jython = sys.platform.startswith('java')
  107. is_pypy = hasattr(sys, 'pypy_version_info')
  108. def strip_comments(l):
  109. return l.split('#', 1)[0].strip()
  110. def reqs(f):
  111. return filter(None, [strip_comments(l) for l in open(
  112. os.path.join(os.getcwd(), 'requirements', f)).readlines()])
  113. install_requires = reqs('default-py3k.txt' if is_py3k else 'default.txt')
  114. if is_jython:
  115. install_requires.extend(reqs('jython.txt'))
  116. if py_version[0:2] == (2, 6):
  117. install_requires.extend(reqs('py26.txt'))
  118. # -*- Tests Requires -*-
  119. if is_py3k:
  120. tests_require = reqs('test-py3k.txt')
  121. elif is_pypy:
  122. tests_require = reqs('test-pypy.txt')
  123. else:
  124. tests_require = reqs('test.txt')
  125. # -*- Long Description -*-
  126. if os.path.exists('README.rst'):
  127. long_description = codecs.open('README.rst', 'r', 'utf-8').read()
  128. else:
  129. long_description = 'See http://pypi.python.org/pypi/celery'
  130. # -*- Entry Points -*- #
  131. console_scripts = entrypoints['console_scripts'] = [
  132. 'celery = celery.bin.celery:main',
  133. 'celeryd = celery.bin.celeryd:main',
  134. 'celerybeat = celery.bin.celerybeat:main',
  135. 'camqadm = celery.bin.camqadm:main',
  136. 'celeryev = celery.bin.celeryev:main',
  137. 'celeryctl = celery.bin.celeryctl:main',
  138. 'celeryd-multi = celery.bin.celeryd_multi:main',
  139. ]
  140. # bundles: Only relevant for Celery developers.
  141. entrypoints['bundle.bundles'] = ['celery = celery.contrib.bundles:bundles']
  142. # -*- %%% -*-
  143. setup(
  144. name=NAME,
  145. version=meta['VERSION'],
  146. description=meta['doc'],
  147. author=meta['author'],
  148. author_email=meta['contact'],
  149. url=meta['homepage'],
  150. platforms=['any'],
  151. license='BSD',
  152. packages=find_packages(exclude=['ez_setup', 'tests', 'tests.*']),
  153. zip_safe=False,
  154. install_requires=install_requires,
  155. tests_require=tests_require,
  156. test_suite='nose.collector',
  157. cmdclass={'quicktest': quicktest},
  158. classifiers=classifiers,
  159. entry_points=entrypoints,
  160. long_description=long_description,
  161. **extra)