setup.py 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197
  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. is_setuptools = True
  7. except ImportError:
  8. raise
  9. from ez_setup import use_setuptools
  10. use_setuptools()
  11. from setuptools import setup, find_packages # noqa
  12. from setuptools.command.test import test # noqa
  13. is_setuptools = False
  14. import os
  15. import sys
  16. import codecs
  17. CELERY_COMPAT_PROGRAMS = int(os.environ.get('CELERY_COMPAT_PROGRAMS', 1))
  18. if sys.version_info < (2, 6):
  19. raise Exception('Celery 3.1 requires Python 2.6 or higher.')
  20. downgrade_packages = [
  21. 'celery.app.task',
  22. ]
  23. orig_path = sys.path[:]
  24. for path in (os.path.curdir, os.getcwd()):
  25. if path in sys.path:
  26. sys.path.remove(path)
  27. try:
  28. import imp
  29. import shutil
  30. for pkg in downgrade_packages:
  31. try:
  32. parent, module = pkg.rsplit('.', 1)
  33. print('- Trying to upgrade %r in %r' % (module, parent))
  34. parent_mod = __import__(parent, None, None, [parent])
  35. _, mod_path, _ = imp.find_module(module, parent_mod.__path__)
  36. if mod_path.endswith('/' + module):
  37. print('- force upgrading previous installation')
  38. print(' - removing {0!r} package...'.format(mod_path))
  39. try:
  40. shutil.rmtree(os.path.abspath(mod_path))
  41. except Exception:
  42. sys.stderr.write('Could not remove {0!r}: {1!r}\n'.format(
  43. mod_path, sys.exc_info[1]))
  44. except ImportError:
  45. print('- upgrade %s: no old version found.' % module)
  46. except:
  47. pass
  48. finally:
  49. sys.path[:] = orig_path
  50. NAME = 'celery'
  51. entrypoints = {}
  52. extra = {}
  53. # -*- Classifiers -*-
  54. classes = """
  55. Development Status :: 5 - Production/Stable
  56. License :: OSI Approved :: BSD License
  57. Topic :: System :: Distributed Computing
  58. Topic :: Software Development :: Object Brokering
  59. Programming Language :: Python
  60. Programming Language :: Python :: 2
  61. Programming Language :: Python :: 2.6
  62. Programming Language :: Python :: 2.7
  63. Programming Language :: Python :: 3
  64. Programming Language :: Python :: 3.3
  65. Programming Language :: Python :: Implementation :: CPython
  66. Programming Language :: Python :: Implementation :: PyPy
  67. Programming Language :: Python :: Implementation :: Jython
  68. Operating System :: OS Independent
  69. Operating System :: POSIX
  70. Operating System :: Microsoft :: Windows
  71. Operating System :: MacOS :: MacOS X
  72. """
  73. classifiers = [s.strip() for s in classes.split('\n') if s]
  74. PY3 = sys.version_info[0] == 3
  75. JYTHON = sys.platform.startswith('java')
  76. PYPY = hasattr(sys, 'pypy_version_info')
  77. # -*- Distribution Meta -*-
  78. import re
  79. re_meta = re.compile(r'__(\w+?)__\s*=\s*(.*)')
  80. re_vers = re.compile(r'VERSION\s*=.*?\((.*?)\)')
  81. re_doc = re.compile(r'^"""(.+?)"""')
  82. rq = lambda s: s.strip("\"'")
  83. def add_default(m):
  84. attr_name, attr_value = m.groups()
  85. return ((attr_name, rq(attr_value)), )
  86. def add_version(m):
  87. v = list(map(rq, m.groups()[0].split(', ')))
  88. return (('VERSION', '.'.join(v[0:3]) + ''.join(v[3:])), )
  89. def add_doc(m):
  90. return (('doc', m.groups()[0]), )
  91. pats = {re_meta: add_default,
  92. re_vers: add_version,
  93. re_doc: add_doc}
  94. here = os.path.abspath(os.path.dirname(__file__))
  95. with open(os.path.join(here, 'celery/__init__.py')) as meta_fh:
  96. meta = {}
  97. for line in meta_fh:
  98. if line.strip() == '# -eof meta-':
  99. break
  100. for pattern, handler in pats.items():
  101. m = pattern.match(line.strip())
  102. if m:
  103. meta.update(handler(m))
  104. # -*- Installation Requires -*-
  105. py_version = sys.version_info
  106. def strip_comments(l):
  107. return l.split('#', 1)[0].strip()
  108. def reqs(*f):
  109. return [
  110. r for r in (
  111. strip_comments(l) for l in open(
  112. os.path.join(os.getcwd(), 'requirements', *f)).readlines()
  113. ) if r]
  114. install_requires = reqs('default.txt')
  115. if JYTHON:
  116. install_requires.extend(reqs('jython.txt'))
  117. # -*- Tests Requires -*-
  118. tests_require = reqs('test3.txt' if PY3 else 'test.txt')
  119. # -*- Long Description -*-
  120. if os.path.exists('README.rst'):
  121. long_description = codecs.open('README.rst', 'r', 'utf-8').read()
  122. else:
  123. long_description = 'See http://pypi.python.org/pypi/celery'
  124. # -*- Entry Points -*- #
  125. console_scripts = entrypoints['console_scripts'] = [
  126. 'celery = celery.__main__:main',
  127. ]
  128. if CELERY_COMPAT_PROGRAMS:
  129. console_scripts.extend([
  130. 'celeryd = celery.__main__:_compat_worker',
  131. 'celerybeat = celery.__main__:_compat_beat',
  132. 'celeryd-multi = celery.__main__:_compat_multi',
  133. ])
  134. if is_setuptools:
  135. extras = lambda *p: reqs('extras', *p)
  136. # Celery specific
  137. specific_list = ['auth', 'cassandra', 'memcache', 'couchbase', 'threads',
  138. 'eventlet', 'gevent', 'msgpack', 'yaml', 'redis',
  139. 'mongodb', 'sqs', 'couchdb', 'beanstalk', 'zookeeper',
  140. 'zeromq', 'sqlalchemy', 'librabbitmq', 'pyro', 'slmq']
  141. extras_require = dict((x, extras(x + '.txt')) for x in specific_list)
  142. extra['extras_require'] = extras_require
  143. # -*- %%% -*-
  144. setup(
  145. name=NAME,
  146. version=meta['VERSION'],
  147. description=meta['doc'],
  148. author=meta['author'],
  149. author_email=meta['contact'],
  150. url=meta['homepage'],
  151. platforms=['any'],
  152. license='BSD',
  153. packages=find_packages(exclude=['ez_setup', 'tests', 'tests.*']),
  154. zip_safe=False,
  155. install_requires=install_requires,
  156. tests_require=tests_require,
  157. test_suite='nose.collector',
  158. classifiers=classifiers,
  159. entry_points=entrypoints,
  160. long_description=long_description,
  161. **extra)