Commits

Andrew Godwin committed ad3aa55

Add a new setup.py in the root.

Comments (0)

Files changed (1)

+#! /usr/bin/python
+#
+# See README for usage instructions.
+
+# We must use setuptools, not distutils, because we need to use the
+# namespace_packages option for the "google" package.
+from setuptools import setup, Extension
+from distutils.spawn import find_executable
+import sys
+import os
+import subprocess
+
+maintainer_email = "andrew@aeracode.org"
+
+# Find the Protocol Compiler.
+if os.path.exists("src/protoc"):
+  protoc = "src/protoc"
+elif os.path.exists("src/protoc.exe"):
+  protoc = "src/protoc.exe"
+elif os.path.exists("vsprojects/Debug/protoc.exe"):
+  protoc = "vsprojects/Debug/protoc.exe"
+elif os.path.exists("vsprojects/Release/protoc.exe"):
+  protoc = "vsprojects/Release/protoc.exe"
+else:
+  protoc = find_executable("protoc")
+
+def generate_proto(source):
+  """Invokes the Protocol Compiler to generate a _pb2.py from the given
+  .proto file.  Does nothing if the output already exists and is newer than
+  the input."""
+
+  output = source.replace(".proto", "_pb2.py").replace("src/", "")
+
+  if not os.path.exists(source):
+    print "Can't find required file: " + source
+    sys.exit(-1)
+
+  if (not os.path.exists(output) or
+      (os.path.exists(source) and
+       os.path.getmtime(source) > os.path.getmtime(output))):
+    print "Generating %s..." % output
+
+    if protoc == None:
+      sys.stderr.write(
+          "protoc is not installed nor found in src.  Please compile it "
+          "or install the binary package.\n")
+      sys.exit(-1)
+
+    protoc_command = [ protoc, "-Isrc", "-I.", "--python_out=python", source ]
+    if subprocess.call(protoc_command) != 0:
+      sys.exit(-1)
+
+def MakeTestSuite():
+  # This is apparently needed on some systems to make sure that the tests
+  # work even if a previous version is already installed.
+  if 'google' in sys.modules:
+    del sys.modules['google']
+
+  generate_proto("src/google/protobuf/unittest.proto")
+  generate_proto("src/google/protobuf/unittest_custom_options.proto")
+  generate_proto("src/google/protobuf/unittest_import.proto")
+  generate_proto("src/google/protobuf/unittest_mset.proto")
+  generate_proto("src/google/protobuf/unittest_no_generic_services.proto")
+  generate_proto("python/google/protobuf/internal/more_extensions.proto")
+  generate_proto("python/google/protobuf/internal/more_messages.proto")
+
+  import unittest
+  import google.protobuf.internal.generator_test     as generator_test
+  import google.protobuf.internal.descriptor_test    as descriptor_test
+  import google.protobuf.internal.reflection_test    as reflection_test
+  import google.protobuf.internal.service_reflection_test \
+    as service_reflection_test
+  import google.protobuf.internal.text_format_test   as text_format_test
+  import google.protobuf.internal.wire_format_test   as wire_format_test
+
+  loader = unittest.defaultTestLoader
+  suite = unittest.TestSuite()
+  for test in [ generator_test,
+                descriptor_test,
+                reflection_test,
+                service_reflection_test,
+                text_format_test,
+                wire_format_test ]:
+    suite.addTest(loader.loadTestsFromModule(test))
+
+  return suite
+
+if __name__ == '__main__':
+  # TODO(kenton):  Integrate this into setuptools somehow?
+  if len(sys.argv) >= 2 and sys.argv[1] == "clean":
+    # Delete generated _pb2.py files and .pyc files in the code tree.
+    for (dirpath, dirnames, filenames) in os.walk("."):
+      for filename in filenames:
+        filepath = os.path.join(dirpath, filename)
+        if filepath.endswith("_pb2.py") or filepath.endswith(".pyc") or \
+          filepath.endswith(".so") or filepath.endswith(".o"):
+          os.remove(filepath)
+  else:
+    # Generate necessary .proto file if it doesn't exist.
+    # TODO(kenton):  Maybe we should hook this into a distutils command?
+    generate_proto("src/google/protobuf/descriptor.proto")
+
+  python_c_extension = Extension("google.protobuf.internal._net_proto2___python",
+                                 [ "python/google/protobuf/pyext/python_descriptor.cc",
+                                   "python/google/protobuf/pyext/python_protobuf.cc",
+                                   "python/google/protobuf/pyext/python-proto2.cc",
+                                   ],
+                                 include_dirs = [ "src", ".", ],
+                                 libraries = [ "protobuf" ],
+                                 runtime_library_dirs = [ "src/.libs" ],
+                                 library_dirs = [ "src/.libs" ],
+                                 )
+
+  setup(name = 'protobuf',
+        version = '2.4.0-pre',
+        packages = [ 'google' ],
+        namespace_packages = [ 'google' ],
+        test_suite = 'setup.MakeTestSuite',
+        # Must list modules explicitly so that we don't install tests.
+        py_modules = [
+          'google.protobuf.internal.api_implementation',
+          'google.protobuf.internal.containers',
+          'google.protobuf.internal.cpp_message',
+          'google.protobuf.internal.decoder',
+          'google.protobuf.internal.encoder',
+          'google.protobuf.internal.message_listener',
+          'google.protobuf.internal.python_message',
+          'google.protobuf.internal.type_checkers',
+          'google.protobuf.internal.wire_format',
+          'google.protobuf.descriptor',
+          'google.protobuf.descriptor_pb2',
+          'google.protobuf.message',
+          'google.protobuf.reflection',
+          'google.protobuf.service',
+          'google.protobuf.service_reflection',
+          'google.protobuf.text_format' ],
+        package_dir = {'': 'python'},
+        #ext_modules = [ python_c_extension ],
+        url = 'http://code.google.com/p/protobuf/',
+        maintainer = maintainer_email,
+        maintainer_email = 'protobuf@googlegroups.com',
+        license = 'New BSD License',
+        description = 'Protocol Buffers',
+        long_description =
+          "Protocol Buffers are Google's data interchange format.",
+        )