33 lines
869 B
Python
33 lines
869 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):
|
|
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
|