🔥 sessionkey.py Model Update

Major:

📦 sessionkey.py -> wrote whole modle with valiators

Minor:

🔥 apikey.py -> removed testing prints

Author: pheeef

models/apikey.py

@@ -1,10 +1,10 @@
 from typing import Optional
-from pydantic import BaseModel, Field, root_validator
+from pydantic import BaseModel, root_validator
 from assets.database import openDBConnection
 
 
 class apikey(BaseModel):
-    ID: Optional[int] = Field(None, description="")
+    ID: Optional[int]
     NAME: Optional[str]
     APIKEY: str
     PERMISSION: Optional[int]

models/sessionkey.py

@@ -0,0 +1,27 @@
+from typing import Optional
+from pydantic import BaseModel, root_validator, ValidationError
+from assets.database import openDBConnection
+
+
+class SessionKey(BaseModel):
+    key: str
+    valid: Optional[bool] = False
+
+    @root_validator
+    def check_session_key(cls, values):
+        db = openDBConnection()
+        cursor = db.cursor()
+
+        cursor.execute("SELECT SESSION_KEY FROM USERDATA WHERE SESSION_KEY = %s", (values['key'],))
+        data = cursor.fetchone()
+
+        if data[0] is not None:
+            values['valid'] = True
+            return values
+
+        raise ValidationError('SessionKey is not valid')
+
+
+rakic = SessionKey(key="d2587b14f80e1f07c1335184e89e7a6c2e6200626ae05d6b99c4d2354d789edd")
+
+print(rakic.valid)