def save(self, commit=True):
new_user = super(UserCreationForm, self).save(commit=False)
password = self.cleaned_data['password1']
email = self.cleaned_data['email']
new_user.set_password(password)
new_user.email = email
new_profile = Profile(user=new_user, role=self.cleaned_data['role'])
if commit:
new_user.save()
new_profile.user = new_user
new_profile.save()
return new_user
def register(request):
form = UserCreationForm(request.POST or None)
if request.POST:
if form.is_valid():
user = form.save()
profile = Profile()
profile.user = user
profile.save()
return redirect('login')
else:
for error in form.errors.items():
messages.warning(request, error)
context = {'form': form}
return render(request, "main/register.html", context)
def user_register_plan(request, plan_url):
"plan_url of 'free' means free plan"
if request.user.is_authenticated():
return change_plan(request, plan_url)
if request.method == 'POST':
form = RegisterForm(request.POST)
if form.is_valid():
plan_id = 0
plan = None
# create the user
user = User.objects.create_user(form.cleaned_data.get('username'),
form.cleaned_data.get('email'),
form.cleaned_data.get('password'))
user.save()
# create a band
band = Band()
band.title = form.cleaned_data.get('artist_name')
band.total_space = settings.BAND_INIT_SPACE
band.save()
# create a profile
profile = Profile()
profile.user = user
profile.solo_band = band
profile.activated = False
profile.activate_code = create_hash(32)
profile.logon_count = 0
profile.band_count_limit = settings.FREE_BAND_LIMIT
profile.save()
# make them a manager
manager = BandMember()
manager.user = user
manager.band = band
manager.role = BandMember.MANAGER
manager.save()
# send an activation email
subject = design.account_confirmation
context = Context({
'user': user,
'activate_url': request.build_absolute_uri(reverse('confirm', args=[user.username, profile.activate_code])),
'host': request.get_host(),
})
message_txt = get_template('email/activation.txt').render(context)
message_html = get_template('email/activation.html').render(context)
send_html_mail(subject, message_txt, message_html, [user.email])
return HttpResponseRedirect(reverse("register_pending"))
else:
try:
plan = AccountPlan.objects.get(url=plan_url)
plan_id = plan.id
except AccountPlan.DoesNotExist:
plan = None
plan_id = 0
form = RegisterForm(initial={'plan': plan_id})
return render_to_response('register.html', {'form': form}, context_instance=RequestContext(request))
def user_register_plan(request, plan_url):
"plan_url of 'free' means free plan"
if request.user.is_authenticated():
return change_plan(request, plan_url)
if request.method == 'POST':
form = RegisterForm(request.POST)
if form.is_valid():
plan_id = 0
plan = None
# create the user
user = User.objects.create_user(form.cleaned_data.get('username'),
form.cleaned_data.get('email'),
form.cleaned_data.get('password'))
user.save()
# create a band
band = Band()
band.title = form.cleaned_data.get('artist_name')
band.total_space = settings.BAND_INIT_SPACE
band.save()
# create a profile
profile = Profile()
profile.user = user
profile.solo_band = band
profile.activated = False
profile.activate_code = create_hash(32)
profile.logon_count = 0
profile.band_count_limit = settings.FREE_BAND_LIMIT
profile.save()
# make them a manager
manager = BandMember()
manager.user = user
manager.band = band
manager.role = BandMember.MANAGER
manager.save()
# send an activation email
subject = design.account_confirmation
context = Context({
'user':
user,
'activate_url':
request.build_absolute_uri(
reverse('confirm',
args=[user.username, profile.activate_code])),
'host':
request.get_host(),
})
message_txt = get_template('email/activation.txt').render(context)
message_html = get_template('email/activation.html').render(
context)
send_html_mail(subject, message_txt, message_html, [user.email])
return HttpResponseRedirect(reverse("register_pending"))
else:
try:
plan = AccountPlan.objects.get(url=plan_url)
plan_id = plan.id
except AccountPlan.DoesNotExist:
plan = None
plan_id = 0
form = RegisterForm(initial={'plan': plan_id})
return render_to_response('register.html', {'form': form},
context_instance=RequestContext(request))
