Restructure

1 files changed, 21 insertions, 0 deletions

diff --git a/kosten/app/login.py b/kosten/app/login.py
new file mode 100644
index 0000000..850cc8a
--- /dev/null
+++ b/kosten/app/login.py
@@ -0,0 +1,21 @@
+from flask_login import LoginManager, UserMixin
+from passlib.apps import custom_app_context as pwd_context
+from . import app, model
+
+# just for exporting
+from flask_login import login_user, logout_user, login_required, current_user
+
+login_manager = LoginManager()
+login_manager.init_app(app)
+login_manager.login_message = "Bitte einloggen!"
+
+class User (model.User, UserMixin):
+    def check_password(self, pwd):
+        return pwd_context.verify(pwd, self.pwd)
+
+    def set_password(self, pwd):
+        self.pwd = pwd_context.encrypt(pwd)
+
+@login_manager.user_loader
+def load_user(id):
+    return User.get(id)