setup.py 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  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, 5):
  16. raise Exception('Celery requires Python 2.5 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.5
  53. Programming Language :: Python :: 2.6
  54. Programming Language :: Python :: 2.7
  55. Programming Language :: Python :: 3
  56. Programming Language :: Python :: 3.2
  57. Programming Language :: Python :: 3.3
  58. Programming Language :: Python :: Implementation :: CPython
  59. Programming Language :: Python :: Implementation :: PyPy
  60. Programming Language :: Python :: Implementation :: Jython
  61. Operating System :: OS Independent
  62. Operating System :: POSIX
  63. Operating System :: Microsoft :: Windows
  64. Operating System :: MacOS :: MacOS X
  65. """
  66. classifiers = [s.strip() for s in classes.split('\n') if s]
  67. # -*- Python 3 -*-
  68. is_py3k = sys.version_info[0] == 3
  69. if is_py3k:
  70. extra.update(use_2to3=True)
  71. # -*- Distribution Meta -*-
  72. import re
  73. re_meta = re.compile(r'__(\w+?)__\s*=\s*(.*)')
  74. re_vers = re.compile(r'VERSION\s*=\s*\((.*?)\)')
  75. re_doc = re.compile(r'^"""(.+?)"""')
  76. rq = lambda s: s.strip("\"'")
  77. def add_default(m):
  78. attr_name, attr_value = m.groups()
  79. return ((attr_name, rq(attr_value)), )
  80. def add_version(m):
  81. v = list(map(rq, m.groups()[0].split(', ')))
  82. return (('VERSION', '.'.join(v[0:3]) + ''.join(v[3:])), )
  83. def add_doc(m):
  84. return (('doc', m.groups()[0]), )
  85. pats = {re_meta: add_default,
  86. re_vers: add_version,
  87. re_doc: add_doc}
  88. here = os.path.abspath(os.path.dirname(__file__))
  89. meta_fh = open(os.path.join(here, 'celery/__init__.py'))
  90. try:
  91. meta = {}
  92. for line in meta_fh:
  93. if line.strip() == '# -eof meta-':
  94. break
  95. for pattern, handler in pats.items():
  96. m = pattern.match(line.strip())
  97. if m:
  98. meta.update(handler(m))
  99. finally:
  100. meta_fh.close()
  101. # -*- Custom Commands -*-
  102. class quicktest(test):
  103. extra_env = dict(SKIP_RLIMITS=1, QUICKTEST=1)
  104. def run(self, *args, **kwargs):
  105. for env_name, env_value in self.extra_env.items():
  106. os.environ[env_name] = str(env_value)
  107. test.run(self, *args, **kwargs)
  108. # -*- Installation Requires -*-
  109. py_version = sys.version_info
  110. is_jython = sys.platform.startswith('java')
  111. is_pypy = hasattr(sys, 'pypy_version_info')
  112. def strip_comments(l):
  113. return l.split('#', 1)[0].strip()
  114. def reqs(f):
  115. return filter(None, [strip_comments(l) for l in open(
  116. os.path.join(os.getcwd(), 'requirements', f)).readlines()])
  117. install_requires = reqs('default-py3k.txt' if is_py3k else 'default.txt')
  118. if is_jython:
  119. install_requires.extend(reqs('jython.txt'))
  120. if py_version[0:2] == (2, 6):
  121. install_requires.extend(reqs('py26.txt'))
  122. elif py_version[0:2] == (2, 5):
  123. install_requires.extend(reqs('py25.txt'))
  124. # -*- Tests Requires -*-
  125. if is_py3k:
  126. tests_require = reqs('test-py3k.txt')
  127. elif is_pypy:
  128. tests_require = reqs('test-pypy.txt')
  129. else:
  130. tests_require = reqs('test.txt')
  131. if py_version[0:2] == (2, 5):
  132. tests_require.extend(reqs('test-py25.txt'))
  133. # -*- Long Description -*-
  134. if os.path.exists('README.rst'):
  135. long_description = codecs.open('README.rst', 'r', 'utf-8').read()
  136. else:
  137. long_description = 'See http://pypi.python.org/pypi/celery'
  138. # -*- Entry Points -*- #
  139. console_scripts = entrypoints['console_scripts'] = [
  140. 'celery = celery.bin.celery:main',
  141. 'celeryd = celery.bin.celeryd:main',
  142. 'celerybeat = celery.bin.celerybeat:main',
  143. 'camqadm = celery.bin.camqadm:main',
  144. 'celeryev = celery.bin.celeryev:main',
  145. 'celeryctl = celery.bin.celeryctl:main',
  146. 'celeryd-multi = celery.bin.celeryd_multi:main',
  147. ]
  148. # bundles: Only relevant for Celery developers.
  149. entrypoints['bundle.bundles'] = ['celery = celery.contrib.bundles:bundles']
  150. # -*- %%% -*-
  151. setup(
  152. name=NAME,
  153. version=meta['VERSION'],
  154. description=meta['doc'],
  155. author=meta['author'],
  156. author_email=meta['contact'],
  157. url=meta['homepage'],
  158. platforms=['any'],
  159. license='BSD',
  160. packages=find_packages(exclude=['ez_setup', 'tests', 'tests.*']),
  161. zip_safe=False,
  162. install_requires=install_requires,
  163. tests_require=tests_require,
  164. test_suite='nose.collector',
  165. cmdclass={'quicktest': quicktest},
  166. classifiers=classifiers,
  167. entry_points=entrypoints,
  168. long_description=long_description,
  169. **extra)