Restore 0.1.5 version from stash
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user