From 2257e0806365775deb429abef0efca6a861c82e8 Mon Sep 17 00:00:00 2001
From: tototomate123 <rushwars201966@gmail.com>
Date: Tue, 24 Feb 2026 17:45:43 +0100
Subject: [PATCH] switch to dataclass, cleanup

---
 app/__init__.py    | 3 ++-
 app/models/user.py | 7 ++++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/app/__init__.py b/app/__init__.py
index 1d44a5e..ea2ebe3 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -1,3 +1,4 @@
+from typing import Optional
 from flask import Flask
 from flask_socketio import SocketIO
 from flask_login import LoginManager
@@ -8,7 +9,7 @@ sIO = SocketIO()
 login_manager = LoginManager()
 
 @login_manager.user_loader
-def load_user(user_id):
+def load_user(user_id: int | str) -> Optional["User"]:
     return User.get(user_id)
 
 def create_app():
diff --git a/app/models/user.py b/app/models/user.py
index 170c538..432a917 100644
--- a/app/models/user.py
+++ b/app/models/user.py
@@ -1,11 +1,12 @@
+from dataclasses import dataclass
 from typing import Optional
 from app.db import get_db
 from flask_login import UserMixin
 
+@dataclass
 class User(UserMixin):
-    def __init__(self, id: int, username: str ):
-        self.id: int = id
-        self.username: str = username
+    id: int
+    username: str
 
     @staticmethod
     def get(user_id: int | str) -> Optional["User"]: