Counter Strike : Global Offensive Source Code
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

196 lines
7.7 KiB

  1. #! /usr/bin/python
  2. #
  3. # See README for usage instructions.
  4. import sys
  5. import os
  6. import subprocess
  7. # We must use setuptools, not distutils, because we need to use the
  8. # namespace_packages option for the "google" package.
  9. try:
  10. from setuptools import setup, Extension
  11. except ImportError:
  12. try:
  13. from ez_setup import use_setuptools
  14. use_setuptools()
  15. from setuptools import setup, Extension
  16. except ImportError:
  17. sys.stderr.write(
  18. "Could not import setuptools; make sure you have setuptools or "
  19. "ez_setup installed.\n")
  20. raise
  21. from distutils.command.clean import clean as _clean
  22. from distutils.command.build_py import build_py as _build_py
  23. from distutils.spawn import find_executable
  24. maintainer_email = "[email protected]"
  25. # Find the Protocol Compiler.
  26. if 'PROTOC' in os.environ and os.path.exists(os.environ['PROTOC']):
  27. protoc = os.environ['PROTOC']
  28. elif os.path.exists("../src/protoc"):
  29. protoc = "../src/protoc"
  30. elif os.path.exists("../src/protoc.exe"):
  31. protoc = "../src/protoc.exe"
  32. elif os.path.exists("../vsprojects/Debug/protoc.exe"):
  33. protoc = "../vsprojects/Debug/protoc.exe"
  34. elif os.path.exists("../vsprojects/Release/protoc.exe"):
  35. protoc = "../vsprojects/Release/protoc.exe"
  36. else:
  37. protoc = find_executable("protoc")
  38. def generate_proto(source):
  39. """Invokes the Protocol Compiler to generate a _pb2.py from the given
  40. .proto file. Does nothing if the output already exists and is newer than
  41. the input."""
  42. output = source.replace(".proto", "_pb2.py").replace("../src/", "")
  43. if (not os.path.exists(output) or
  44. (os.path.exists(source) and
  45. os.path.getmtime(source) > os.path.getmtime(output))):
  46. print "Generating %s..." % output
  47. if not os.path.exists(source):
  48. sys.stderr.write("Can't find required file: %s\n" % source)
  49. sys.exit(-1)
  50. if protoc == None:
  51. sys.stderr.write(
  52. "protoc is not installed nor found in ../src. Please compile it "
  53. "or install the binary package.\n")
  54. sys.exit(-1)
  55. protoc_command = [ protoc, "-I../src", "-I.", "--python_out=.", source ]
  56. if subprocess.call(protoc_command) != 0:
  57. sys.exit(-1)
  58. def GenerateUnittestProtos():
  59. generate_proto("../src/google/protobuf/unittest.proto")
  60. generate_proto("../src/google/protobuf/unittest_custom_options.proto")
  61. generate_proto("../src/google/protobuf/unittest_import.proto")
  62. generate_proto("../src/google/protobuf/unittest_import_public.proto")
  63. generate_proto("../src/google/protobuf/unittest_mset.proto")
  64. generate_proto("../src/google/protobuf/unittest_no_generic_services.proto")
  65. generate_proto("google/protobuf/internal/test_bad_identifiers.proto")
  66. generate_proto("google/protobuf/internal/more_extensions.proto")
  67. generate_proto("google/protobuf/internal/more_extensions_dynamic.proto")
  68. generate_proto("google/protobuf/internal/more_messages.proto")
  69. generate_proto("google/protobuf/internal/factory_test1.proto")
  70. generate_proto("google/protobuf/internal/factory_test2.proto")
  71. def MakeTestSuite():
  72. # This is apparently needed on some systems to make sure that the tests
  73. # work even if a previous version is already installed.
  74. if 'google' in sys.modules:
  75. del sys.modules['google']
  76. GenerateUnittestProtos()
  77. import unittest
  78. import google.protobuf.internal.generator_test as generator_test
  79. import google.protobuf.internal.descriptor_test as descriptor_test
  80. import google.protobuf.internal.reflection_test as reflection_test
  81. import google.protobuf.internal.service_reflection_test \
  82. as service_reflection_test
  83. import google.protobuf.internal.text_format_test as text_format_test
  84. import google.protobuf.internal.wire_format_test as wire_format_test
  85. import google.protobuf.internal.unknown_fields_test as unknown_fields_test
  86. import google.protobuf.internal.descriptor_database_test \
  87. as descriptor_database_test
  88. import google.protobuf.internal.descriptor_pool_test as descriptor_pool_test
  89. import google.protobuf.internal.message_factory_test as message_factory_test
  90. import google.protobuf.internal.message_cpp_test as message_cpp_test
  91. import google.protobuf.internal.reflection_cpp_generated_test \
  92. as reflection_cpp_generated_test
  93. loader = unittest.defaultTestLoader
  94. suite = unittest.TestSuite()
  95. for test in [ generator_test,
  96. descriptor_test,
  97. reflection_test,
  98. service_reflection_test,
  99. text_format_test,
  100. wire_format_test ]:
  101. suite.addTest(loader.loadTestsFromModule(test))
  102. return suite
  103. class clean(_clean):
  104. def run(self):
  105. # Delete generated files in the code tree.
  106. for (dirpath, dirnames, filenames) in os.walk("."):
  107. for filename in filenames:
  108. filepath = os.path.join(dirpath, filename)
  109. if filepath.endswith("_pb2.py") or filepath.endswith(".pyc") or \
  110. filepath.endswith(".so") or filepath.endswith(".o") or \
  111. filepath.endswith('google/protobuf/compiler/__init__.py'):
  112. os.remove(filepath)
  113. # _clean is an old-style class, so super() doesn't work.
  114. _clean.run(self)
  115. class build_py(_build_py):
  116. def run(self):
  117. # Generate necessary .proto file if it doesn't exist.
  118. generate_proto("../src/google/protobuf/descriptor.proto")
  119. generate_proto("../src/google/protobuf/compiler/plugin.proto")
  120. GenerateUnittestProtos()
  121. # Make sure google.protobuf.compiler is a valid package.
  122. open('google/protobuf/compiler/__init__.py', 'a').close()
  123. # _build_py is an old-style class, so super() doesn't work.
  124. _build_py.run(self)
  125. if __name__ == '__main__':
  126. ext_module_list = []
  127. # C++ implementation extension
  128. if os.getenv("PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION", "python") == "cpp":
  129. print "Using EXPERIMENTAL C++ Implmenetation."
  130. ext_module_list.append(Extension(
  131. "google.protobuf.internal._net_proto2___python",
  132. [ "google/protobuf/pyext/python_descriptor.cc",
  133. "google/protobuf/pyext/python_protobuf.cc",
  134. "google/protobuf/pyext/python-proto2.cc" ],
  135. include_dirs = [ "." ],
  136. libraries = [ "protobuf" ]))
  137. setup(name = 'protobuf',
  138. version = '2.5.0',
  139. packages = [ 'google' ],
  140. namespace_packages = [ 'google' ],
  141. test_suite = 'setup.MakeTestSuite',
  142. # Must list modules explicitly so that we don't install tests.
  143. py_modules = [
  144. 'google.protobuf.internal.api_implementation',
  145. 'google.protobuf.internal.containers',
  146. 'google.protobuf.internal.cpp_message',
  147. 'google.protobuf.internal.decoder',
  148. 'google.protobuf.internal.encoder',
  149. 'google.protobuf.internal.enum_type_wrapper',
  150. 'google.protobuf.internal.message_listener',
  151. 'google.protobuf.internal.python_message',
  152. 'google.protobuf.internal.type_checkers',
  153. 'google.protobuf.internal.wire_format',
  154. 'google.protobuf.descriptor',
  155. 'google.protobuf.descriptor_pb2',
  156. 'google.protobuf.compiler.plugin_pb2',
  157. 'google.protobuf.message',
  158. 'google.protobuf.descriptor_database',
  159. 'google.protobuf.descriptor_pool',
  160. 'google.protobuf.message_factory',
  161. 'google.protobuf.reflection',
  162. 'google.protobuf.service',
  163. 'google.protobuf.service_reflection',
  164. 'google.protobuf.text_format' ],
  165. cmdclass = { 'clean': clean, 'build_py': build_py },
  166. install_requires = ['setuptools'],
  167. ext_modules = ext_module_list,
  168. url = 'http://code.google.com/p/protobuf/',
  169. maintainer = maintainer_email,
  170. maintainer_email = '[email protected]',
  171. license = 'New BSD License',
  172. description = 'Protocol Buffers',
  173. long_description =
  174. "Protocol Buffers are Google's data interchange format.",
  175. )