mirror of
https://github.com/Llloooggg/Jarvis.git
synced 2026-03-06 03:56:23 +03:00
Модель сессии и логина переписана и работает
This commit is contained in:
3
.gitignore
vendored
3
.gitignore
vendored
@@ -1,3 +1,4 @@
|
||||
./idea
|
||||
./venv
|
||||
./vscode
|
||||
./vscode
|
||||
./data.db
|
||||
27
__init__.py
27
__init__.py
@@ -9,9 +9,9 @@ login_manager = LoginManager(app)
|
||||
|
||||
|
||||
@login_manager.user_loader
|
||||
def load_user(user):
|
||||
user_id = user.UserID
|
||||
return user_id
|
||||
def load_user(user_id):
|
||||
userID = db_routing.find_user(id=user_id).UserID
|
||||
return userID
|
||||
|
||||
|
||||
@app.route('/', methods=['GET'])
|
||||
@@ -22,21 +22,22 @@ def index():
|
||||
@app.route('/registration', methods=['GET', 'POST'])
|
||||
def register():
|
||||
if request.method == 'POST':
|
||||
userLogin = request.form['RegUserLogin']
|
||||
userName = request.form['RegUserLogin']
|
||||
userPassw = request.form['RegUserPassw']
|
||||
if db_routing.add_user(userLogin, passw_hash(userPassw)):
|
||||
return redirect('/content')
|
||||
if db_routing.add_user(userName, passw_hash(userPassw)):
|
||||
login_user(db_routing.find_user(username=userName))
|
||||
return redirect('content.html')
|
||||
return render_template('registration.html')
|
||||
|
||||
|
||||
@app.route('/login', methods=['POST'])
|
||||
def login():
|
||||
userLogin = request.form['LogUserLogin']
|
||||
userName = request.form['LogUserLogin']
|
||||
userPassw = request.form['LogUserPassw']
|
||||
user = verify_password(userLogin, userPassw)
|
||||
user = verify_password(userName, userPassw)
|
||||
if user:
|
||||
login_user(user)
|
||||
return redirect('/content')
|
||||
return render_template('content.html')
|
||||
|
||||
|
||||
@app.route('/content', methods=['GET'])
|
||||
@@ -58,11 +59,11 @@ def passw_hash(user_passw, salt=os.urandom(32)):
|
||||
return storage
|
||||
|
||||
|
||||
def verify_password(user_login, user_passw):
|
||||
User = db_routing.find_user(user_login)
|
||||
def verify_password(username, password):
|
||||
User = db_routing.find_user(username=username)
|
||||
if User:
|
||||
userSalt = User.UserPassw[:32]
|
||||
if passw_hash(user_passw, userSalt) == User.UserPassw:
|
||||
userSalt = User.password[:32]
|
||||
if passw_hash(password, userSalt) == User.password:
|
||||
return User
|
||||
else:
|
||||
print('Неверный пароль')
|
||||
|
||||
@@ -1,43 +1,48 @@
|
||||
from flask_sqlalchemy import SQLAlchemy
|
||||
from flask import Flask
|
||||
|
||||
from flask_login import UserMixin
|
||||
|
||||
app = Flask('Jarvis', static_folder='static', template_folder='templates')
|
||||
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///data.db'
|
||||
app.config['SECRET_KEY'] = 'Radius'
|
||||
db = SQLAlchemy(app)
|
||||
|
||||
|
||||
class User(db.Model):
|
||||
class User(UserMixin, db.Model):
|
||||
__tablename__ = 'Users'
|
||||
UserID = db.Column(db.Integer, primary_key=True)
|
||||
UserName = db.Column(db.String(80), unique=True, nullable=False)
|
||||
UserPassw = db.Column(db.String(120), nullable=False)
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
username = db.Column(db.String(80), unique=True, nullable=False)
|
||||
password = db.Column(db.String(120), nullable=False)
|
||||
|
||||
def __init__(self, username, password):
|
||||
self.username = username
|
||||
self.password = password
|
||||
|
||||
|
||||
class Trigger(db.Model):
|
||||
__tablename__ = 'Triggers'
|
||||
TriggerID = db.Column(db.Integer, primary_key=True)
|
||||
TriggerName = db.Column(db.String(80), unique=True, nullable=False)
|
||||
TriggerArgs = db.Column(db.String(200))
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
triggername = db.Column(db.String(80), unique=True, nullable=False)
|
||||
triggerargs = db.Column(db.String(200))
|
||||
|
||||
|
||||
class Action(db.Model):
|
||||
__tablename__ = 'Actions'
|
||||
ActionID = db.Column(db.Integer, primary_key=True)
|
||||
ActionName = db.Column(db.String(80), unique=True, nullable=False)
|
||||
ActionArgs = db.Column(db.String(200))
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
actionname = db.Column(db.String(80), unique=True, nullable=False)
|
||||
actionargs = db.Column(db.String(200))
|
||||
|
||||
|
||||
class Scenario(db.Model):
|
||||
__tablename__ = 'Scenarios'
|
||||
ScenarioID = db.Column(db.Integer, primary_key=True)
|
||||
ScenarioTrigger = db.Column(db.Integer, nullable=False)
|
||||
ScenarioAction = db.Column(db.Integer, nullable=False)
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
scenariotrigger = db.Column(db.Integer, nullable=False)
|
||||
scenarioaction = db.Column(db.Integer, nullable=False)
|
||||
|
||||
|
||||
def add_user(user_name, passw_hash):
|
||||
if not find_user(user_name):
|
||||
new_user = User(UserName=user_name, UserPassw=passw_hash)
|
||||
new_user = User(username=user_name, password=passw_hash)
|
||||
db.session.add(new_user)
|
||||
db.session.commit()
|
||||
return new_user
|
||||
@@ -46,5 +51,8 @@ def add_user(user_name, passw_hash):
|
||||
return False
|
||||
|
||||
|
||||
def find_user(user_name):
|
||||
return User.query.filter_by(UserName=user_name).first()
|
||||
def find_user(id=None, username=None):
|
||||
if id:
|
||||
return User.query.filter_by(id=id).first()
|
||||
if username:
|
||||
return User.query.filter_by(username=username).first()
|
||||
|
||||
@@ -22,22 +22,20 @@
|
||||
<div class="header_right">
|
||||
<div class="contacts">
|
||||
<div class="drop">
|
||||
<div href="" class="contacts__schedule">Войти</div>
|
||||
<div class="contacts__schedule">Войти</div>
|
||||
<div class="dropdown">
|
||||
<form action="" method="POST">
|
||||
<div class="drop-block">
|
||||
<div class="drop-block">
|
||||
<form action="/login" method="POST">
|
||||
<div class="triangle"></div>
|
||||
<label class="form__cell-header ">
|
||||
<input type="text" class="form__input-text-mod" placeholder="Логин" name= "LogUserLogin">
|
||||
</label>
|
||||
<label class="form__cell-header">
|
||||
<input type="password" class="form__input-text-mod" placeholder="Пароль" name= "LogUserLogin">
|
||||
<input type="password" class="form__input-text-mod" placeholder="Пароль" name= "LogUserPassw">
|
||||
</label>
|
||||
<a href="/login" class="input">
|
||||
<input type="button" href="/login" name="LogButton" class="header_btn" value="Войти" >
|
||||
</a>
|
||||
</div>
|
||||
</form>
|
||||
<button type="submit" value="Войти" class="header_btn" name="LogButton" >Вход</button>
|
||||
</form>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
@@ -63,7 +61,7 @@
|
||||
<p class="text_content">Ваш индивидуальный помощник на каждый день</p>
|
||||
</div>
|
||||
<div class="input-button">
|
||||
<a href="" class="input">
|
||||
<a href="/registration" class="input">
|
||||
<input type="button" href="/registration" name="scriptbutton" class="welcome__btn" value="Регистрация" >
|
||||
</a>
|
||||
</div>
|
||||
|
||||
@@ -31,7 +31,7 @@
|
||||
<input type="text" class="form__input-text-mod" placeholder="Логин" name= "LogUserLogin">
|
||||
</label>
|
||||
<label class="form__cell-header">
|
||||
<input type="password" class="form__input-text-mod" placeholder="Пароль" name= "LogUserLogin">
|
||||
<input type="password" class="form__input-text-mod" placeholder="Пароль" name= "LogUserPassw">
|
||||
</label>
|
||||
<a href="" class="input">
|
||||
<input type="button" href="/content.html" name="LogButton" class="header_btn" value="Войти" >
|
||||
@@ -88,7 +88,7 @@
|
||||
<div class="form__controls">
|
||||
<div class="form__controls-btns clearfix">
|
||||
<div class="form__controls-right">
|
||||
<button type="submit" value="Отправить" class="form__submit" name="RegButton" href="/content" >Регистрация</button>
|
||||
<button type="submit" value="Отправить" class="form__submit" name="RegButton" >Регистрация</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
Reference in New Issue
Block a user