from sqlalchemy.orm import Session
from app.models.employees import Employee
from app.schemas.employees import EmployeeCreate, EmployeeUpdate
from datetime import datetime
def get_employee(db: Session, employee_id: int):
    return db.query(Employee).filter(Employee.id == employee_id).first()

def get_employees(db: Session, skip: int = 0, limit: int = 10):
    return db.query(Employee).offset(skip).limit(limit).all()



def create_employee(db: Session, employee_data):
    new_employee = Employee(
        first_name=employee_data.first_name,
        last_name=employee_data.last_name,
        matricule=employee_data.matricule,
        mail=employee_data.mail,
        category_id=employee_data.category_id,
        is_active=employee_data.is_active,
        created_at=datetime.utcnow(),  # Ajout de la date de création
        last_update=datetime.utcnow()  # Ajout de la date de mise à jour
    )
    db.add(new_employee)
    db.commit()
    db.refresh(new_employee)
    return new_employee


def update_employee(db: Session, employee_id: int, employee_data: EmployeeUpdate):
    db_employee = db.query(Employee).filter(Employee.id == employee_id).first()
    if db_employee:
        for key, value in employee_data.dict().items():
            setattr(db_employee, key, value)
        db.commit()
        db.refresh(db_employee)
    return db_employee

def delete_employee(db: Session, employee_id: int):
    db_employee = db.query(Employee).filter(Employee.id == employee_id).first()
    if db_employee:
        db.delete(db_employee)
        db.commit()
    return db_employee
