mirror of
https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git
synced 2025-08-14 00:25:46 +02:00
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
"""Employee CRUD operations"""
|
|
from sqlalchemy.orm import Session
|
|
from typing import Optional, List
|
|
from ..models.employee import Employee
|
|
from ..schemas.employee import EmployeeCreate, EmployeeUpdate
|
|
from ..utils.loggers import auth_logger
|
|
|
|
def get_employees(db: Session, skip: int = 0, limit: int = 100) -> List[Employee]:
|
|
"""Get all employees"""
|
|
return db.query(Employee).offset(skip).limit(limit).all()
|
|
|
|
def get_employee(db: Session, employee_id: int) -> Optional[Employee]:
|
|
"""Get employee by ID"""
|
|
return db.query(Employee).filter(Employee.id == employee_id).first()
|
|
|
|
def get_employee_by_email(db: Session, email: str) -> Optional[Employee]:
|
|
"""Get employee by email"""
|
|
return db.query(Employee).filter(Employee.email == email).first()
|
|
|
|
def create_employee(db: Session, employee: EmployeeCreate, hashed_password: str) -> Employee:
|
|
"""Create new employee"""
|
|
try:
|
|
db_employee = Employee(
|
|
email=employee.email,
|
|
full_name=employee.full_name,
|
|
hashed_password=hashed_password,
|
|
is_active=employee.is_active,
|
|
is_admin=employee.is_admin,
|
|
department=employee.department
|
|
)
|
|
db.add(db_employee)
|
|
db.commit()
|
|
db.refresh(db_employee)
|
|
return db_employee
|
|
except Exception as e:
|
|
auth_logger.error(f"Error creating employee: {e}")
|
|
db.rollback()
|
|
raise
|
|
|
|
def update_employee(db: Session, employee_id: int, employee: EmployeeUpdate) -> Optional[Employee]:
|
|
"""Update employee"""
|
|
db_employee = get_employee(db, employee_id)
|
|
if not db_employee:
|
|
return None
|
|
|
|
for field, value in employee.model_dump(exclude_unset=True).items():
|
|
setattr(db_employee, field, value)
|
|
|
|
try:
|
|
db.commit()
|
|
db.refresh(db_employee)
|
|
return db_employee
|
|
except Exception as e:
|
|
auth_logger.error(f"Error updating employee: {e}")
|
|
db.rollback()
|
|
raise
|
|
|
|
def delete_employee(db: Session, employee_id: int) -> Optional[Employee]:
|
|
"""Delete employee"""
|
|
db_employee = get_employee(db, employee_id)
|
|
if db_employee:
|
|
try:
|
|
db.delete(db_employee)
|
|
db.commit()
|
|
return db_employee
|
|
except Exception as e:
|
|
auth_logger.error(f"Error deleting employee: {e}")
|
|
db.rollback()
|
|
raise
|
|
return None |