Llloooggg/Jarvis
1
0
Fork 0
mirror of https://github.com/Llloooggg/Jarvis.git synced 2026-03-06 03:56:23 +03:00
Code Issues 21 Packages Projects Releases Wiki Activity

Переписана система сессий и логина

Browse Source
This commit is contained in:
Denis Podshivalov
2020-03-15 20:12:40 +03:00
parent 14865444da
commit 1f4e0c11f0
4 changed files with 177 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
__init__.py
View File

@@ -1,16 +1,17 @@
from flask import Flask, render_template, request
from flask import render_template, request, redirect
import db_routing
from flask_httpauth import HTTPBasicAuth
from flask_login import LoginManager, current_user, login_user, login_required
from db_routing import app, db
import os
import hashlib
login_manager = LoginManager(app)
if not os.path.exists('./data.db'):
db_routing.db.create_all()
app = Flask('Jarvis', static_folder='static', template_folder='templates')
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///data.db'
auth = HTTPBasicAuth()
@login_manager.user_loader
def load_user(user):
user_id = user.UserID
return user_id
@app.route('/', methods=['GET'])
@@ -21,26 +22,25 @@ def index():
@app.route('/registration', methods=['GET', 'POST'])
def register():
if request.method == 'POST':
userLogin = request.form['UserLogin']
userPassw = request.form['UserPassw']
userLogin = request.form['RegUserLogin']
userPassw = request.form['RegUserPassw']
if db_routing.add_user(userLogin, passw_hash(userPassw)):
return render_template('content.html')
return redirect('/content')
return render_template('registration.html')
@auth.verify_password
def verify_password(user_login, user_passw):
User = db_routing.find_user(user_login)
if User:
userSalt = User.UserPassw[:32]
if passw_hash(user_passw, userSalt) == User.UserPassw:
return True
else:
return False
@app.route('/login', methods=['POST'])
def login():
userLogin = request.form['LogUserLogin']
userPassw = request.form['LogUserPassw']
user = verify_password(userLogin, userPassw)
if user:
login_user(user)
return redirect('/content')
@app.route('/content', methods=['GET'])
@auth.login_required
@login_required
def content():
return render_template('content.html')
@@ -58,5 +58,18 @@ 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)
if User:
userSalt = User.UserPassw[:32]
if passw_hash(user_passw, userSalt) == User.UserPassw:
return User
else:
print('Неверный пароль')
return False
if __name__ == '__main__':
if not os.path.exists('./data.db'):
db.create_all()
app.run()

29
db_routing.py
View File

@@ -1,23 +1,46 @@
from flask_sqlalchemy import SQLAlchemy
from __init__ import app
from flask import Flask
app = Flask('Jarvis', static_folder='static', template_folder='templates')
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///data.db'
db = SQLAlchemy(app)
class User(db.Model):
__tablename__ = 'users'
__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)
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))
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))
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)
def add_user(user_name, passw_hash):
if not find_user(user_name):
new_user = User(UserName=user_name, UserPassw=passw_hash)
db.session.add(new_user)
db.session.commit()
return True
return new_user
else:
print('Логин занят')
return False

2
requirements.txt
View File

@@ -3,7 +3,7 @@ Flask==1.1.1
Flask-HTTPAuth==3.3.0
Flask-SQLAlchemy==2.4.1
itsdangerous==1.1.0
Jinja2==2.11.1
Jinja2==3.0.0a1
MarkupSafe==1.1.1
SQLAlchemy==1.3.15
Werkzeug==1.0.0

234
templates/index.html
View File

@@ -1,118 +1,118 @@
<!DOCTYPE html>
<html lang="ru-RU">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta http-equiv="X-UA-Compatible" content="ie=edge">
<title>jarvis</title>
<link rel="stylesheet" type="text/css" href="{{ url_for('static', filename = './css/main.css') }}">
</head>
<body>
<div class="wrapper">
<!-- header-- -->
<header class="header">
<div class="container clearfix">
<div class="header_left clearfix">
<a href="/" class="logo-decor">
<img class="logo_icon" src="{{ url_for('static', filename='./img/815a836e51c6f20226cb07df7f32895c.png') }}" alt="">
<p class="logo-text">J.a.r.v.i.s </p>
</a>
</div>
<div class="header_right">
<div class="contacts">
<div class="drop">
<a href="" class="contacts__schedule">Войти</a>
<div class="dropdown">
<div class="drop-block">
<div class="triangle"></div>
<label class="form__cell-header ">
<input type="text" class="form__input-text-mod" placeholder="Логин" >
</label>
<label class="form__cell-header">
<input type="password" class="form__input-text-mod" placeholder="Пароль">
</label>
<a href="" class="input">
<input type="button" href="/content.html" name="scriptbutton" class="header_btn" value="Войти" >
</a>
</div>
</div>
</div>
</div>
</div>
</div>
</header>
<section class="hero">
<div class="container">
<div class="hero_con clearfix">
<div class="rotateblock_orenge">
<div class="ava_logo">
<div class="rotateblock">
<img src="{{ url_for('static', filename='./img/ava.png') }}" alt="" class="ava">
</div>
</div>
</div>
<div class="content_form">
<div class="animate_text">
<h class="h_content">Jarvis</h>
</div>
<div class="content_form-text">
<p class="text_content">Ваш индивидуальный помощник на каждый день</p>
</div>
<div class="input-button">
<a href="/registration" class="input">
<input type="button" href="/registration" name="scriptbutton" class="welcome__btn" value="Регистрация" >
</a>
</div>
</div>
</div>
</div>
</section>
</div>
<!-- заглушка -->
<footer class="footer clearfix">
<div class="container">
<div class="footer-q clearfix">
<p class="footer-full-text">Сделано с любовью и старанием на курсе ВГУ "Тестирование ПО" <b class="footer-text_name"></b> </p>
</div>
<div class="footer-q clearfix">
<a href="/" class="logo-text">j.a.r.v.i.s</a>
</div>
<div class="footer-q clearfix">
<p class="footer-messeger_text">Наши социальные сети</p>
<ul class="footer-messeger">
<li class="footer-messeger__item ">
<a href="" class="footer-messeger__link">
<img src="{{ url_for('static', filename='./img/vk.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
<li class="footer-messeger__item">
<a href="" class="footer-messeger__link ">
<img src="{{ url_for('static', filename='./img/fb.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
<li class="footer-messeger__item">
<a href="" class="footer-messeger__link">
<img src="{{ url_for('static', filename='./img/google.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
<li class="footer-messeger__item">
<a href="" class="footer-messeger__link">
<img src="{{ url_for('static', filename='./img/twitter.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
<li class="footer-messeger__item">
<a href="" class="footer-messeger__link">
<img src="{{ url_for('static', filename='./img/mail.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
</ul>
</div>
</div>
</footer>
</body>
<!DOCTYPE html>
<html lang="ru-RU">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta http-equiv="X-UA-Compatible" content="ie=edge">
<title>jarvis</title>
<link rel="stylesheet" type="text/css" href="{{ url_for('static', filename = './css/main.css') }}">
</head>
<body>
<div class="wrapper">
<!-- header-- -->
<header class="header">
<div class="container clearfix">
<div class="header_left clearfix">
<a class="logo-decor">
<img class="logo_icon" src="{{ url_for('static', filename='./img/815a836e51c6f20226cb07df7f32895c.png') }}" alt="">
<p class="logo-text">J.a.r.v.i.s </p>
</a>
</div>
<div class="header_right">
<div class="contacts">
<div class="drop">
<a href="" class="contacts__schedule">Войти</a>
<div class="dropdown">
<div class="drop-block">
<div class="triangle"></div>
<label class="form__cell-header ">
<input type="text" class="form__input-text-mod" placeholder="Логин" >
</label>
<label class="form__cell-header">
<input type="password" class="form__input-text-mod" placeholder="Пароль">
</label>
<a href="/login" class="input">
<input type="button" href="/login" name="scriptbutton" class="header_btn" value="Войти" >
</a>
</div>
</div>
</div>
</div>
</div>
</div>
</header>
<section class="hero">
<div class="container">
<div class="hero_con clearfix">
<div class="rotateblock_orenge">
<div class="ava_logo">
<div class="rotateblock">
<img src="{{ url_for('static', filename='./img/ava.png') }}" alt="" class="ava">
</div>
</div>
</div>
<div class="content_form">
<div class="animate_text">
<h class="h_content">Jarvis</h>
</div>
<div class="content_form-text">
<p class="text_content">Ваш индивидуальный помощник на каждый день</p>
</div>
<div class="input-button">
<a href="/registration" class="input">
<input type="button" href="/registration" name="scriptbutton" class="welcome__btn" value="Регистрация" >
</a>
</div>
</div>
</div>
</div>
</section>
</div>
<!-- заглушка -->
<footer class="footer clearfix">
<div class="container">
<div class="footer-q clearfix">
<p class="footer-full-text">Сделано с любовью и старанием на курсе ВГУ "Тестирование ПО" <b class="footer-text_name"></b> </p>
</div>
<div class="footer-q clearfix">
<a href="/" class="logo-text">j.a.r.v.i.s</a>
</div>
<div class="footer-q clearfix">
<p class="footer-messeger_text">Наши социальные сети</p>
<ul class="footer-messeger">
<li class="footer-messeger__item ">
<a href="" class="footer-messeger__link">
<img src="{{ url_for('static', filename='./img/vk.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
<li class="footer-messeger__item">
<a href="" class="footer-messeger__link ">
<img src="{{ url_for('static', filename='./img/fb.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
<li class="footer-messeger__item">
<a href="" class="footer-messeger__link">
<img src="{{ url_for('static', filename='./img/google.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
<li class="footer-messeger__item">
<a href="" class="footer-messeger__link">
<img src="{{ url_for('static', filename='./img/twitter.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
<li class="footer-messeger__item">
<a href="" class="footer-messeger__link">
<img src="{{ url_for('static', filename='./img/mail.png') }}" class="footer-messeger__icon" alt="">
</a>
</li>
</ul>
</div>
</div>
</footer>
</body>
</html>