app.py

import os

from flask_restful import reqparse, abort, Api, Resource

from config import *

os.environ['NLS_LANG'] = 'SIMPLIFIED CHINESE_CHINA.UTF8'
from flask import Flask

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

db = SQlManager()
db1 = db.get_list('select todo_id from TODOS')
db2 = db.get_list('select task from TODOS')

res1 = [item[key] for item in db1 for key in item]
TODOS = dict(map(lambda x, y: [x, y], res1, db2))


def abort_if_todo_doesnt_exist(todo_id):
    if todo_id not in TODOS:
        abort(404, message="Todo {} doesn't exist".format(todo_id))


parser = reqparse.RequestParser()
parser.add_argument('task', type=str)


# Todo
# show a single todo item and lets you delete them
class Todo(Resource):
    def get(self, todo_id):
        abort_if_todo_doesnt_exist(todo_id)
        return TODOS[todo_id]

    def delete(self, todo_id):
        abort_if_todo_doesnt_exist(todo_id)
        del TODOS[todo_id]
        db.moddify('delete from TODOS where todo_id = "{}"'.format(todo_id))
        return '', 204

    def put(self, todo_id):
        args = parser.parse_args()
        task = {'task': args['task']}
        TODOS[todo_id] = task
        db.moddify('update TODOS set task="{}" where todo_id="{}";'.format(args['task'], todo_id))
        return task, 201


# TodoList
#   shows a list of all todos, and lets you POST to add new tasks
class TodoList(Resource):
    def get(self):
        return TODOS

    def post(self):
        args = parser.parse_args()
        todo_id = int(max(TODOS.keys()).lstrip('todo')) + 1
        todo_id = 'todo%i' % todo_id
        TODOS[todo_id] = {'task': args['task']}
        db.moddify('insert into TODOS (todo_id,task) values("{}","{}");'.format(todo_id, args['task']))
        return TODOS[todo_id], 201


##
## Actually setup the Api resource routing here
##

api.add_resource(TodoList, '/todos')
api.add_resource(Todo, '/todos/<todo_id>')

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