From ae67d53c5e3dc7d95b5bc165b1d1413a1a4e15fe Mon Sep 17 00:00:00 2001
From: Stef <stbuwalda@gmail.com>
Date: Tue, 15 Apr 2025 11:26:59 +0200
Subject: [PATCH] Fixes squiggle lines

---
 app.py                     | 2 +-
 application/auth/models.py | 5 +++++
 application/dash/models.py | 4 ++++
 application/decorators.py  | 7 ++++---
 4 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/app.py b/app.py
index ddcf5de..89ab1c5 100644
--- a/app.py
+++ b/app.py
@@ -1,6 +1,6 @@
 from application import app
 from flask import redirect, url_for
-from flask_login import current_user, login_required  # type: ignore
+from flask_login import login_required  # type: ignore
 
 
 # home route
diff --git a/application/auth/models.py b/application/auth/models.py
index 73a539a..ab4ce40 100644
--- a/application/auth/models.py
+++ b/application/auth/models.py
@@ -7,3 +7,8 @@ class User(db.Model, UserMixin):
     username = db.Column(db.String(150), unique=True, nullable=False)
     password = db.Column(db.String(150), nullable=False)
     is_admin = db.Column(db.Boolean, default=False)
+
+    def __init__(self, username: str, password: str, is_admin: bool = False):
+        self.username = username
+        self.password = password
+        self.is_admin = is_admin
diff --git a/application/dash/models.py b/application/dash/models.py
index b468005..8789483 100644
--- a/application/dash/models.py
+++ b/application/dash/models.py
@@ -5,3 +5,7 @@ class Service(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     name = db.Column(db.String, nullable=False)
     url = db.Column(db.String, nullable=False)
+
+    def __init__(self, name: str, url: str):
+        self.name = name
+        self.url = url
diff --git a/application/decorators.py b/application/decorators.py
index c38d108..4112de3 100644
--- a/application/decorators.py
+++ b/application/decorators.py
@@ -1,11 +1,12 @@
-from flask_login import current_user
+from typing import Callable, Any
+from flask_login import current_user  # type: ignore
 from functools import wraps
 from flask import redirect, url_for, flash
 
 
-def admin_required(f):
+def admin_required(f: Callable[..., Any]) -> Callable[..., Any]:
     @wraps(f)
-    def decorated_function(*args, **kwargs):
+    def decorated_function(*args: ..., **kwargs: ...):
         if not current_user.is_authenticated:
             return redirect(url_for("login"))
         if not current_user.is_admin: