from flask import Flask, Blueprint
from flask_restplus_marshmallow import abort, Api, Schema, Resource, Namespace, JSONParameters
from marshmallow import fields
from pony.orm import db_session
import typing

app = Flask(__name__)
blueprint = Blueprint('my-api', __name__)
api = Api(
  blueprint,
  title='My REST API',
  version='1.0',
  description='My super helpful description for my REST API'
)


auth_ns = Namespace(
  name='authentication',
  description='My authentication related routes',
  path="/authentication",
  db_context=db_session
)

class AuthenticationRequestSchema(JSONParameters):

  password = fields.String(required=True)
  email = fields.Email(required=True)

class AuthenticationSuccessfulResponseSchema(Schema):

  user = fields.Nested('MyUserSchema')
  token = fields.String()

@auth_ns.route("/login")
class LoginRoutes(Resource):
  
  @auth_ns.parameters(AuthenticationRequestSchema())
  @auth_ns.response(code=401)
  @auth_ns.response(AuthenticationSuccessfulResponseSchema())
  def post(self, data: typing.Dict) -> typing.Dict:
    successfully_authenticated = ... # some authentication related business logic here
    if successfully_authenticated:
      return {
        "user": {}, # JSON response here
        "token": "foo_bar_baz"
      }
    return abort(401)


api.add_namespace(auth_ns)