32 lines
939 B
Python
32 lines
939 B
Python
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):
|
|
new_user = User(**data, created_at=get_current_timestamp(), updated_at=get_current_timestamp())
|
|
db.add(new_user)
|
|
db.commit()
|
|
db.refresh(new_user)
|
|
return new_user
|
|
|
|
def update_user(db: Session, id: int, data: dict):
|
|
user = db.query(User).filter(User.id == id)
|
|
if not user.first():
|
|
return None
|
|
user.update({**data, "updated_at": get_current_timestamp()})
|
|
db.commit()
|
|
return user.first()
|
|
|
|
def delete_user(db: Session, id: int):
|
|
user = db.query(User).filter(User.id == id).first()
|
|
if not user:
|
|
return None
|
|
db.delete(user)
|
|
db.commit()
|
|
return user |