Add dependencies to uv in post_gen hook

Author: foarsitter

Diff for: hooks/post_gen_project.py

@@ -1,7 +1,10 @@
 import json
+import os
 import random
 import shutil
 import string
+import subprocess
+import sys
 from pathlib import Path
 
 try:
@@ -502,8 +505,52 @@ def main():
     if "{{ cookiecutter.use_async }}".lower() == "n":
         remove_async_files()
 
+    setup_dependencies()
+
     print(SUCCESS + "Project initialized, keep up the good work!" + TERMINATOR)
 
 
+def setup_dependencies():
+    print("Installing python dependencies using uv...")
+
+    if "{{ cookiecutter.use_docker }}".lower() == "y":
+        # Build the Docker service using Docker Compose
+        try:
+            subprocess.run(["docker", "compose", "-f", "docker-compose.local.yml", "build", "django"], check=True)
+        except subprocess.CalledProcessError as e:
+            print(f"Error building Docker service: {e}", file=sys.stderr)
+            sys.exit(1)
+
+        # Use Docker to run the uv command
+        uv_cmd = ["docker", "compose", "-f", "docker-compose.local.yml", "run", "--rm", "django", "uv"]
+    else:
+        # Use uv command directly
+        uv_cmd = ["uv"]
+
+    # Install production dependencies
+    try:
+        subprocess.run(uv_cmd + ["add", "-r", "requirements/production.txt"], check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Error installing production dependencies: {e}", file=sys.stderr)
+        sys.exit(1)
+
+    # Install local (development) dependencies
+    try:
+        subprocess.run(uv_cmd + ["add", "--dev", "-r", "requirements/local.txt"], check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Error installing local dependencies: {e}", file=sys.stderr)
+        sys.exit(1)
+
+    # Remove the requirements directory
+    if os.path.exists("requirements"):
+        try:
+            shutil.rmtree("requirements")
+        except Exception as e:
+            print(f"Error removing 'requirements' folder: {e}", file=sys.stderr)
+            sys.exit(1)
+
+    print("Setup complete!")
+
+
 if __name__ == "__main__":
     main()

Diff for: tests/test_cookiecutter_generation.py

@@ -150,7 +150,14 @@ def _fixture_id(ctx):
 
 def build_files_list(base_path: Path):
     """Build a list containing absolute paths to the generated files."""
-    return [dirpath / file_path for dirpath, subdirs, files in base_path.walk() for file_path in files]
+    f = []
+    for dirpath, subdirs, files in base_path.walk():
+        if ".venv" in subdirs:
+            subdirs.remove(".venv")
+
+        for file_path in files:
+            f.append(dirpath / file_path)
+    return f
 
 
 def check_paths(paths: Iterable[Path]):

Diff for: tox.ini

@@ -4,7 +4,7 @@ envlist = py312,black-template
 
 [testenv]
 passenv = AUTOFIXABLE_STYLES
-commands = pytest -n auto {posargs:./tests}
+commands = pytest --instafail -n auto {posargs:./tests}
 
 [testenv:black-template]
 deps = black

Diff for: uv.lock

@@ -1,5 +1,3 @@
--r production.txt
-
 Werkzeug[watchdog]==3.1.3 # https://github.com/pallets/werkzeug
 ipdb==0.13.13  # https://github.com/gotcha/ipdb
 {%- if cookiecutter.use_docker == 'y' %}