def save(request, address_id=None):
user=request.user
if request.method =='POST':
form=AddressForm(request.POST)
if form.is_valid():
if form.cleaned_data['address_id']:
address=Address.objects.get(pk=form.cleaned_data['address_id'])
else:
address=Address()
address.name=form.cleaned_data['name']
address.address=form.cleaned_data['address']
address.postal_code=form.cleaned_data['postal_code']
address.user=user
address.save()
return HttpResponseRedirect('/address/')
else:
if address_id:
address=Address.objects.get(pk=address_id)
if address.user==request.user:
form=AddressForm(initial={'address_id': address.id,
'name': address.name, 'address': address.address,
'postal_code': address.postal_code})
else:
form=AddressForm()
return render_to_response('address_form.html', {
'form': form},
context_instance=RequestContext(request))
def save(self, *args, **kwargs):
if self.instance.address_id:
addr = self.instance.address
else:
addr = Address()
addr.address = self.cleaned_data['address']
addr.zipcode = self.cleaned_data['zipcode']
addr.city = self.cleaned_data['city']
addr.country = self.cleaned_data['country']
addr.province_code = self.cleaned_data['province_code']
addr.save()
self.instance.address = addr
brand = super(SABrandForm, self).save(*args, **kwargs)
return brand
def update_address(self, address, prefix):
if address is None:
address = Address()
address.address = '\n'.join([
self.cleaned_data[prefix + 'address1'],
self.cleaned_data[prefix + 'address2'],
self.cleaned_data[prefix + 'city']
])
address.post_code = self.cleaned_data[prefix + 'post_code']
address_region_ser = RegionSerializer(
data={
'country': self.cleaned_data[prefix + 'country'],
'name': self.cleaned_data[prefix + 'state'],
'code': self.cleaned_data[prefix + 'state']
})
address_region_ser.is_valid(raise_exception=True)
address_region = address_region_ser.save()
address.region = address_region
address.save()
return address
