diff --git a/main.py b/main.py index 1b46f29..ffef5f5 100644 --- a/main.py +++ b/main.py @@ -30,6 +30,7 @@ async def ver_cardapio_campus(campus: str, response: Response): response.status_code = status.HTTP_200_OK return {'cardapios': cardapio} + @app.get("/campus/{campus}/dia/{dia}") async def ver_cardapio_campus_dia(campus: str, dia: int, response: Response): bs = get_cardapio(campus) diff --git a/webscraping/get_data.py b/webscraping/get_data.py index 157e941..4119ac7 100644 --- a/webscraping/get_data.py +++ b/webscraping/get_data.py @@ -17,9 +17,13 @@ def normalize_url(url: str): def get_cardapio(campus: str): try: if campus == 'realeza': - html = urlopen(f"https://www.uffs.edu.br/campi/{normalize_url(campus)}/restaurante_universitario/apresentacao-do-ru") + url = f"https://www.uffs.edu.br/campi/{normalize_url(campus)}/restaurante_universitario/apresentacao-do-ru" + elif campus == 'passo-fundo': + url = f"https://www.uffs.edu.br/campi/{normalize_url(campus)}/restaurante-universitario" else: - html = urlopen(f"https://www.uffs.edu.br/campi/{normalize_url(campus)}/restaurante_universitario") + url = f"https://www.uffs.edu.br/campi/{normalize_url(campus)}/restaurante_universitario" + html = urlopen(url) + except HTTPError: return False if html.code != 200: