from sqlalchemy.orm import Session
from app.api.v1.models.user import User
from app.api.utils import get_current_timestamp

def get_user(db: Session, id: int):
    return db.query(User).filter(User.id == id).first()

def get_all_user(db: Session):
    return db.query(User).all()

def create_user(db: Session, data: dict):
    obj = User(**data, created_at=get_current_timestamp())
    db.add(obj)
    db.commit()
    db.refresh(obj)
    return obj

def update_user(db: Session, id: int, data: dict):
    obj = db.query(User).filter(User.id == id).first()
    if obj:
        for key, value in data.items():
            setattr(obj, key, value)
        db.commit()
        db.refresh(obj)
    return obj

def delete_user(db: Session, id: int):
    obj = db.query(User).filter(User.id == id).first()
    if obj:
        db.delete(obj)
        db.commit()
    return obj