hodne veci se zmenilo

templates/base.html

@@ -25,7 +25,9 @@
 {% endif %}
 </p>
 {% for category, message in get_flashed_messages(with_categories = true)  %}
-<p class="flash" {{category}}>{{message}}</p>
+<p class="flash {{category}}">
+<button class ="hide">X</button>
+{{message}}</p>
 {% endfor %}
 
 
@@ -78,6 +80,20 @@ <h1>Flask</h1>
 
 
 
+<script>
+function hide(){
+  this.parentElement.style.display = "none";
 
+}
+
+var tlacitka = document.getElementsByClassName("hide");
+for (var i = 0; i<tlacitka.length; i++ ){
+  tlacitka[i].onclick =hide;
+}
+
+
+
+
+</script>
 </body>
 </html>

webface.py

@@ -35,6 +35,9 @@ def info():
 
 @app.route("/abc/")
 def abc():
+    if 'uzivatel' not in session:
+        flash("Nejsi prihlasen blazne", "error")
+        return redirect(url_for("login", page = request.full_path))
     return render_template("abc.html", slova=slova)
 
 
@@ -44,7 +47,7 @@ def abc():
 def malina():
     if 'uzivatel' not in session:
         flash("Nejsi prihlasen blazne", "error")
-        return redirect(url_for("login"))
+        return redirect(url_for("login", page = request.full_path))
 
     hmotnost = request.args.get('hmotnost')
     vyska = request.args.get('vyska')
@@ -74,11 +77,16 @@ def login():
 def login_post():
     jmeno = request.form.get('jmeno')
     heslo = request.form.get('heslo')
+    page = request.args.get("page")
     if jmeno == "marek" and heslo == "lokomotiva":
+        flash("jsi přihlašen", "message")
         session["uzivatel"] = jmeno
-
-
-
+        if page:
+            return redirect(page)   
+    else:
+        flash("nespravne udaje", "error")
+    if page:
+        return redirect(url_for("login", page= page)) 
     return redirect(url_for('login'))