app.py

from datetime import datetime, timedelta

from flask import Flask, jsonify, request
from flask_jwt import JWT, jwt_required
from werkzeug.security import safe_str_cmp

from data import alchemy
from model import show, episode, user

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///data.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.config['PROPAGATE_EXCEPTIONS'] = True
app.secret_key = 'supersecreto'


# Funções necessárias para o JWT
def authenticate(username,password):
    auth_user = user.UserModel.find_by_name(username)
    dict_error = {'erro':'Usuário ou senha inválidos'}
    if auth_user and safe_str_cmp(auth_user.password.encode('utf-8'),password.encode('utf-8')):
        return auth_user
    return auth_user, 401

def identity(payload):
    user_id = payload['user_name']
    return user.UserModel.find_by_name(user_id)

jwt = JWT(app,authenticate,identity)

@jwt.jwt_payload_handler
def make_payload(jwt_identity):
    expiration = datetime.now() +timedelta(hours=10)
    nbf = datetime.now() + timedelta(seconds=1)
    iat = datetime.now()
    return {
        'user_id' : jwt_identity.id,
        'user_name' : jwt_identity.name,
        'iat':iat,
        'exp':expiration,
        'nbf':nbf
    }

@app.before_first_request
def create_tables():
    alchemy.create_all()


@app.route('/', methods=['GET'])
@jwt_required()
def home():
    return "API Funcionando", 200

@app.route("/signup",methods=['POST'])
def signup():
  request_data = request.get_json()
  # Caso o email já exista, devolve conflito (status 409)
  if (user.UserModel.find_by_name(request_data['username'])):
      return {'message': 'email já encontrado'}, 409
  new_user = user.UserModel(name=request_data['username'],password=request_data['password'])
  new_user.save_to_db()
  return new_user.json()

# post /show data: {name :}
@app.route('/show', methods=['POST'])
@jwt_required()
def create_show():
    request_data = request.get_json()
    new_show = show.ShowModel(request_data['name'])
    new_show.save_to_db()
    result = show.ShowModel.find_by_id(new_show.id)
    return jsonify(result.json())


# get /show/<name> data: {name :}
@app.route('/show/<string:name>')
@jwt_required()
def get_show(name):
    result = show.ShowModel.find_by_name(name);
    if result:
        return result.json();
    return {'message': 'Série não encontrada'}, 404


# post /show/<name> data: {name :}
@app.route('/show/<string:name>/item', methods=['POST'])
@jwt_required()
def create_episode_in_show(name):
    request_data = request.get_json()
    parent = show.ShowModel.find_by_name(name)
    if parent:
        new_episode = episode.EpisodeModel(name=request_data['name'], season=request_data['season'], show_id=parent.id)
        new_episode.save_to_db()
        return new_episode.json()
    else:
        return {'message': 'Não existe a série'}, 404



if __name__ == '__main__':
    from data import alchemy

    alchemy.init_app(app)
    app.run(port=5000, debug=True)