university-final-iot-backend/app/api/resources/token.py

34 lines
1.0 KiB
Python
Raw Normal View History

from flask_restful import Resource, abort
from webargs import fields
from marshmallow import Schema
from webargs.flaskparser import use_args
2018-05-07 14:13:07 +00:00
from flasgger import swag_from
from app.api.auth_protection import ProtectedResource
import app.accounts.api as accounts
2018-05-06 19:42:21 +00:00
class UserInfoSchema(Schema):
username = fields.Str(required=True)
password = fields.Str(required=True)
2018-05-06 19:42:21 +00:00
class TokenResource(Resource):
@use_args(UserInfoSchema(), locations=('json',))
2018-05-07 14:13:07 +00:00
@swag_from('swagger/create_token_spec.yaml')
def post(self, args):
2018-05-06 19:42:21 +00:00
try:
token = accounts.create_token(
args['username'],
args['password'])
if token:
return {'status': 'success', 'token': token}, 200
2018-10-15 17:51:51 +00:00
except ValueError as e:
abort(401, message=str(e), status='error')
2018-05-22 14:20:14 +00:00
class ValidateTokenResource(ProtectedResource):
@swag_from('swagger/validate_token_spec.yaml')
def get(self):
return {'status': 'success', 'message': 'Valid token'}, 200