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.email import send_email
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 . import auth
from .forms import (LoginForm, PasswordResetForm, PasswordResetRequestForm,
RegistrationForm)
import os
@auth.route('/login', methods=['GET', 'POST'])
@ -46,6 +48,14 @@ def register():
username=form.username.data)
db.session.add(user)
db.session.commit()
dir = os.path.join(current_app.config['OPAQUE_STORAGE_DIRECTORY'],
str(job.user_id), 'jobs', str(job.id))
try:
os.makedirs(dir)
except OSError:
flash('[ERROR]!')
user.delete()
else:
token = user.generate_confirmation_token()
send_email(user.email,
'Confirm Your Account',