first commit

mdrsclient/models/user.py

@@ -0,0 +1,50 @@
+import time
+
+import jwt
+from pydantic import parse_obj_as
+from pydantic.dataclasses import dataclass
+
+
+@dataclass(frozen=True)
+class DecodedJWT:
+    token_type: str
+    exp: int
+    iat: int
+    jti: str
+    user_id: int
+
+
+@dataclass(frozen=True)
+class Token:
+    access: str
+    refresh: str
+
+    @property
+    def user_id(self) -> int:
+        access_decoded = self.__decode(self.access)
+        return access_decoded.user_id
+
+    @property
+    def is_expired(self) -> bool:
+        now = int(time.time()) + 10
+        refresh_decoded = self.__decode(self.refresh)
+        return now > refresh_decoded.exp
+
+    @property
+    def is_refresh_required(self) -> bool:
+        now = int(time.time()) + 10
+        access_decoded = self.__decode(self.access)
+        refresh_decoded = self.__decode(self.refresh)
+        return now > access_decoded.exp and now < refresh_decoded.exp
+
+    def __decode(self, token: str) -> DecodedJWT:
+        data = jwt.decode(token, options={"verify_signature": False})
+        return parse_obj_as(DecodedJWT, data)
+
+
+@dataclass(frozen=True)
+class User:
+    id: int
+    username: str
+    laboratory_id: int
+    laboratory: str