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