From d43e4079e7384f24f779ebd67d91f6e8b9c56a36 Mon Sep 17 00:00:00 2001 From: "diego.diogenes" Date: Thu, 8 Aug 2019 17:02:21 -0300 Subject: [PATCH] =?UTF-8?q?Adiciona=20m=C3=A9todo=20de=20not=5Ffound=20e?= =?UTF-8?q?=20not=5Frelation=20#47?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- odufrn_downloader/modules/Env.py | 16 ++++++++++++++-- odufrn_downloader/modules/Group.py | 2 +- odufrn_downloader/modules/Package.py | 7 ++----- odufrn_downloader/modules/Tag.py | 5 +---- 4 files changed, 18 insertions(+), 12 deletions(-) diff --git a/odufrn_downloader/modules/Env.py b/odufrn_downloader/modules/Env.py index 928557e..ea51862 100644 --- a/odufrn_downloader/modules/Env.py +++ b/odufrn_downloader/modules/Env.py @@ -21,13 +21,24 @@ def __init__(self): self.url_action = self.url_base + 'api/action/' def _print_exception(self, ex: Exception): - """Imprime mensagem padrão para exceções.""" + """Imprime mensagem padrão para exceções. + """ print('\033[91m{}\033[0m'.format(ex)) print( "Ocorreu algum erro durante o download do pacote. " "Verifique sua conexão, o nome do conjunto de dados " "e tente novamente." ) + + def _print_not_found(self, name: str, type_name: str): + """Imprime mensagem padrão para nome de dados não encontrados. + """ + print('{} de dados "{}" não foi encontrado.'.format(type_name, name)) + + def _print_not_relation(self, name: str, type_name: str): + """Imprime mensagem padrão para nome de dados semelhantes não encontrados. + """ + print('Não há {} semelhante a {}'.format(type_name, name)) def _print_list(self, name: str, variable: list): """Mostra na tela a lista desejada.""" @@ -41,7 +52,8 @@ def _load_list(self, option: str) -> list: Parâmetros ---------- option: str - indica o que se deseja consultar pelo request.""" + indica o que se deseja consultar pelo request. + """ try: packages = requests.get(self.url_action + option).json() return packages['result'] diff --git a/odufrn_downloader/modules/Group.py b/odufrn_downloader/modules/Group.py index ac5ff01..c5d17af 100644 --- a/odufrn_downloader/modules/Group.py +++ b/odufrn_downloader/modules/Group.py @@ -71,7 +71,7 @@ def download_group(self, name: str, path: str = os.getcwd(), # Checa se o grupo está disponível if not (name in self.available_groups): - print("O grupo de pacotes \"{}\" não foi encontrado.".format(name)) + self._print_not_found('Grupo', name) return groups = self._request_get(self.url_group + name) diff --git a/odufrn_downloader/modules/Package.py b/odufrn_downloader/modules/Package.py index cb8aa98..05dc6d7 100644 --- a/odufrn_downloader/modules/Package.py +++ b/odufrn_downloader/modules/Package.py @@ -58,7 +58,7 @@ def download_package(self, name: str, path: str = os.getcwd(), # Checa se o pacote está disponível if not (name in self.available_packages): - print('O conjunto de dados "{}" não foi encontrado.'.format(name)) + self._print_not_found('Grupo', name) return response = self._request_get(self.url_package + name) @@ -140,10 +140,7 @@ def search_related_packages(self, keyword: str, # Imprime exceção se não houver pacotes similares if not len(related): - print( - "Não há nenhum pacote de dados semelhante" - " a \"{}\".".format(keyword) - ) + self._print_not_relation(keyword, 'package') return related diff --git a/odufrn_downloader/modules/Tag.py b/odufrn_downloader/modules/Tag.py index 1ae2149..ecb0727 100644 --- a/odufrn_downloader/modules/Tag.py +++ b/odufrn_downloader/modules/Tag.py @@ -42,10 +42,7 @@ def search_by_tag(self, tag: str) -> list: tags = self.search_related(tag, self.available_tags, False) # Imprime exceção se não houver pacotes if not len(tags): - print( - "Não há nenhuma etiqueta semelhante" - " a \"{}\".".format(tag) - ) + self._print_not_relation(tag, 'tag') packages = [] for key in tags: