71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import uuid
|
|
|
|
from flask_restful import Resource
|
|
from flask import request, current_app, abort, request
|
|
|
|
from fred import flaskred
|
|
from config import ENCODED_SECRET_KEY
|
|
from schemas import UserSchema, ListSchema, TrackSchema
|
|
from aes_encrypt import EncryptedUserRedis
|
|
|
|
"""
|
|
Flask Restful endpoints
|
|
"""
|
|
|
|
__author__ = '@tormakris'
|
|
__copyright__ = "Copyright 2020, onSpot Team"
|
|
__module_name__ = "resources"
|
|
__version__text__ = "1"
|
|
|
|
INVALID_JSON_SCHEMA_MSG = "invalid json schema"
|
|
|
|
|
|
class LoginApi(Resource):
|
|
"""
|
|
See: https://swagger.kmlabz.com/?urls.primaryName=onSpot%20Backend#/backend/logon
|
|
"""
|
|
|
|
userschema = UserSchema(many=False)
|
|
encryptor = EncryptedUserRedis(ENCODED_SECRET_KEY)
|
|
|
|
def post(self):
|
|
body = request.get_json()
|
|
|
|
try:
|
|
userobj = self.userschema.load(body)
|
|
except Exception as e:
|
|
current_app.logger.warning(e)
|
|
abort(417, INVALID_JSON_SCHEMA_MSG)
|
|
|
|
self.encryptor.store(body)
|
|
|
|
token = str(uuid.uuid4())
|
|
|
|
flaskred.set(token, userobj['name'].encode('UTF-8'))
|
|
|
|
return {
|
|
'token': token
|
|
}, 200
|
|
|
|
|
|
class LogoffApi(Resource):
|
|
"""
|
|
See: https://swagger.kmlabz.com/?urls.primaryName=onSpot%20Backend#/backend/logoff
|
|
"""
|
|
|
|
def delelete(self):
|
|
flaskred.delete(flaskred.get(request.headers.get('Authorization')).decode('UTF-8'))
|
|
flaskred.delete(request.headers.get('Authorization'))
|
|
return 204
|
|
|
|
|
|
class MeApi(Resource):
|
|
"""
|
|
See: https://swagger.kmlabz.com/?urls.primaryName=onSpot%20Backend#/backend/currentUser
|
|
"""
|
|
|
|
def get(self):
|
|
return {"name": flaskred.get(request.headers.get('Authorization')).decode('UTF-8')}, 200
|