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))
