From 89316d51b7646a390b11059add76f90e9da6d267 Mon Sep 17 00:00:00 2001 From: Sourcery AI <> Date: Mon, 13 Nov 2023 14:35:34 +0000 Subject: [PATCH] 'Refactored by Sourcery' --- agenda_api/models/pessoa_read.py | 3 +-- agenda_api/repositories.py | 6 +++--- 2 files changed, 4 insertions(+), 5 deletions(-) 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: