setup.py

import glob
import os
import subprocess

import torch
from setuptools import find_packages, setup
from torch.utils.cpp_extension import CUDA_HOME, CppExtension, CUDAExtension

version = "v1.5"
package_name = "gdino"
cwd = os.path.dirname(os.path.abspath(__file__))

sha = "Unknown"
try:
 sha = subprocess.check_output(["git", "rev-parse", "HEAD"],
 cwd=cwd).decode("ascii").strip()
except Exception:
 pass


def write_version_file():
 version_path = os.path.join(cwd, "gdino/", "version.py")
 with open(version_path, "w") as f:
 f.write(f"__version__ = '{version}'\n")
 # f.write(f"git_version = {repr(sha)}\n")


def parse_requirements(fname="requirements.txt", with_version=True):
 """Parse the package dependencies listed in a requirements file but strips
 specific versioning information.

 Args:
 fname (str): path to requirements file
 with_version (bool, default=False): if True include version specs

 Returns:
 List[str]: list of requirements items

 CommandLine:
 python -c "import setup; print(setup.parse_requirements())"
 """
 import re
 import sys
 from os.path import exists

 require_fpath = fname

 def parse_line(line):
 """Parse information from a line in a requirements text file."""
 if line.startswith("-r "):
 # Allow specifying requirements in other files
 target = line.split(" ")[1]
 for info in parse_require_file(target):
 yield info
 else:
 info = {"line": line}
 if line.startswith("-e "):
 info["package"] = line.split("#egg=")[1]
 elif "@git+" in line:
 info["package"] = line
 else:
 # Remove versioning from the package
 pat = "(" + "|".join([">=", "==", ">"]) + ")"
 parts = re.split(pat, line, maxsplit=1)
 parts = [p.strip() for p in parts]

 info["package"] = parts[0]
 if len(parts) > 1:
 op, rest = parts[1:]
 if ";" in rest:
 # Handle platform specific dependencies
 # http://setuptools.readthedocs.io/en/latest/setuptools.html#declaring-platform-specific-dependencies
 version, platform_deps = map(str.strip,
 rest.split(";"))
 info["platform_deps"] = platform_deps
 else:
 version = rest # NOQA
 info["version"] = (op, version)
 yield info

 def parse_require_file(fpath):
 with open(fpath, "r") as f:
 for line in f.readlines():
 line = line.strip()
 if line and not line.startswith("#"):
 for info in parse_line(line):
 yield info

 def gen_packages_items():
 if exists(require_fpath):
 for info in parse_require_file(require_fpath):
 parts = [info["package"]]
 if with_version and "version" in info:
 parts.extend(info["version"])
 if not sys.version.startswith("3.4"):
 # apparently package_deps are broken in 3.4
 platform_deps = info.get("platform_deps")
 if platform_deps is not None:
 parts.append(";" + platform_deps)
 item = "".join(parts)
 yield item

 packages = list(gen_packages_items())
 return packages


if __name__ == "__main__":
 print(f"Building wheel {package_name}-{version}")

 with open("LICENSE", "r", encoding="utf-8") as f:
 license = f.read()

 write_version_file()

 setup(
 name="gdino",
 version="v1.5",
 author="International Digital Economy Academy, CVR",
 url="https://github.com/IDEA-Research/Grounding-DINO-1.5-API",
 description="Grounding DINO 1.5 API wrapper.",
 license=license,
 install_requires=parse_requirements("requirements.txt"),
 packages=find_packages(exclude=("tests", )),
 ext_modules=None,
 cmdclass={"build_ext": torch.utils.cpp_extension.BuildExtension},
 )