Draft: Authentication Service Model and Views

requirements.txt

@@ -7,3 +7,4 @@ jinja2==3.1.3
 pyyaml==6.0.1
 pytest
 types-requests
+passlib

src/data/database.py

@@ -1,8 +1,10 @@
 from typing import Any, Generator
 from sqlmodel import SQLModel, create_engine, Session
 from sqlalchemy_utils import database_exists, create_database
+from sqlalchemy.ext.declarative import declarative_base
 import constants
 
+Base = declarative_base()
 
 def create_tables(db_engine) -> None:
     from models.turnilo_dashboard import TurniloDashboard

src/models/authentication.py

@@ -0,0 +1,14 @@
+from sqlalchemy import Column, Integer, String, Boolean
+
+from data.database import Base
+
+
+class User(Base):
+    __tablename__ = "users"
+
+    id = Column(Integer, primary_key=True, index=True)
+    tenant_id = Column(Integer, index=True) # Not sure about this one
+    email = Column(String, unique=True, index=True)
+    hashed_password = Column(String)
+    is_active = Column(Boolean, default=True)
+    oauth_provider = Column(String, default=None)

src/views/authentication.py

@@ -0,0 +1,25 @@
+from pydantic import BaseModel
+from passlib.context import CryptContext
+
+from models.authentication import User as UserModel
+
+pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
+
+
+def get_password_hash(password) -> str:
+    return pwd_context.hash(password)
+
+
+class UserRequest(BaseModel):
+    email: str
+    password: str
+
+    def to_model(self, tenant_id: int, oauth_provider: str) -> UserModel:
+        return UserModel(
+            tenant_id=tenant_id, 
+            email=self.email, 
+            hashed_password=get_password_hash(self.password), # TODO: Interline gibberish for extra protection
+            oauth_provider=oauth_provider,
+        )
+
+