make return value of crud.put_item easier to understand

This commit is contained in:
clonejo 2024-01-14 18:57:32 +01:00
parent eba9b410c6
commit 99df802a40
2 changed files with 11 additions and 2 deletions

11
crud.py
View file

@ -1,3 +1,4 @@
from enum import Enum
from sqlalchemy import DECIMAL, cast
from sqlalchemy.orm import Session
@ -17,7 +18,15 @@ def put_item(db: Session, id: str, item: schemas.Item):
if not updated:
db.add(models.Item(**item.dict()))
db.commit()
return updated
if updated:
return PutItemResult.UPDATED
else:
return PutItemResult.ADDED
class PutItemResult(Enum):
ADDED = 1
UPDATED = 2
def delete_item(db: Session, id: str):