|
|
- import os
- import json
-
- from flask import Flask, request
- from flask_restful import Resource, Api, abort
-
- app = Flask(__name__)
- api = Api(app)
-
- with open(os.path.join(app.root_path, 'ratings.json'), 'r') as f:
- ratings = json.load(f)
-
- class Ratings(Resource):
- def get(self):
- rating = [
- {
- 'id' : v['id'],
- 'name': v['name'],
- 'desc': v['desc'],
- 'img' : v['img']
- }
- for v in ratings
- ]
- return rating
-
- def post(self):
- """
- Example POST Data:
- name=<rating_name>&
- desc=<rating_desc>& # OPTIONAL
- img=<rating_img>& # OPTIONAL
- """
- args = request.form
- rating_id = len(ratings) + 1
- rating = {
- 'id': rating_id,
- 'name': args['name'],
- 'desc': args.get('desc'),
- 'img' : args.get('img'),
- 'rates': []
- }
-
- ratings.append(rating)
-
- with open(os.path.join(app.root_path, 'ratings.json'), 'w') as f:
- json.dump(ratings, f, indent=4)
-
- return rating
-
- class Rating(Resource):
- def get(self, rating_id):
- try:
- return ratings[rating_id - 1]
- except:
- abort(404, error="Rating {} doesn't exist".format(rating_id))
-
- class Rate(Resource):
- def get(self):
- """
- Example URL Query:
- /rate?
- rating_id=<rating_id>>&
- score=<score>&
- note=<note> # ADDITIONAL
- """
- rating_id = int(request.args['rating_id'])
- score = int(request.args['score'])
- if 0 >= score >= 10:
- abort(500, 'Score should be between 0 and 10')
- note = request.args.get('note')
- ratings[rating_id - 1]['rates'].append({
- 'id': len(ratings[rating_id - 1]['rates']) + 1,
- 'score': score,
- 'note': note
- })
- with open(os.path.join(app.root_path, 'ratings.json'), 'w') as f:
- json.dump(ratings, f, indent=4)
-
- return ratings[rating_id - 1]
-
- if __name__ == '__main__':
- api.add_resource(Ratings, '/ratings', '/ratings/')
- api.add_resource(Rating, '/ratings/<int:rating_id>', '/ratings/<int:rating_id>/')
- api.add_resource(Rate, '/rate', '/rate/')
-
- app.run(host='0.0.0.0', port=5000)
|