Improved SparseVectorField forms for Django [skip ci]

ankane · ankane · commit 6d41b794b64e · 2024-06-12T15:10:30.000-07:00
diff --git a/pgvector/django/sparsevec.py b/pgvector/django/sparsevec.py
@@ -39,7 +39,16 @@ def formfield(self, **kwargs):
         return super().formfield(form_class=SparseVectorFormField, **kwargs)
 
 
+class SparseVectorWidget(forms.TextInput):
+    def format_value(self, value):
+        if isinstance(value, SparseVector):
+            value = value.to_text()
+        return super().format_value(value)
+
+
 class SparseVectorFormField(forms.CharField):
+    widget = SparseVectorWidget
+
     def to_python(self, value):
         if isinstance(value, str) and value == '':
             return None
diff --git a/tests/test_django.py b/tests/test_django.py
@@ -373,8 +373,8 @@ def test_sparsevec_form_instance(self):
         Item(id=1, sparse_embedding=[1, 2, 3]).save()
         item = Item.objects.get(pk=1)
         form = SparseVectorForm(instance=item)
-        # TODO fix
-        # assert 'value="[1.0, 2.0, 3.0]"' in form.as_div()
+        # TODO improve
+        assert 'value="{1:1.0,2:2.0,3:3.0}/3"' in form.as_div()
 
     def test_sparsevec_form_save(self):
         Item(id=1, sparse_embedding=[1, 2, 3]).save()
