def main()

in build_pip_pkg.py [0:0]

• 58 lines of code
• 5 McCabe index (conditional complexity)

def main(srcdir):
  tempdir = tempfile.mkdtemp()
  atexit.register(shutil.rmtree, tempdir)

  pkgdir = os.path.join(tempdir, "tensorflow_compression")
  shutil.copytree(os.path.join(srcdir, "tensorflow_compression"), pkgdir)
  shutil.copy2(os.path.join(srcdir, "MANIFEST.in"), tempdir)
  shutil.copy2(os.path.join(srcdir, "LICENSE"), pkgdir)
  shutil.copy2(os.path.join(srcdir, "README.md"), pkgdir)

  if not os.path.exists(
      os.path.join(pkgdir, "cc/libtensorflow_compression.so")):
    raise RuntimeError("libtensorflow_compression.so not found. "
                       "Did you 'bazel run?'")

  with open(os.path.join(srcdir, "requirements.txt"), "r") as f:
    install_requires = f.readlines()

  print("=== Building wheel")
  atexit.register(os.chdir, os.getcwd())
  os.chdir(tempdir)
  setuptools.setup(
      name="tensorflow_compression",
      version=__version__,
      description="Data compression in TensorFlow",
      url="https://tensorflow.github.io/compression/",
      author="Google LLC",
      # Contained modules and scripts.
      packages=setuptools.find_packages(),
      install_requires=install_requires,
      script_args=["sdist", "bdist_wheel"],
      # Add in any packaged data.
      include_package_data=True,
      zip_safe=False,
      distclass=BinaryDistribution,
      # PyPI package information.
      classifiers=[
          "Development Status :: 5 - Production/Stable",
          "Intended Audience :: Developers",
          "Intended Audience :: Education",
          "Intended Audience :: Science/Research",
          "License :: OSI Approved :: Apache Software License",
          "Programming Language :: Python :: 3",
          "Topic :: Scientific/Engineering :: Mathematics",
          "Topic :: Software Development :: Libraries :: Python Modules",
          "Topic :: Software Development :: Libraries",
      ],
      project_urls={
          "Documentation":
              "https://tensorflow.github.io/compression/docs/api_docs/python/tfc.html",
          "Discussion":
              "https://groups.google.com/forum/#!forum/tensorflow-compression",
          "Source": "https://github.com/tensorflow/compression",
          "Tracker": "https://github.com/tensorflow/compression/issues",
      },
      license="Apache 2.0",
      keywords=("compression data-compression tensorflow machine-learning "
                "python deep-learning deep-neural-networks neural-network ml")
  )

  destdir = "/tmp/tensorflow_compression"
  print("=== Copying wheel to " + destdir)
  if not os.path.exists(destdir): os.mkdir(destdir)
  for path in glob.glob(os.path.join(tempdir, "dist", "*.whl")):
    print("Copying into " + os.path.join(destdir, os.path.basename(path)))
    shutil.copy(path, destdir)