def add(request):
if request.method == 'POST':
form = CustomerProfileForm(request.POST)
if form.is_valid():
c = form.save(commit=False)
c.user = request.user
c.save()
return HttpResponseRedirect('/accounts/customers')
else:
form = CustomerProfileForm()
return render(request, 'customers/add.html', {'form': form})
def edit(request, customer_id):
if request.method == 'POST':
instance = CustomerProfile.objects.get(pk=customer_id)
form = CustomerProfileForm(request.POST, instance=instance)
if form.is_valid():
c = form.save(commit=False)
c.save()
return HttpResponseRedirect('/accounts/customers/')
else:
try:
customer = CustomerProfile.objects.get(pk=customer_id)
form = CustomerProfileForm(instance=customer)
except CustomerProfile.DoesNotExist:
return HttpResponse(u'客户信息不存在')
return render(request, 'customers/edit.html', {'form': form, 'customer': customer})
