✨️ user.py Model

Author: pheeef

functions/hashing.py

@@ -0,0 +1,16 @@
+from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
+from passlib.context import CryptContext
+
+pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
+oauth2_sheme = OAuth2PasswordBearer(tokenUrl="token")
+
+SECRET_KEY = "09d25e094faa6ca2556c818166b7a9563b93f7099f6f0f4caa6cf63b88e8d3e7"
+ALOGRITHM = "HS256"
+ACCESS_TOKEN_EXPIRE_MINUTES = 30
+
+def verify_password(plain_password, hashed_password):
+    return pwd_context.verify(plain_password, hashed_password)
+
+
+def get_password_hash(password):
+    return pwd_context.hash(password)

functions/user.py

@@ -0,0 +1,36 @@
+from assets.database import openDBConnection
+from models.sessionkey import SessionKey
+from models.user import User
+
+
+def get_user(ID=None, EMAIL=None, SESSION_KEY=None):
+    db = openDBConnection()
+    cursor = db.cursor()
+
+    if ID is not None:
+        cursor.execute("SELECT * FROM USERDATA WHERE ID = %s", (ID,))
+        return fetch_data(cursor)
+    if EMAIL is not None:
+        cursor.execute("SELECT * FROM USERDATA WHERE ID = %s", (EMAIL,))
+        return fetch_data(cursor)
+    if SESSION_KEY is not None:
+        cursor.execute("SELECT * FROM USERDATA WHERE ID = %s", (SESSION_KEY,))
+        return fetch_data(cursor)
+    else:
+        raise ValueError('USER not Valid')
+
+
+def fetch_data(cursor):
+    data = cursor.fetchone()
+    return User(
+        ID=data[0],
+        EMAIL=data[1],
+        PASSWORD_HASH=data[2],
+        NAME=data[3],
+        LASTNAME=data[4],
+        CLASS=data[5],
+        PERMISSION_LEVEL=data[6],
+        LAST_IP=data[7],
+        ACTIVE=data[8],
+        SESSION_KEY=SessionKey(key=data[9])
+    )

main.py

@@ -1,4 +1,13 @@
 from fastapi import FastAPI
 
+import functions
+import models
+import assets
+
 app = FastAPI()
 
+
+
+
+
+

models/apikey.py

@@ -3,7 +3,7 @@
 from assets.database import openDBConnection
 
 
-class apikey(BaseModel):
+class ApiKey(BaseModel):
     ID: Optional[int]
     NAME: Optional[str]
     APIKEY: str

models/sessionkey.py

@@ -19,4 +19,4 @@ def check_session_key(cls, values):
             values['valid'] = True
             return values
 
-        raise ValidationError('SessionKey is not valid')
+        raise ValueError('SessionKey is not valid')

models/user.py

@@ -0,0 +1,40 @@
+from typing import Optional
+from pydantic import BaseModel, validator, ValidationError, IPvAnyAddress
+from assets.database import openDBConnection
+import re
+from models.sessionkey import SessionKey
+
+
+class User(BaseModel):
+    ID: int
+    EMAIL: Optional[str] = None
+    PASSWORD_HASH: Optional[str] = None
+    NAME: Optional[str] = None
+    LASTNAME: Optional[str] = None
+    CLASS: Optional[str] = None
+    PERMISSION_LEVEL: Optional[int] = None
+    LAST_IP: Optional[IPvAnyAddress] = None
+    ACTIVE: Optional[bool] = None
+    SESSION_KEY: Optional[SessionKey] = None
+
+    @validator('ID')
+    def is_positiv(cls, values):
+        if values > 0:
+            return values
+        raise ValueError('ID <= 0 is not possible')
+
+    @validator('EMAIL')
+    def is_valid_email(cls, values):
+        regex = r"^([A-z]+\.[A-z]+(\d\d)?@htl-salzburg.ac.at)$"
+        if re.search(regex, values):
+            return values
+        else:
+            raise ValueError('EMAIL not Valid')
+
+    @validator('CLASS')
+    def is_valid_class(cls, values):
+        regex = r"^([1-8][A-I]([A-Z]{2,4})|(LEHRER))$"
+        if re.search(regex, values):
+            return values
+        else:
+            raise ValueError('CLASS in not Valid')