diff --git a/backend/bin/server.py b/backend/bin/server.py index d6cecb1..29e1841 100644 --- a/backend/bin/server.py +++ b/backend/bin/server.py @@ -1,21 +1,28 @@ version = "0.1" from flask import Flask +from flask_restx import Resource, Api + app = Flask(__name__) +app.url_map.strict_slashes = False +api = Api(app, version=version, title='CyCAT.org API', description='CyberSecurity catalog API services', doc='/doc/') import uuid import inspect import redis -r = redis.Redis(host='127.0.0.1', port='3033') -@app.route('/info') -def info(): - return "CyCAT backend {}".format(version) -@app.route('/generate/uuid') -def generateUUID(): - genuuid = uuid.uuid4() - k = "stats:f:{}".format(inspect.stack()[0][3].lower()) - r.incr(k, 1) - return "{}".format(genuuid) +r = redis.Redis(host='127.0.0.1', port='3033') +@api.route('/info') +class info(Resource): + def get(self): + return "CyCAT backend {}".format(version) + +@api.route('/generate/uuid') +class generateUUID(Resource): + def get(self): + genuuid = uuid.uuid4() + k = "stats:f:{}".format(inspect.stack()[0][3].lower()) + r.incr(k, 1) + return "{}".format(genuuid) if __name__ == '__main__': app.run() diff --git a/backend/requirements.txt b/backend/requirements.txt index 1a5dc97..039899b 100644 --- a/backend/requirements.txt +++ b/backend/requirements.txt @@ -1,2 +1,3 @@ flask redis +flask_restplus