mirror of
https://github.com/Llloooggg/Jarvis.git
synced 2026-03-06 03:56:23 +03:00
133 lines
4.2 KiB
Python
133 lines
4.2 KiB
Python
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_routing import app, db
|
|
|
|
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 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']
|
|
triggerName = request.form['TriggerID']
|
|
triggerArgs = request.form['TriggerArgs']
|
|
actionName = request.form['ActionID']
|
|
actionArgs = request.form['ActionArgs']
|
|
db_routing.add_scenario(current_user.get_id(), newScenarioName, triggerName, triggerArgs, actionName,
|
|
actionArgs)
|
|
return redirect(url_for('workshop'))
|
|
|
|
if 'TGUsername' in request.form:
|
|
new_tg_username = request.form['TGUsername']
|
|
db_routing.tg_username_update(current_user.get_id(), new_tg_username)
|
|
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_username = User.tg_username
|
|
else:
|
|
tg_username = None
|
|
return render_template('workshop.html', triggers_list=triggers_list, actions_list=actions_list,
|
|
user_scripts_list=user_scripts_list, tg_username=tg_username)
|
|
|
|
|
|
@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()
|
|
app.run()
|