class UbigeoFormField(forms.MultiValueField):
def __init__(self, *args, **kwargs):
ubigeos = Ubigeo.objects.filter(parent__isnull=True)
regiones = ubigeos.exclude(ubigeo__startswith='9')
if 'ubigeo' in kwargs:
if kwargs['ubigeo'] == constant.ONLY_INTERNATIONAL:
regiones = regiones.filter(ubigeo__startswith='9')
elif kwargs['ubigeo'] == constant.ALL:
regiones = Ubigeo.objects.filter(parent__isnull=True)
provincias = Ubigeo.objects.filter(parent=regiones[0])
distritos = Ubigeo.objects.filter(parent=provincias[0])
self.fields = (
ModelChoiceField(queryset=regiones),
ModelChoiceField(queryset=provincias),
ModelChoiceField(queryset=distritos),
)
self.widget = UbigeoWidget(
regiones = self.fields[0]._get_choices(),
provincias = self.fields[1]._get_choices(),
distritos = self.fields[2]._get_choices()
)
super(UbigeoFormField, self).__init__(
self.fields,
self.widget,
*args
)
def compress(self, data_list):
if data_list:
return data_list[2]
return None
def clean(self, value):
if value[2]:
distrito = Ubigeo.objects.get(pk=value[2])
self.fields[1].queryset = Ubigeo.objects.filter(
parent=distrito.parent.parent
)
self.fields[2].queryset = Ubigeo.objects.filter(parent=distrito.parent)
return distrito
return None
def prepare_value(self, value):
if value is None:
value=constant.DISTRICT_DEFAULT
distrito = Ubigeo.objects.get(pk=value[2] if isinstance(value,list) else value)
self.fields[1].queryset = Ubigeo.objects.filter(
parent=distrito.parent.parent
)
self.fields[2].queryset = Ubigeo.objects.filter(parent=distrito.parent)
self.widget.provincias = self.fields[1]._get_choices()
self.widget.distritos = self.fields[2]._get_choices()
self.widget.decompress(distrito)
return value
def __init__(self, *args, **kwargs):
ubigeos = Ubigeo.objects.filter(parent__isnull=True)
regiones = ubigeos.exclude(ubigeo__startswith='9')
if 'ubigeo' in kwargs:
if kwargs['ubigeo'] == constant.ONLY_INTERNATIONAL:
regiones = regiones.filter(ubigeo__startswith='9')
elif kwargs['ubigeo'] == constant.ALL:
regiones = Ubigeo.objects.filter(parent__isnull=True)
provincias = Ubigeo.objects.filter(parent=regiones[0])
distritos = Ubigeo.objects.filter(parent=provincias[0])
self.fields = (
ModelChoiceField(queryset=regiones),
ModelChoiceField(queryset=provincias),
ModelChoiceField(queryset=distritos),
)
self.widget = UbigeoWidget(
regiones = self.fields[0]._get_choices(),
provincias = self.fields[1]._get_choices(),
distritos = self.fields[2]._get_choices()
)
super(UbigeoFormField, self).__init__(
self.fields,
self.widget,
*args
)
class UbigeoFormField(forms.MultiValueField):
def __init__(self, *args, **kwargs):
regiones = Ubigeo.objects.filter(parent__isnull=True).order_by('name')
if 'ubigeo' in kwargs:
if kwargs['ubigeo'] == constant.ONLY_INTERNATIONAL:
regiones = regiones.filter(
ubigeo__startswith='9').order_by('name')
elif kwargs['ubigeo'] == constant.ONLY_PERU:
regiones = regiones.exclude(ubigeo__startswith='9')
provincias = Ubigeo.objects.filter(parent=regiones[0]).order_by('name')
distritos = Ubigeo.objects.filter(
parent=provincias[0]).order_by('name')
self.fields = (
ModelChoiceField(queryset=regiones),
ModelChoiceField(queryset=provincias),
ModelChoiceField(queryset=distritos),
)
self.widget = UbigeoWidget(regiones=self.fields[0]._get_choices(),
provincias=self.fields[1]._get_choices(),
distritos=self.fields[2]._get_choices())
super(UbigeoFormField, self).__init__(self.fields, self.widget, *args)
def compress(self, data_list):
if data_list:
return data_list[2]
return None
def clean(self, value):
distrito = Ubigeo.objects.get(pk=value[2])
self.fields[1].queryset = Ubigeo.objects.filter(
parent=distrito.parent.parent)
self.fields[2].queryset = Ubigeo.objects.filter(parent=distrito.parent)
return distrito
def prepare_value(self, value):
if value is None:
value = constant.DISTRICT_DEFAULT
distrito = Ubigeo.objects.get(
pk=value[2] if isinstance(value, list) else value)
self.fields[1].queryset = Ubigeo.objects.filter(
parent=distrito.parent.parent)
self.fields[2].queryset = Ubigeo.objects.filter(parent=distrito.parent)
self.widget.provincias = self.fields[1]._get_choices()
self.widget.distritos = self.fields[2]._get_choices()
self.widget.decompress(distrito)
return value
def __init__(self, *args, **kwargs):
regiones = Ubigeo.objects.filter(parent__isnull=True).order_by('name')
if 'ubigeo' in kwargs:
if kwargs['ubigeo'] == constant.ONLY_INTERNATIONAL:
regiones = regiones.filter(
ubigeo__startswith='9').order_by('name')
elif kwargs['ubigeo'] == constant.ONLY_PERU:
regiones = regiones.exclude(ubigeo__startswith='9')
provincias = Ubigeo.objects.filter(parent=regiones[0]).order_by('name')
distritos = Ubigeo.objects.filter(
parent=provincias[0]).order_by('name')
self.fields = (
ModelChoiceField(queryset=regiones),
ModelChoiceField(queryset=provincias),
ModelChoiceField(queryset=distritos),
)
self.widget = UbigeoWidget(regiones=self.fields[0]._get_choices(),
provincias=self.fields[1]._get_choices(),
distritos=self.fields[2]._get_choices())
super(UbigeoFormField, self).__init__(self.fields, self.widget, *args)