Нейминг

__main__.py

@@ -0,0 +1,159 @@
+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()