Improve JSON typing

Author: emyller

api/projects/code_references/models.py

@@ -2,6 +2,8 @@
 from django.db import models
 from django.db.models.expressions import Func
 
+from projects.code_references.types import JSONCodeReference
+
 
 class FeatureFlagCodeReferencesScan(models.Model):
     """
@@ -16,7 +18,7 @@ class FeatureFlagCodeReferencesScan(models.Model):
 
     repository_url = models.URLField()  # Provider-agnostic
     revision = models.CharField(max_length=100)
-    code_references = models.JSONField(default=list)
+    code_references = models.JSONField[list[JSONCodeReference]](default=list)
 
     created_at = models.DateTimeField(auto_now_add=True, db_index=True)
 

api/projects/code_references/types.py

@@ -1,5 +1,12 @@
 from dataclasses import dataclass
 from datetime import datetime
+from typing import TypedDict
+
+
+class JSONCodeReference(TypedDict):
+    feature_name: str
+    file_path: str
+    line_number: int
 
 
 @dataclass