Setup update license badge fix #57 (#79)

* Setup update license badge

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Setup no license -> remove badge

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Escape - in badge; Refactoring

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* ruff

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Reinder Vos de Wael <reinder.vosdewael@childmind.org>

Author: 3 people

setup/licenses.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 import json
+import re
 from datetime import datetime
 from typing import Optional
 from urllib import request
@@ -81,6 +82,33 @@ def request_license() -> Optional[dict[str, str]]:
     return final_license
 
 
+def replace_license_badge(content: str, repo_license: Optional[dict[str, str]]) -> str:
+    """Replaces the license badge with the specified license.
+
+    Args:
+        content: Content of any file that might contain a license badge.
+        repo_license: The license to replace the current license with. If None,
+        the current license badge will be deleted.
+
+    Returns:
+        The content with the license badge replaced.
+    """
+    if repo_license is None:
+        # remove line containing license badge
+        return re.sub(
+            r"\[!\[.*License\]\(https://img\.shields\.io/badge/license.*\)\]\(.*\)\n",
+            "",
+            content,
+        )
+    # shield.io uses -- as an escape character, so we need to replace - with --
+    license_name_upper = repo_license["key"].upper().replace("-", "--")
+    return content.replace(
+        "![MIT License]" "(https://img.shields.io/badge/license-MIT-blue.svg)]",
+        f"![{license_name_upper} License]"
+        f"(https://img.shields.io/badge/license-{license_name_upper}-blue.svg)]",
+    )
+
+
 def replace_license(repo_license: Optional[dict[str, str]]) -> None:
     """Replaces the license file in the repository with the specified license.
 

setup_template.py

@@ -86,6 +86,8 @@ def main() -> None:
         content = content.replace("ENTER_YOUR_EMAIL_ADDRESS", email)
         content = content.replace("Reinder Vos de Wael", username)
 
+        content = licenses.replace_license_badge(content, repo_license)
+
         if content != content_before:
             print(f"Updating {file.relative_to(DIR_REPO)}")
             with open(file, "w", encoding="utf-8") as f: