mirror of
https://github.com/Llloooggg/Jarvis.git
synced 2026-03-06 03:56:23 +03:00
Рефракторинг
This commit is contained in:
140
__init__.py
140
__init__.py
@@ -1,140 +0,0 @@
|
||||
import hashlib
|
||||
import os
|
||||
from re import match
|
||||
|
||||
from flask import render_template, request, redirect, url_for
|
||||
from flask_login import LoginManager, login_user, login_required, logout_user, current_user
|
||||
|
||||
import db_routing
|
||||
from db_init import db_fill
|
||||
from db_routing import app, db
|
||||
from scenario_manager import Executor
|
||||
|
||||
login_manager = LoginManager(app)
|
||||
|
||||
|
||||
@login_manager.user_loader
|
||||
def load_user(user_id):
|
||||
return db_routing.get_user(id=user_id)
|
||||
|
||||
|
||||
@app.route('/', methods=['GET'])
|
||||
def index():
|
||||
return render_template('index.html')
|
||||
|
||||
|
||||
@app.route('/registration', methods=['GET', 'POST'])
|
||||
def registration():
|
||||
if current_user.is_authenticated:
|
||||
return redirect(url_for('universal_error'))
|
||||
if request.method == 'POST':
|
||||
userName = request.form['RegUserLogin']
|
||||
userPassw = request.form['RegUserPassw']
|
||||
if string_check(userName) and string_check(userPassw):
|
||||
if db_routing.add_user(userName, passw_hash(userPassw)):
|
||||
login_user(db_routing.get_user(username=userName))
|
||||
return redirect(url_for('workshop'))
|
||||
return render_template('registration.html')
|
||||
|
||||
|
||||
@app.route('/login', methods=['POST'])
|
||||
def login():
|
||||
userName = request.form['LogUserLogin']
|
||||
userPassw = request.form['LogUserPassw']
|
||||
if string_check(userName) and string_check(userPassw):
|
||||
user = verify_password(userName, userPassw)
|
||||
if user:
|
||||
login_user(user)
|
||||
return redirect(url_for('workshop'))
|
||||
else:
|
||||
return redirect(url_for('registration'))
|
||||
|
||||
else:
|
||||
return redirect(url_for('registration'))
|
||||
|
||||
|
||||
@app.route('/logout')
|
||||
def logout():
|
||||
logout_user()
|
||||
return redirect(url_for('index'))
|
||||
|
||||
|
||||
@app.route('/workshop', methods=['GET', 'POST'])
|
||||
@login_required
|
||||
def workshop():
|
||||
if request.method == 'POST':
|
||||
if 'NewScenarioName' in request.form:
|
||||
newScenarioName = request.form['NewScenarioName']
|
||||
triggerID = request.form['TriggerID']
|
||||
triggerArgs = request.form['TriggerArgs']
|
||||
actionID = request.form['ActionID']
|
||||
actionArgs = request.form['ActionArgs']
|
||||
new_scenario = db_routing.add_scenario(current_user.get_id(), newScenarioName, triggerID, triggerArgs,
|
||||
actionID,
|
||||
actionArgs)
|
||||
activeScenario = Executor(new_scenario, current_user.get_tg_id())
|
||||
activeScenario.start()
|
||||
return redirect(url_for('workshop'))
|
||||
|
||||
if 'TGID' in request.form:
|
||||
new_tg_id = request.form['TGID']
|
||||
db_routing.tg_id_update(current_user.get_id(), new_tg_id)
|
||||
return redirect(url_for('workshop'))
|
||||
|
||||
if 'ScenarioID' in request.form:
|
||||
scnarioID = request.form['ScenarioID']
|
||||
db_routing.delete_scenario(scnarioID)
|
||||
return redirect(url_for('workshop'))
|
||||
|
||||
triggers_list = db_routing.get_trigers()
|
||||
actions_list = db_routing.get_actions()
|
||||
user_scripts_list = db_routing.get_user_scripts(current_user.get_id())
|
||||
User = db_routing.get_user(id=current_user.get_id())
|
||||
if User:
|
||||
tg_id = User.tg_id
|
||||
else:
|
||||
tg_id = None
|
||||
return render_template('workshop.html', triggers_list=triggers_list, actions_list=actions_list,
|
||||
user_scripts_list=user_scripts_list, tg_id=tg_id)
|
||||
|
||||
|
||||
@app.errorhandler(Exception)
|
||||
def universal_error(error):
|
||||
return render_template('error.html'), 404
|
||||
|
||||
|
||||
def string_check(string):
|
||||
if 2 < len(string) < 7:
|
||||
if match('^[0-9A-Za-z]*$', string) and not ('\\' in string):
|
||||
return True
|
||||
else:
|
||||
print(
|
||||
'Некорректный ввод! Строка должно включать только английские буквы или цифры. Содержать не менее 3 и не '
|
||||
'более 6 символов')
|
||||
return False
|
||||
|
||||
|
||||
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
|
||||
|
||||
|
||||
def verify_password(username, password):
|
||||
User = db_routing.get_user(username=username)
|
||||
if User:
|
||||
userSalt = User.password[:32]
|
||||
if passw_hash(password, userSalt) == User.password:
|
||||
return User
|
||||
else:
|
||||
print('Неверный пароль')
|
||||
return False
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if not os.path.exists('./data.db'):
|
||||
db.create_all()
|
||||
db_fill()
|
||||
app.run()
|
||||
|
||||
Reference in New Issue
Block a user