[dist] Update tweak_setup_cfg_for_headless.py to modify pyproject.toml

Author: Breakthrough

.github/workflows/build.yml

@@ -50,12 +50,15 @@ jobs:
           version: "0.5.11"
           python-version: ${{ matrix.python-version }}
 
-      # TODO(#257): Unpin OpenCV versions once we figure out why frame numbers are mismatching in tests.
       - name: Install Dependencies
         run: |
           uv pip install --upgrade build wheel virtualenv
-          uv pip install opencv-python-headless\<4.13 opencv-contrib-python-headless\<4.13 --only-binary :all:
-          uv pip install -r requirements_headless.txt -r docs/requirements.txt
+          # TODO: Figure out how to avoid installing the original OpenCV package.
+          uv pip install -e ".[dev]" -r docs/requirements.txt
+          # TODO(#257): Unpin OpenCV versions once we figure out why frame numbers are mismatching in tests.
+          # Use headless OpenCV variant on CI to reduce download size.
+          uv pip uninstall opencv-python opencv-contrib-python
+          uv pip install opencv-python-headless<4.13 opencv-contrib-python-headless<4.13 --only-binary :all:
 
       - name: Setup FFmpeg
         # TODO: This action currently does not work for non-x64 builders (e.g. macos-14):
@@ -79,7 +82,7 @@ jobs:
       - name: Build Package (Headless)
         shell: bash
         run: |
-          python dist/tweak_setup_cfg_for_headless.py
+          python dist/build_headless.py
           python -m build
           git reset --hard HEAD
 

.github/workflows/check-code-format.yml

@@ -35,8 +35,7 @@ jobs:
       - name: Install Dependencies
         run: |
           uv pip install --upgrade pip
-          uv pip install --upgrade yapf toml
-          uv pip install -r requirements_headless.txt
+          uv pip install --upgrade yapf toml ruff
 
       - name: Check Code Format (yapf)
         if: ${{ hashFiles('.style.yapf') != '' }}

.github/workflows/publish-pypi.yaml

@@ -103,13 +103,13 @@ jobs:
         run: |
           uv pip install --upgrade build wheel virtualenv
           uv pip install opencv-python-headless opencv-contrib-python-headless --only-binary :all:
-          uv pip install -r requirements_headless.txt -r docs/requirements.txt
+          uv pip install -e . -r docs/requirements.txt
 
       - name: Build Package
         run: |
           python dist/pre_release.py
           python -m build
-          python dist/tweak_setup_cfg_for_headless.py
+          python dist/build_headless.py
           python -m build
           mkdir pkg
           mv dist/*.tar.gz pkg/

.gitignore

@@ -10,8 +10,6 @@ site/
 build/
 # Working directory for distribution assembly
 dist/dvr-scan/*
-dist/dvr-scan-*
-dist/dvr_scan-*
 
 # Editors/environment
 .DS_Store

dist/.gitignore

@@ -0,0 +1,4 @@
+*
+!*.py
+!*.txt
+!.gitignore

dist/build_headless.py

@@ -0,0 +1,48 @@
+#
+#      DVR-Scan: Video Motion Event Detection & Extraction Tool
+#   --------------------------------------------------------------
+#       [  Site: https://www.dvr-scan.com/                 ]
+#       [  Repo: https://github.com/Breakthrough/DVR-Scan  ]
+#
+# Copyright (C) 2016 Brandon Castellano <http://www.bcastell.com>.
+#
+
+# This script generates a headless (no GUI) version of the package
+# by modifying the pyproject.toml file. When the project is next built,
+# the headless variant will be used.
+#
+# *WARNING*: This modifies the existing pyproject.toml file in place.
+# The changes must be reverted to restore the full package.
+#
+
+import os
+import re
+
+# Correctly locate pyproject.toml relative to the script's location
+script_dir = os.path.dirname(os.path.abspath(__file__))
+project_root = os.path.abspath(os.path.join(script_dir, ".."))
+pyproject_path = os.path.join(project_root, "pyproject.toml")
+
+with open(pyproject_path, "r") as f:
+    content = f.read()
+
+# Rename package to headless variant
+assert 'name = "dvr-scan"' in content
+content = content.replace('name = "dvr-scan"', 'name = "dvr-scan-headless"', 1)
+
+# Remove dvr-scan-app entry point
+assert "dvr-scan-app" in content
+content = re.sub(r'^dvr-scan-app\s*=.*\n', "", content, flags=re.MULTILINE)
+
+# Swap GUI opencv for headless opencv
+assert "opencv-contrib-python<4.13" in content
+content = content.replace('"opencv-contrib-python<4.13"', '"opencv-contrib-python-headless<4.13"', 1)
+
+# Remove screeninfo (GUI dependency)
+assert "screeninfo" in content
+content = re.sub(r'^\s*"screeninfo",?\n', "", content, flags=re.MULTILINE)
+
+with open(pyproject_path, "w") as f:
+    f.write(content)
+
+print("Successfully generated headless pyproject.toml.")

dist/tweak_setup_cfg_for_headless.py

@@ -52,18 +52,12 @@ dependencies = [
     "scenedetect>=0.6.2",
     "tqdm",
     "pillow",
-    # TODO(#257): Unpin OpenCV versions once we figure out why frame numbers are mismatching in tests.
-    "opencv-contrib-python-headless<4.13",
-]
-
-[project.optional-dependencies]
-gui = [
     # TODO(#257): Unpin OpenCV versions once we figure out why frame numbers are mismatching in tests.
     "opencv-contrib-python<4.13",
     "screeninfo",
 ]
 
-[dependency-groups]
+[project.optional-dependencies]
 dev = [
     "pytest",
     "ruff",