[dist] Simplify generation of version.py

Author: Breakthrough

MANIFEST.in

@@ -1,7 +1,6 @@
 include LICENSE-3RD-PARTY.txt
 include LICENSE.txt
 include README.md
-include find_version.py
 include setup.py
 include pyproject.toml
 include _build_backend/backend.py

find_version.py

@@ -1,9 +1,9 @@
+from datetime import date
 import glob
 import io
 import os
 import os.path
 import sys
-import runpy
 import subprocess
 import re
 import sysconfig
@@ -22,6 +22,7 @@ def main():
     build_java = "ON" if get_build_env_var_by_name("java") else "OFF"
     build_rolling = get_build_env_var_by_name("rolling")
 
+    # TODO(@Breakthrough): What architectures should we ship?
     # fermi     = "2.0"
     # kepler    = "3.0;3.5;3.7"
     # maxwell   = "5.0;5.2"
@@ -32,7 +33,7 @@ def main():
     # lovelace  = "8.9"
     # hopper    = "9.0"
     # blackwell =  "10.0;12.0"
-    cuda_arch_bin = "6.0;6.1;7.0;7.5,8.0"
+    cuda_arch_bin = "5.0;5.2;6.0;6.1;7.0;7.5,8.0"
     cuda_arch_ptx = "8.0"
 
     # NOTE: since 2.3.0 numpy upgraded from manylinux2014 to manylinux_2_28
@@ -81,8 +82,8 @@ def main():
                     ["submodule", "update", "--init", "--recursive", "opencv_contrib"]
                 )
 
-    package_version, build_contrib, build_headless, build_rolling = get_and_set_info(
-        build_contrib, build_headless, build_rolling, is_CI_build, cuda_arch_bin, cuda_arch_ptx
+    package_version, build_contrib, build_headless, build_rolling = generate_version_file(
+        build_contrib, build_headless, build_rolling, cuda_arch_bin, cuda_arch_ptx
     )
 
     # https://stackoverflow.com/questions/1405913/python-32bit-or-64bit-mode
@@ -500,24 +501,83 @@ def _classify_installed_files_override(
         )
 
 
-def get_and_set_info(contrib, headless, rolling, ci_build, cuda_arch_bin, cuda_arch_ptx):
-    # cv2/version.py should be generated by running find_version.py
+def generate_version_file(contrib, headless, rolling, cuda_arch_bin, cuda_arch_ptx):
     version = {}
-    here = os.path.abspath(os.path.dirname(__file__))
-    version_file = os.path.join(here, "cv2", "version.py")
 
-    # generate a fresh version.py always when Git repository exists
-    # (in sdists the version.py file already exists)
-    if os.path.exists(".git"):
-        old_args = sys.argv.copy()
-        sys.argv = ["", str(contrib), str(headless), str(rolling), str(ci_build), str(cuda_arch_bin), str(cuda_arch_ptx)]
-        runpy.run_path("find_version.py", run_name="__main__")
-        sys.argv = old_args
+    # generate version.py
+    opencv_version = ""
+    # dig out the version from OpenCV sources
+    version_file_path = "opencv/modules/core/include/opencv2/core/version.hpp"
 
-    with open(version_file) as fp:
-        exec(fp.read(), version)
+    with open(version_file_path, "r") as f:
+        for line in f:
+            words = line.split()
 
-    return version["opencv_version"], version["contrib"], version["headless"], version["rolling"]
+            if "CV_VERSION_MAJOR" in words:
+                opencv_version += words[2]
+                opencv_version += "."
+
+            if "CV_VERSION_MINOR" in words:
+                opencv_version += words[2]
+                opencv_version += "."
+
+            if "CV_VERSION_REVISION" in words:
+                opencv_version += words[2]
+                break
+
+    # used in local dev releases
+    git_hash = (
+        subprocess.check_output(["git", "rev-parse", "--short", "HEAD"])
+        .splitlines()[0]
+        .decode()
+    )
+    # this outputs the annotated tag if we are exactly on a tag, otherwise <tag>-<n>-g<shortened sha-1>
+    try:
+        tag = (
+            subprocess.check_output(
+                ["git", "describe", "--tags"], stderr=subprocess.STDOUT
+            )
+            .splitlines()[0]
+            .decode()
+            .split("-")
+        )
+    except subprocess.CalledProcessError as e:
+        # no tags reachable (e.g. on a topic branch in a fork), see
+        # https://stackoverflow.com/questions/4916492/git-describe-fails-with-fatal-no-names-found-cannot-describe-anything
+        if e.output.rstrip() == b"fatal: No names found, cannot describe anything.":
+            tag = []
+        else:
+            print(e.output)
+            raise
+
+
+    # TODO(@Breakthrough): Update this so that we just use the tag name directly as the version
+    # identifier. We might need to strip off a "v" suffix if we include that in the repo, but it
+    # will greatly simplify things. This can also get out of sync since we cache artifacts now.
+
+    if len(tag) == 1:
+        # tag identifies the build and should be a sequential revision number
+        version = tag[0]
+        opencv_version += ".{}".format(version)
+    # rolling has converted into string using generate_version_file() function in setup.py
+    elif rolling == "True":
+        # rolling version identifier, will be published in a dedicated rolling PyPI repository
+        version = date.today().strftime('%Y%m%d')
+        opencv_version += ".{}".format(version)
+    else:
+        # local version identifier, not to be published on PyPI
+        version = git_hash
+        opencv_version += "+{}".format(version)
+
+    with open("cv2/version.py", "w") as f:
+        f.write('opencv_version = "{}"\n'.format(opencv_version))
+        f.write("contrib = {}\n".format(contrib))
+        f.write("headless = {}\n".format(headless))
+        f.write("rolling = {}\n".format(rolling))
+        f.write("cuda_arch_bin = {}".format(cuda_arch_bin))
+        f.write("cuda_arch_ptx = {}".format(cuda_arch_ptx))
+
+    return opencv_version, contrib, headless, rolling
 
 
 def get_build_env_var_by_name(flag_name):