forked from mirrors/bookwyrm
67 lines
No EOL
1.9 KiB
Python
67 lines
No EOL
1.9 KiB
Python
'''Quick and dirty shim for JSONField and ArrayField compatibility on sqlite.
|
|
|
|
For more info and original code, see:
|
|
- https://medium.com/@philamersune/using-postgresql-jsonfield-in-sqlite-95ad4ad2e5f1
|
|
- https://gist.github.com/pvsune/2e5f9f9ae356d0bff633d896bc7d168b#file-django-sqlite-fields-py
|
|
'''
|
|
|
|
import json
|
|
|
|
from django.conf import settings
|
|
from django.contrib.postgres.fields import (
|
|
JSONField as DjangoJSONField,
|
|
ArrayField as DjangoArrayField,
|
|
)
|
|
from django.db.models import Field
|
|
|
|
|
|
class JSONField(DjangoJSONField):
|
|
pass
|
|
|
|
|
|
class ArrayField(DjangoArrayField):
|
|
pass
|
|
|
|
|
|
if 'sqlite' in settings.DATABASES['default']['ENGINE']:
|
|
class JSONField(Field):
|
|
def db_type(self, connection):
|
|
return 'text'
|
|
|
|
def from_db_value(self, value, expression, connection):
|
|
if value is not None:
|
|
return self.to_python(value)
|
|
return value
|
|
|
|
def to_python(self, value):
|
|
if value is not None:
|
|
try:
|
|
return json.loads(value)
|
|
except (TypeError, ValueError):
|
|
return value
|
|
return value
|
|
|
|
def get_prep_value(self, value):
|
|
if value is not None:
|
|
return str(json.dumps(value))
|
|
return value
|
|
|
|
def value_to_string(self, obj):
|
|
return self.value_from_object(obj)
|
|
|
|
|
|
class ArrayField(JSONField):
|
|
def __init__(self, base_field, size=None, **kwargs):
|
|
"""Care for DjangoArrayField's kwargs."""
|
|
self.base_field = base_field
|
|
self.size = size
|
|
return super().__init__(**kwargs)
|
|
|
|
def deconstruct(self):
|
|
"""Need to create migrations properly."""
|
|
name, path, args, kwargs = super().deconstruct()
|
|
kwargs.update({
|
|
'base_field': self.base_field.clone(),
|
|
'size': self.size,
|
|
})
|
|
return name, path, args, kwargs |