56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
|
from django import forms
|
||
|
|
from django.db.models import Field
|
||
|
|
from ..utils import SparseVector
|
||
|
|
|
||
|
|
|
||
|
|
# https://docs.djangoproject.com/en/5.0/howto/custom-model-fields/
|
||
|
|
class SparseVectorField(Field):
|
||
|
|
description = 'Sparse vector'
|
||
|
|
empty_strings_allowed = False
|
||
|
|
|
||
|
|
def __init__(self, *args, dimensions=None, **kwargs):
|
||
|
|
self.dimensions = dimensions
|
||
|
|
super().__init__(*args, **kwargs)
|
||
|
|
|
||
|
|
def deconstruct(self):
|
||
|
|
name, path, args, kwargs = super().deconstruct()
|
||
|
|
if self.dimensions is not None:
|
||
|
|
kwargs['dimensions'] = self.dimensions
|
||
|
|
return name, path, args, kwargs
|
||
|
|
|
||
|
|
def db_type(self, connection):
|
||
|
|
if self.dimensions is None:
|
||
|
|
return 'sparsevec'
|
||
|
|
return 'sparsevec(%d)' % self.dimensions
|
||
|
|
|
||
|
|
def from_db_value(self, value, expression, connection):
|
||
|
|
return SparseVector._from_db(value)
|
||
|
|
|
||
|
|
def to_python(self, value):
|
||
|
|
return SparseVector._from_db(value)
|
||
|
|
|
||
|
|
def get_prep_value(self, value):
|
||
|
|
return SparseVector._to_db(value)
|
||
|
|
|
||
|
|
def value_to_string(self, obj):
|
||
|
|
return self.get_prep_value(self.value_from_object(obj))
|
||
|
|
|
||
|
|
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
|
||
|
|
return super().to_python(value)
|