Fix: Old pastes don't have annotations

PythonistaGuild · May 29, 2024 · 90ccf83 · 90ccf83
1 parent 60d163b
commit 90ccf83
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/core/models.py b/core/models.py
@@ -66,7 +66,7 @@ def __init__(self, record: asyncpg.Record | dict[str, Any]) -> None:
         self.loc: int = record["loc"]
         self.charcount: int = record["charcount"]
         self.index: int = record["file_index"]
-        self.annotation: str = record["annotation"]
+        self.annotation: str | None = record["annotation"]
         self.warning_positions: list[int] = record["warning_positions"]
 
 

diff --git a/views/htmx.py b/views/htmx.py
@@ -54,7 +54,7 @@ def highlight_code(self, *, files: list[dict[str, Any]]) -> str:
             filename = "_".join(filename.splitlines())
 
             raw_url: str = f'/raw/{file["parent_id"]}'
-            annotation: str = file["annotation"]
+            annotation: str = file["annotation"] or ""
             positions: list[int] = file.get("warning_positions", [])
             original: str = file["content"]