restore original languages in fixtures

Author: elfkuzco

backend/src/cms_backend/mill/__init__.py

@@ -7,12 +7,12 @@
 def update_language_codes():
     for code in MillContext.disallowed_language_codes:
         try:
-            pycountry.languages.remove_entry(alpha_3=code)
+            pycountry.languages.remove_entry(alpha_3=code)  # pyright: ignore[reportUnknownMemberType]
         except Exception as exc:
             logger.warning(f"failed to remove language code '{code}': {exc}")
 
     for code in MillContext.custom_language_codes:
-        pycountry.languages.add_entry(alpha_3=code)
+        pycountry.languages.add_entry(alpha_3=code)  # pyright: ignore[reportUnknownMemberType]
 
 
 update_language_codes()

backend/src/cms_backend/mill/context.py

@@ -23,7 +23,7 @@ def _parse_custom_language_codes(language_code: str | None) -> list[str]:
 
 def _validate_language_codes(language_codes: list[str]) -> list[str]:
     for code in language_codes:
-        if pycountry.languages.get(alpha_3=code) is None:
+        if pycountry.languages.get(alpha_3=code) is None:  # pyright: ignore[reportUnknownMemberType]
             raise ValueError(f"Code '{code}' is not a valid ISO 639-3 code.")
     return language_codes
 

backend/src/cms_backend/mill/processors/book.py

@@ -48,7 +48,7 @@ def check_book_zim_spec(book: Book) -> bool:
             return False
 
         language_code = book.zim_metadata["Language"]
-        if pycountry.languages.get(alpha_3=language_code) is None:
+        if pycountry.languages.get(alpha_3=language_code) is None:  # pyright: ignore[reportUnknownMemberType]
             book.events.append(
                 f"{getnow()}: book has unknown language code {language_code}"
             )

backend/tests/mill/processors/test_zimfarm_notification.py

@@ -7,8 +7,11 @@
 
 from collections.abc import Callable
 from pathlib import Path
+from typing import Any
 from uuid import UUID
 
+import pycountry
+import pytest
 from pytest import MonkeyPatch
 from sqlalchemy.orm import Session as OrmSession
 
@@ -24,7 +27,7 @@
 from cms_backend.mill import update_language_codes
 from cms_backend.mill.processors.zimfarm_notification import process_notification
 
-VALID_NOTIFICATION_CONTENT = {
+VALID_NOTIFICATION_CONTENT: dict[str, Any] = {
     "article_count": 1000,
     "media_count": 500,
     "size": 1000000,
@@ -48,6 +51,28 @@
 }
 
 
+@pytest.fixture(autouse=True)
+def restore_language_codes():
+    """Fixture to restore pycountry language codes after test modifications."""
+    original_entries = list(pycountry.languages)  # pyright: ignore[reportUnknownVariableType]
+    yield
+    current_entries = list(pycountry.languages)  # pyright: ignore[reportUnknownVariableType]
+    for entry in current_entries:  # pyright: ignore[reportUnknownVariableType]
+        try:
+            pycountry.languages.remove_entry(alpha_3=entry.alpha_3)  # pyright: ignore[reportUnknownMemberType,reportUnknownArgumentType]
+        except Exception:  # noqa: S110
+            pass
+
+    for entry in original_entries:  # pyright: ignore[reportUnknownVariableType]
+        try:
+            pycountry.languages.add_entry(  # pyright: ignore[reportUnknownMemberType]
+                alpha_3=entry.alpha_3,  # pyright: ignore[reportUnknownMemberType,reportUnknownArgumentType]
+                name=entry.name,  # pyright: ignore[reportUnknownMemberType,reportUnknownArgumentType]
+            )
+        except Exception:  # noqa: S110
+            pass
+
+
 class TestBadNotifications:
     """Test notifications that fail validation and are marked as bad_notification."""