def register():
form = RegForm()
if form.validate_on_submit():
user_name = form.username.data
password = form.password.data
repassword = form.repassword.data
content = form.content.data
filename = secure_filename(content.filename)
user_model = UsersModel(db.get_connection())
print(user_name)
exists = user_model.login_used(user_name)
if exists:
return render_template('register.html',
title='Регистрация',
form=form)
else:
if password == repassword:
if os.path.isfile(os.path.join('static', 'img', filename)):
while os.path.isfile(
os.path.join('static', 'img', filename)):
filename = filename.split('.')
filename = '.'.join([filename[0] + 'A', filename[-1]])
content.save(os.path.join('static', 'img', filename))
user_model.insert(user_name, password, filename)
exists = user_model.exists(user_name, password)
session['username'] = user_name
session['user_id'] = exists[1][0]
session['content'] = exists[1][-1]
else:
return render_template('register.html',
title='Регистрация',
form=form)
return redirect("/index")
return render_template('register.html', title='Регистрация', form=form)
def login():
form = LoginForm()
if form.validate_on_submit():
user_name = form.username.data
password = form.password.data
user_model = UsersModel(db.get_connection())
exists = user_model.exists(user_name, password)
if (exists[0]):
session['username'] = user_name
session['user_id'] = exists[1]
return redirect("/index")
return render_template('login.html', title='Авторизация', form=form)
def login():
if request.method == 'GET':
return render_template(
'login.html',
title='Please fill in this form to sign in an account:')
elif request.method == 'POST':
um = UsersModel(db.get_connection())
um.init_table()
print(um.get_all())
if um.exists(request.form['email'], request.form['pswd']):
username = um.get_username(request.form['email'])
session['username'] = username
return redirect('/main')
else:
return render_template('login.html',
title='Wrong email or password')
def register():
if 'username' in session:
return redirect('/')
form = RegisterModel()
if form.validate_on_submit():
user_name = form.user_name.data
password = form.password.data
user = UsersModel(db.get_connection())
flag = user.is_username_busy(user_name)
if flag and user_name not in admins:
user.insert(user_name, password)
session['username'] = user_name
exists = user.exists(user_name, password)
session['user_id'] = exists[1]
return redirect("/")
else:
return render_template('register.html', form=form, error=1)
return render_template('register.html', form=form)
def login(error=None):
form = LoginForm()
if form.validate_on_submit():
user_name = form.username.data
password = form.password.data
user_model = UsersModel(db.get_connection())
exists = user_model.exists(user_name, password)
if exists[0]:
session['username'] = user_name
session['user_id'] = exists[1]
else:
return redirect('/login/notexist')
return redirect("/")
return render_template('login.html',
title='Login',
text="Авторизоваться",
form=form,
error=error)
def login(error=None
): # Now there are User1, User2, User3 with the same passwords
form = LoginForm()
if form.validate_on_submit():
user_name = form.username.data
password = form.password.data
user_model = UsersModel(db.get_connection())
exists = user_model.exists(
user_name,
str(md5(bytes(password,
encoding='utf-8')).hexdigest())) # encoding
if exists[0]:
session['username'] = user_name
session['user_id'] = exists[1]
else:
return redirect('/login/notexist')
return redirect("/index")
return render_template('login.html', title='Login', form=form, error=error)
def login():
global flag_perm
form = LoginForm()
if form.validate_on_submit():
user_name = form.username.data
password = form.password.data
perm = form.remember_me.data
user_model = UsersModel(db.get_connection())
exists = user_model.exists(user_name, password)
if exists[0]:
session['username'] = user_name
session['user_id'] = exists[1]
if perm:
session.permanent = True
flag_perm = True
else:
session.permanent = False
flag_perm = True
return redirect("/index")
else:
return render_template('login.html', form=form, error=1)
return render_template('login.html', title='Авторизация', form=form)
def login(error=None):
form = LoginForm()
if form.validate_on_submit(
): # checking if a login form fields are all valid
user_name = form.username.data
password = form.password.data
user_model = UsersModel(db.get_connection())
exists = user_model.exists(
user_name, str(md5(bytes(password, encoding='utf-8')).hexdigest())
) # checking if a user with login user_name and password hash md5(password)
if exists[0]:
session['username'] = user_name
session['user_id'] = exists[1]
else:
return redirect(
'/login/notexist'
) # if password and login are invalid redirect to /login/
return redirect("/index")
return render_template('login.html',
title='Авторизация',
form=form,
error=error)
