first commit

swingmusic/models/user.py

@@ -0,0 +1,36 @@
+import json
+from dataclasses import asdict, dataclass, field
+
+
+@dataclass(slots=True)
+class User:
+    id: int
+    image: str
+    password: str
+    username: str
+    roles: list[str]
+    extra: dict[str, str] = field(default_factory=dict)
+    password_change_required: bool = False
+
+    # NOTE: roles: ['admin', 'user', 'curator']
+    roles: list[str] = field(default_factory=lambda: ["user"])
+
+    def todict(self):
+        this_dict = asdict(self)
+        del this_dict["password"]
+
+        if type(this_dict["roles"]) is str:
+            # INFO: this is an attempt to fix string roles!
+            try:
+                this_dict["roles"] = json.loads(this_dict["roles"])
+            except json.JSONDecodeError:
+                this_dict["roles"] = []
+
+        return this_dict
+
+    def todict_simplified(self):
+        return {
+            "id": self.id,
+            "username": self.username,
+            "firstname": self.extra["firstname"] if self.extra else "",
+        }