from functools import wraps from flask import request from api.models.trainer import Trainer from .errors import AuthenticationFailure from api.app import app import requests import json import jwt class HTTPError(Exception): def __init__(self, message): self.message = message class NotFound(Exception): pass def get_or_not_found(callback): try: resource = callback() if resource is None: raise NotFound() return resource except: raise NotFound() def get_trainer_fail(id): return get_or_not_found(lambda : Trainer.query.get(id)) def get_trainer_by_nick_fail(nickname): return get_or_not_found(lambda : Trainer.query.filter_by(nickname=nickname).one()) def get_pokemon_fail(trainer, id): return get_or_not_found(lambda : trainer.pokemons_list.filter_by(id=id).one()) # authenticação do trainer (decorator) def token_required(f): @wraps(f) def decorated(*args, **kwargs): try: token = request.headers["authorization"] data = jwt.decode(token, app.config["SECRET_KEY"], algorithms=["HS256"]) trainer = get_trainer_by_nick_fail(data["username"]) except (TypeError, KeyError): return AuthenticationFailure("JWT token required") except NotFound: return AuthenticationFailure("Trainer not found") except: return AuthenticationFailure("JWT token is invalid or expired") return f(trainer, *args, **kwargs) return decorated # seguintes funções puxam informações da pokeapi def set_pokemon_data(pokemon): try: response = requests.get("https://pokeapi.co/api/v2/pokemon/{}".format(pokemon.pokemon_id)) if response.status_code != 200: raise HTTPError("Could not fetch pokemon with id {}".format(pokemon.pokemon_id)) pokemon.pokemon_data = json.loads(response.text) except: raise HTTPError("Could not fetch pokemon with id {}".format(pokemon.pokemon_id)) async def async_set_pokemon_data(session, pokemon): try: response = await session.get("https://pokeapi.co/api/v2/pokemon/{}".format(pokemon.pokemon_id)) if response.status != 200: raise HTTPError("Could not fetch pokemon with id {}".format(pokemon.pokemon_id)) pokemon.pokemon_data = json.loads(await response.text()) except: raise HTTPError("Could not fetch pokemon with id {}".format(pokemon.pokemon_id))