def edit_investor(request):
user = request.user
group = user.groups.first()
if str(group) == "Investor":
investor = get_object_or_404(Investor, user_id=user.id)
if request.method == 'POST':
user_form = UpdateForm(
request.POST,
instance=user,
initial={'fullname': user.first_name + " " + user.last_name})
investor_form = InvestorForm(request.POST,
request.FILES,
instance=investor)
if user_form.is_valid() and investor_form.is_valid():
user_form.save()
investor_form.save()
return HttpResponseRedirect("/profile/" + str(user.id) + "/")
else:
user_form = UpdateForm(
instance=user,
initial={'fullname': user.first_name + " " + user.last_name})
investor_form = InvestorForm(instance=investor)
context = {
'user_form': user_form,
'investor_form': investor_form,
}
return render(request, 'edit_investor.html', context)
else:
return HttpResponseRedirect("/index/")
def edit_person(request):
user = request.user
group = user.groups.first()
if str(group) == "Person":
person = get_object_or_404(Person, user_id=user.id)
address = person.address
if request.method == 'POST':
user_form = UpdateForm(
request.POST,
instance=user,
initial={'fullname': user.first_name + " " + user.last_name})
address_form = AddressForm(request.POST, instance=address)
person_form = PersonForm(request.POST,
request.FILES,
instance=person)
if user_form.is_valid() and address_form.is_valid(
) and person_form.is_valid():
user_form.save()
address_form.save()
person_form.save()
return HttpResponseRedirect("/profile/" + str(user.id) + "/")
else:
user_form = UpdateForm(
instance=user,
initial={'fullname': user.first_name + " " + user.last_name})
address_form = AddressForm(instance=address)
person_form = PersonForm(instance=person)
context = {
'user_form': user_form,
'address_form': address_form,
'person_form': person_form,
}
return render(request, 'edit_person.html', context)
else:
return HttpResponseRedirect("/index/")
def post(self,request,*args,**kwargs):
#print(json.loads(request.body))
print(request.body)
print(request.POST)
form = UpdateForm(request.POST)
if form.is_valid():
obj = form.save(commit=True)
obj_data = obj.serialize1()
return HttpResponse(obj_data,content_type="application/json",status=200)
json_data = json.dumps({
"message":"sorry bro"
})
return HttpResponse(json_data, content_type='application/json', status=200)
def edit_startup(request):
user = request.user
group = user.groups.first()
if str(group) == "Startup":
startup = get_object_or_404(Startup, user_id=user.id)
address = startup.address
if request.method == 'POST':
user_form = UpdateForm(
request.POST,
instance=user,
initial={'fullname': user.first_name + " " + user.last_name})
address_form = AddressForm(request.POST, instance=address)
startup_form = StartupForm(request.POST,
request.FILES,
instance=startup)
if user_form.is_valid() and address_form.is_valid(
) and startup_form.is_valid():
user_form.save()
address_form.save()
startup_form.save()
return HttpResponseRedirect("/profile/" + str(user.id) + "/")
else:
user_form = UpdateForm(
instance=user,
initial={'fullname': user.first_name + " " + user.last_name})
address_form = AddressForm(instance=address)
startup_form = StartupForm(instance=startup)
context = {
'user_form': user_form,
'address_form': address_form,
'startup_form': startup_form,
}
return render(request, 'edit_startup.html', context)
else:
return HttpResponseRedirect("/index/")
