def register(request): if request.method == 'GET': return render_to_response('register.html', {'form' : RegistrationForm()}, context_instance=RequestContext(request)) form = RegistrationForm(request.POST) if form.is_valid(): user_data = {} user_data['username'] = form.cleaned_data['username'] user_data['password'] = form.cleaned_data['password1'] user_data['email'] = form.cleaned_data['email'] user = User.objects.create_user(**user_data) user.first_name = form.cleaned_data['first_name'] user.last_name = form.cleaned_data['last_name'] user.save() # Create a new category with the same name as the username category = Category(name=user.username, owner=user) category.category_parent = category.id category.save() user = authenticate(username=user_data['username'], password=user_data['password']) login(request, user) return redirect('/') return render_to_response('register.html', {'form' : form}, context_instance=RequestContext(request))
def add_category(request): if request.method == 'GET': return render_to_response('add_category.html', {'form' : CategoryForm(user=request.user)}, context_instance=RequestContext(request)) form = CategoryForm(request.POST) if form.is_valid(): category_data = form.cleaned_data category_data['owner'] = request.user category = Category(**category_data) category.save() return redirect('/') return render_to_response('add_category.html', {'form' : form}, context_instance=RequestContext(request))