Remove jinja tags from pyproject.toml and remove dependencies through the post hook

Author: foarsitter

Diff for: hooks/post_gen_project.py

@@ -4,6 +4,8 @@
 import string
 from pathlib import Path
 
+from tomlkit import dumps, parse
+
 try:
     # Inspired by
     # https://github.com/django/django/blob/master/django/utils/crypto.py
@@ -502,8 +504,130 @@ def main():
     if "{{ cookiecutter.use_async }}".lower() == "n":
         remove_async_files()
 
+    manipulate_pyproject()
+
     print(SUCCESS + "Project initialized, keep up the good work!" + TERMINATOR)
 
 
+def manipulate_pyproject():
+
+    path = Path("pyproject.toml")
+    project = parse(path.read_text())
+
+    if "{{cookiecutter.frontend_pipeline}}" != "Gulp":
+        del project["tool"]["pytest"]["ini_options"]["norecursedirs"]
+
+    if "{{cookiecutter.use_drf}}" == "y":
+        project["tool"]["mypy"]["plugins"].append("mypy_drf_plugin.main")
+    else:
+        remove_dependency(project, "djangorestframework")
+        remove_dependency(project, "django-cors-headers")
+        remove_dependency(project, "drf-spectacular")
+
+    if "{{cookiecutter.use_drf}}" == "n":
+        remove_dev_dependency(project, "djangorestframework-stubs")
+
+    if "{{cookiecutter.use_docker}}" == "y":
+        replace_dependency(project, "psycopg", "psycopg[c]")
+    else:
+        replace_dependency(project, "psycopg", "psycopg[binary]")
+
+    if "{{cookiecutter.use_async}}" == "n" and "{{cookiecutter.use_async}}" == "n":
+        remove_dev_dependency(project, "watchfiles")
+
+    if "{{cookiecutter.frontend_pipeline}}" != "Django Compressor":
+        remove_dependency(project, "rcssmin")
+        remove_dependency(project, "django-compressor")
+
+    if "{{cookiecutter.frontend_pipeline}}" != "Webpack":
+        remove_dependency(project, "django-webpack-loader")
+
+    if "{{cookiecutter.use_whitenoise}}" != "y":
+        remove_dependency(project, "whitenoise")
+
+    if "{{cookiecutter.use_whitenoise}}" != "n":
+        remove_dependency(project, "Collectfasta")
+
+    if "{{cookiecutter.use_sentry}}" != "y":
+        remove_dependency(project, "sentry-sdk")
+
+    if "{{cookiecutter.use_docker}}" != "n" or "{{cookiecutter.windows}}" != "y":
+        remove_dependency(project, "hiredis")
+
+    if "{{cookiecutter.use_celery}}" != "y":
+        remove_dev_dependency(project, "celery")
+        remove_dev_dependency(project, "django-celery-beat")
+
+        if "{{cookiecutter.use_docker}}" != "y":
+            remove_dev_dependency(project, "flower")
+
+    if "{{cookiecutter.use_async}}" != "y":
+        remove_dependency(project, "uvicorn[standard]")
+        remove_dependency(project, "uvicorn-worker")
+
+    if "{{cookiecutter.cloud_provider}}":
+        cloud_providers = {
+            "AWS": "s3",
+            "GCP": "google",
+            "Azure": "azure",
+        }
+
+        p = cloud_providers["{{cookiecutter.cloud_provider}}"]
+
+        if p:
+            replace_dependency(project, "django-storages", f"django-storages[{p}]")
+        else:
+            remove_dependency(project, "django-storages")
+
+    if "{{cookiecutter.mail_service}}":
+        mail_service_providers = {
+            "Mailgun": "mailgun",
+            "Amazon SES": "amazon-ses",
+            "Mailjet": "mailjet",
+            "Mandrill": "mandrill",
+            "Postmark": "postmark",
+            "Sendgrid": "sendgrid",
+            "Brevo": "brevo",
+            "SparkPost": "sparkpost",
+            "Other SMTP": "",
+        }
+
+        p = mail_service_providers["{{cookiecutter.mail_service}}"]
+        if p:
+            replace_dependency(project, "django-anymail", f"django-anymail[{p}]")
+        else:
+            remove_dependency(project, "django-anymail")
+
+    path.write_text(dumps(project))
+
+
+def replace_dependency(p, dep, target):
+    x = []
+    for i, d in enumerate(p["project"]["dependencies"]):
+        if d.startswith(dep):
+            x.append(d.replace(dep, target))
+        else:
+            x.append(d)
+
+    p["project"]["dependencies"] = x
+
+
+def remove_dependency(p, dep):
+    dependencies = []
+    for i, d in enumerate(p["project"]["dependencies"]):
+        if not d.startswith(dep):
+            dependencies.append(d)
+    p["project"]["dependencies"] = dependencies
+
+
+def remove_dev_dependency(p, dep):
+    dependencies = []
+    for i, d in enumerate(p["dependency-groups"]["dev"]):
+        if not d.startswith(dep):
+            dependencies.append(d)
+
+    p["dependency-groups"]["dev"] = dependencies
+
+
 if __name__ == "__main__":
     main()

Diff for: pyproject.toml

@@ -45,6 +45,7 @@ dependencies = [
   "requests==2.32.3",
   "ruff==0.11.4",
   "sh==2.1; sys_platform!='win23'",
+  "tomlkit>=0.13.2",
   "tox==4.23.2",
   "tox-uv>=1.17",
 ]

Diff for: tests/test_cookiecutter_generation.py

@@ -439,6 +439,8 @@ def test_pyproject_toml(cookies, context):
     assert data["project"]["authors"][0]["name"] == author_name
     assert data["project"]["name"] == context["project_slug"]
 
+    assert "norecursedirs" not in data["tool"]["pytest"]["ini_options"]
+
 
 def test_pre_commit_without_heroku(cookies, context):
     context.update({"use_heroku": "n"})

Diff for: tests/test_hooks.py

@@ -4,8 +4,14 @@
 from pathlib import Path
 
 import pytest
+from tomlkit import parse
 
-from hooks.post_gen_project import append_to_gitignore_file
+from hooks.post_gen_project import (
+    append_to_gitignore_file,
+    remove_dependency,
+    remove_dev_dependency,
+    replace_dependency,
+)
 
 
 @pytest.fixture
@@ -25,3 +31,21 @@ def test_append_to_gitignore_file(working_directory):
     linesep = os.linesep.encode()
     assert gitignore_file.read_bytes() == b"node_modules/" + linesep + b".envs/*" + linesep
     assert gitignore_file.read_text() == "node_modules/\n.envs/*\n"
+
+
+def test_replace_dependency():
+
+    def has_dependency(dependencies: list[str], dependency: str):
+        return any(dep.startswith(dependency) or dep == dependency for dep in dependencies)
+
+    p = Path(__file__).parent.parent / "{{cookiecutter.project_slug}}" / "pyproject.toml"
+    project = parse(p.read_text())
+
+    remove_dependency(project, "djangorestframework")
+    remove_dev_dependency(project, "djangorestframework-stubs")
+
+    replace_dependency(project, "psycopg", "psycopg[c]")
+
+    assert has_dependency(project["project"]["dependencies"], "psycopg[c]")
+    assert not has_dependency(project["project"]["dependencies"], "djangorestframework")
+    assert not has_dependency(project["dependency-groups"]["dev"], "djangorestframework-stubs")