def post(self, request):
receiver = request.POST.get('username')
address = request.POST.get('address')
postcode = request.POST.get('postcode')
mobile = request.POST.get('mobile')
if not all([receiver, address, postcode, mobile]):
return render(request, 'address.html',
{'errmsg': 'caoyaolu请输入地址信息'})
addr = Address()
addr.receiver = receiver
addr.address = address
addr.tel_phone = mobile
addr.zip_code = postcode
addr.user = request.user
exist_addr = Address.objects.GetDefaultAddress(user=request.user)
default_addr = addr
if not exist_addr:
addr.is_default = True
else:
default_addr = exist_addr
addr.save()
address_list = Address.objects.GetAddressList(request.user)
return render(
request, 'address.html', {
'errmsg': 'caoyaolu请输入地址信息',
'address': default_addr,
'page': 'address',
'address_list': address_list
})
def profile(request):
findprofile = Profile.objects.filter(user_id=request.user.id).first()
if findprofile == None:
print("No profile found, making one..")
tempprofile = Profile()
tempprofile.user = request.user
tempprofile.save()
findpicture = UserImage.objects.filter(user_id=request.user.id).first()
if findpicture == None:
findpicture = UserImage()
findpicture.user = request.user
findpicture.save()
if request.user.is_staff:
return redirect('/user/admin-option')
postform = User.objects.filter(id=request.user.id).first()
address = Address.objects.filter(user_id=request.user.id).first()
if address == None:
address = Address()
address.user = request.user
address.address = None
address.zip_code = None
address.city = None
address.country = None
if request.method == 'POST':
personalform = PersonalInfo(instance=postform, data=request.POST)
if personalform.is_valid():
postform.save()
addressform = AddressInfo(instance=address, data=request.POST)
if addressform.is_valid():
address.save()
messages.success(request, 'Nýjar upplýsingar vistaðar')
else:
messages.error(
request,
'Ekki gékk að vista heimilisfangs upplýsingar, vinsamlegast reyndu aftur.'
)
return redirect('/user/profile')
messages.error(
request,
'Ekki gékk að vista upplýsingar, vinsamlegast reyndu aftur.')
return render(
request, 'user/profile.html', {
'form': PersonalInfo(instance=postform),
'address': AddressInfo(instance=address)
})