diff --git a/data.db b/data.db new file mode 100644 index 0000000..62cac5a Binary files /dev/null and b/data.db differ diff --git a/data.db.sqbpro b/data.db.sqbpro new file mode 100644 index 0000000..d89ad18 --- /dev/null +++ b/data.db.sqbpro @@ -0,0 +1 @@ +
diff --git a/mysqlite.py b/mysqlite.py new file mode 100644 index 0000000..c98f408 --- /dev/null +++ b/mysqlite.py @@ -0,0 +1,13 @@ +import sqlite3 + + +class SQLite(): + def __init__(self, file='db.sqlite3'): + self.file=file + def __enter__(self): + self.conn = sqlite3.connect(self.file) + self.conn.row_factory = sqlite3.Row + return self.conn.cursor() + def __exit__(self, type, value, traceback): + self.conn.commit() + self.conn.close() \ No newline at end of file diff --git a/webface.py b/webface.py index 40c38c0..ff24b1e 100644 --- a/webface.py +++ b/webface.py @@ -1,7 +1,7 @@ from decimal import DivisionByZero from flask import Flask, render_template, request, redirect, url_for, session, flash import functools - +from mysqlite import SQLite # from werkzeug.security import generate_password_hash, check_password_hash app = Flask(__name__) @@ -12,6 +12,8 @@ slova = ("Super", "Perfekt", "Úža", "Flask") + + def prihlasit(function): @functools.wraps(function) def wrapper(*args, **kwargs): @@ -78,7 +80,13 @@ def login_post(): jmeno = request.form.get('jmeno') heslo = request.form.get('heslo') page = request.args.get("page") - if jmeno == "marek" and heslo == "lokomotiva": + + with SQLite("data.db") as cur: + cur.execute("SELECT passwd FROM user WHERE login = ? ", [jmeno] ) + ans = cur.fetchall() + + + if ans and ans[0][0]== heslo: flash("jsi přihlašen", "message") session["uzivatel"] = jmeno if page: