import django
from django.conf import settings

settings.configure(
    DATABASES={'default': {'ENGINE': 'django.db.backends.postgresql', 'NAME': 'pgvector_python_test'}},
    INSTALLED_APPS=[],
)
django.setup()

from django import forms
from django.db import connection, models
from pgvector.django import VectorField
from rest_framework import serializers


class Item(models.Model):
    embedding = VectorField(dimensions=3)

    class Meta:
        app_label = 'repro'


with connection.cursor() as cursor:
    cursor.execute('CREATE EXTENSION IF NOT EXISTS vector')
    cursor.execute('DROP TABLE IF EXISTS repro_item')
with connection.schema_editor() as editor:
    editor.create_model(Item)


class ItemForm(forms.ModelForm):
    class Meta:
        model = Item
        fields = ['embedding']


class ItemSerializer(serializers.ModelSerializer):
    class Meta:
        model = Item
        fields = ['embedding']


def check(name, fn):
    try:
        print(f'{name}: {fn()}')
    except Exception as e:
        print(f'{name}: {type(e).__name__}: {e}')


def form(value):
    f = ItemForm(data={'embedding': value})
    if not f.is_valid():
        return f'invalid {f.errors.get_json_data()}'
    f.save()
    return 'saved'


def serializer(value):
    s = ItemSerializer(data={'embedding': value})
    if not s.is_valid():
        return f'invalid {s.errors}'
    s.save()
    return 'saved'


check('form [1,2,x]', lambda: form('[1,2,x]'))
check('form [1,2]', lambda: form('[1,2]'))
check('serializer [1, 2, "x"]', lambda: serializer([1, 2, 'x']))
check('serializer [1, 2]', lambda: serializer([1, 2]))

form [1,2,x]: ValueError: could not convert string to float: 'x'
form [1,2]: DataError: expected 3 dimensions, not 2
serializer [1, 2, "x"]: ValueError: could not convert string to float: 'x'
serializer [1, 2]: DataError: expected 3 dimensions, not 2

  .../django/forms/models.py", line 503, in _post_clean
  .../django/db/models/base.py", line 1713, in full_clean
  .../django/db/models/base.py", line 1768, in clean_fields
  .../django/db/models/fields/__init__.py", line 875, in clean
  .../pgvector/django/vector.py", line 33, in to_python
  .../pgvector/vector.py", line 112, in _from_db
  .../pgvector/vector.py", line 88, in _from_text
ValueError: could not convert string to float: 'x'