Дописана проверка на существующего пользователя при регистрации

This commit is contained in:
2020-03-15 12:03:26 +03:00
parent f12671df12
commit 2cddf6fa10
2 changed files with 36 additions and 14 deletions

View File

@@ -1,7 +1,10 @@
from flask import Flask, render_template, flash, redirect, url_for, session, logging, request
from flask_sqlalchemy import SQLAlchemy
import os
import hashlib
import sqlite3
app = Flask(__name__, static_folder="static", template_folder="templates")
app = Flask(__name__, static_folder='static', template_folder='templates')
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///data.db'
db = SQLAlchemy(app)
@@ -11,28 +14,47 @@ def index():
return render_template('index.html')
@app.route("/registration", methods=["GET", "POST"])
@app.route('/registraion', methods=['GET', 'POST'])
def register():
if request.method == "POST":
if request.method == 'POST':
UserLogin = request.form['UserLogin']
UserPass = request.form['UserPass']
if not find_user_copy(UserLogin):
db.session.add(UserLogin, passw_hash(UserPass))
db.session.commit()
return render_template('registration.html')
#db.session.add(UserLogin, UserPass)
#db.session.commit()
#return redirect(url_for(""))
@app.route('/content', methods=['GET'])
def content():
return render_template('content.html')
print(UserLogin, ' ', UserPass)
# return страница с контентом
return render_template("registration.html")
@app.errorhandler(404)
def not_found(error):
return render_template('404.html'), 404
def find_user_copy(user_login):
con = sqlite3.connect('data.db')
with con:
cur = con.cursor()
exist = cur.execute('SELECT EXISTS ( SELECT UserLogin FROM Users Where UserLogin = ' + user_login + ' LIMIT 1')
return exist
def passw_hash(user_passw):
salt = os.urandom(32)
key = hashlib.pbkdf2_hmac('sha256', user_passw.encode('utf-8'), salt, 100000)
storage = salt + key
# salt_from_storage = storage[:32] # 32 длина соли
# key_from_storage = storage[32:]
return storage
# @app.route('/login', methods=['GET'])
# def login():
# return render_template('login.html')
if __name__ == "__main__":
if __name__ == '__main__':
app.run()