Add user directory on registration

This commit is contained in:
Patrick Jentsch 2019-11-14 10:01:45 +01:00
parent bab479db20
commit baf06d3106

View File

@ -1,11 +1,13 @@
from app import db from app import db
from app.email import send_email from app.email import send_email
from app.models import User from app.models import User
from flask import flash, redirect, render_template, request, url_for from flask import (current_app, flash, redirect, render_template, request,
url_for)
from flask_login import current_user, login_required, login_user, logout_user from flask_login import current_user, login_required, login_user, logout_user
from . import auth from . import auth
from .forms import (LoginForm, PasswordResetForm, PasswordResetRequestForm, from .forms import (LoginForm, PasswordResetForm, PasswordResetRequestForm,
RegistrationForm) RegistrationForm)
import os
@auth.route('/login', methods=['GET', 'POST']) @auth.route('/login', methods=['GET', 'POST'])
@ -46,14 +48,22 @@ def register():
username=form.username.data) username=form.username.data)
db.session.add(user) db.session.add(user)
db.session.commit() db.session.commit()
token = user.generate_confirmation_token() dir = os.path.join(current_app.config['OPAQUE_STORAGE_DIRECTORY'],
send_email(user.email, str(job.user_id), 'jobs', str(job.id))
'Confirm Your Account', try:
'auth/email/confirm', os.makedirs(dir)
token=token, except OSError:
user=user) flash('[ERROR]!')
flash('A confirmation email has been sent to you by email.') user.delete()
return redirect(url_for('auth.login')) else:
token = user.generate_confirmation_token()
send_email(user.email,
'Confirm Your Account',
'auth/email/confirm',
token=token,
user=user)
flash('A confirmation email has been sent to you by email.')
return redirect(url_for('auth.login'))
return render_template('auth/register.html.j2', return render_template('auth/register.html.j2',
form=form, form=form,
title='Register') title='Register')