def create_poll(request):
current_user = get_current_user(request=request)
if request.method == "POST":
#Create poll
new_poll = Poll()
new_poll.pub_date = datetime.datetime.now()
new_poll.question = request.POST["question"]
new_poll.master = current_user
new_poll.save()
#Create answers
answers = [request.POST["answer1"], request.POST["answer2"], request.POST["answer3"], request.POST["answer4"], request.POST["answer5"], request.POST["answer6"]]
for answer in answers:
if answer != "":
new_choice = Choice()
new_choice.poll = new_poll
new_choice.choice_text = answer
new_choice.save()
return render(request, "add_poll.html", {"current_user": current_user})
