diff --git a/agenda_api/models/pessoa_read.py b/agenda_api/models/pessoa_read.py index 938e265..1d5db0b 100644 --- a/agenda_api/models/pessoa_read.py +++ b/agenda_api/models/pessoa_read.py @@ -25,8 +25,7 @@ class PessoaRead(PessoaCreate): def _set_pessoa_id(cls, data): """altera o campo _id para pessoa_id e o a alias para "pessoa_id" """ - document_id = data.get("_id") - if document_id: + if document_id := data.get("_id"): data["pessoa_id"] = document_id return data diff --git a/agenda_api/repositories.py b/agenda_api/repositories.py index 274929e..2f1768f 100644 --- a/agenda_api/repositories.py +++ b/agenda_api/repositories.py @@ -14,10 +14,10 @@ class PessoasRepository: @staticmethod def get(pessoa_id: str) -> PessoaRead: """Get pessoa por seu ID""" - document = collection.find_one({"_id": pessoa_id}) - if not document: + if document := collection.find_one({"_id": pessoa_id}): + return PessoaRead(**document) + else: raise PessoaNotFoundException(pessoa_id) - return PessoaRead(**document) @staticmethod def list() -> PessoasRead: