server.py

from flask import Flask, request, render_template
from flask_restful import Resource, Api
from content_hash import Content
import json

app = Flask(__name__)
api = Api(app)
data_con = Content()


class HomePage(Resource):
    @staticmethod
    def get():
        #return render_template('index.html')
        return "Welcome to Site!"


class AddNode(Resource):
    @staticmethod
    def post():
        sent_data = json.loads(request.get_json())    # [address_hash, content_hash, url]
        dict_a = {sent_data[0]: sent_data[1]}
        dict_b = {sent_data[1]: sent_data[2]}
        data_con.add_node(dict_a, dict_b)
        return {'result': 'added'}


class ReadNode(Resource):
    @staticmethod
    def get(text):   # address_hash
        address = text.strip()
        response = data_con.read_node(address)
        return response


class ReadReverse(Resource):
    @staticmethod
    def get(text):   # content_hash
        address = text.strip()
        response = data_con.read_reserve(address)
        return response


api.add_resource(HomePage, '/')
api.add_resource(AddNode, '/add/')
api.add_resource(ReadNode, '/read/hash/<text>')
api.add_resource(ReadReverse, '/read/url/<text>')


# if __name__ == '__main__':
#     app.run(debug=True)