44 lines
1.4 KiB
Python
Executable File
44 lines
1.4 KiB
Python
Executable File
from django.db.models import CharField
|
|
|
|
from location_field import forms
|
|
|
|
|
|
class BaseLocationField:
|
|
"""Base class for location fields."""
|
|
|
|
def __init__(self, based_field=None, zoom=2, default=None, *args, **kwargs):
|
|
self._based_field = based_field
|
|
self._zoom = zoom
|
|
self._default = default
|
|
self.default = default
|
|
|
|
def formfield(self, **kwargs):
|
|
return super().formfield(
|
|
form_class=self.formfield_class,
|
|
based_field=self._based_field,
|
|
zoom=self._zoom,
|
|
default=self._default,
|
|
**kwargs,
|
|
)
|
|
|
|
|
|
class PlainLocationField(BaseLocationField, CharField):
|
|
"""A CharField that stores location data as 'latitude,longitude,zoom'."""
|
|
|
|
formfield_class = forms.PlainLocationField
|
|
|
|
def __init__(self, based_field=None, zoom=None, max_length=63, *args, **kwargs):
|
|
BaseLocationField.__init__(
|
|
self, based_field=based_field, zoom=zoom, *args, **kwargs
|
|
)
|
|
CharField.__init__(self, max_length=max_length, *args, **kwargs)
|
|
|
|
def deconstruct(self):
|
|
"""Return enough information to recreate the field."""
|
|
name, path, args, kwargs = super().deconstruct()
|
|
if self._based_field is not None:
|
|
kwargs["based_field"] = self._based_field
|
|
if self._zoom is not None:
|
|
kwargs["zoom"] = self._zoom
|
|
return name, path, args, kwargs
|