Fix database

backend/app/dependencies.py

@@ -0,0 +1,63 @@
+"""Dependencies module"""
+from typing import Generator, Any
+from sqlalchemy.orm import Session
+from fastapi import Depends, HTTPException, status
+from fastapi.security import OAuth2PasswordBearer
+
+from .database import SessionLocal
+from .core.config import settings
+from .utils.jwt import verify_token
+from .models.employee import Employee
+
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="api/auth/login")
+
+def get_db() -> Generator[Session, Any, None]:
+    """Get database session"""
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+async def get_current_employee(
+    db: Session = Depends(get_db),
+    token: str = Depends(oauth2_scheme)
+) -> Employee:
+    """Get current employee"""
+    credentials_exception = HTTPException(
+        status_code=status.HTTP_401_UNAUTHORIZED,
+        detail="Could not validate credentials",
+        headers={"WWW-Authenticate": "Bearer"},
+    )
+    
+    employee_id = verify_token(token)
+    if not employee_id:
+        raise credentials_exception
+    
+    employee = db.query(Employee).filter(Employee.id == employee_id).first()
+    if not employee:
+        raise credentials_exception
+    
+    return employee
+
+async def get_current_active_employee(
+    current_employee: Employee = Depends(get_current_employee),
+) -> Employee:
+    """Get current active employee"""
+    if not current_employee.is_active:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail="Inactive employee"
+        )
+    return current_employee
+
+async def get_current_admin(
+    current_employee: Employee = Depends(get_current_employee),
+) -> Employee:
+    """Get current admin"""
+    if not current_employee.is_admin:
+        raise HTTPException(
+            status_code=status.HTTP_403_FORBIDDEN,
+            detail="The employee doesn't have enough privileges"
+        )
+    return current_employee