flask例子

import json
import os

import pymysql
from flask import Flask, jsonify,abort,make_response,request,url_for
from flask_sqlalchemy import SQLAlchemy

pymysql.install_as_MySQLdb()

app = Flask(__name__)

basedir = os.path.abspath(os.path.dirname(__file__))

app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:123456@localhost/study'
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True

db = SQLAlchemy(app)


class Role(db.Model):
    __tablename__ = 'roles'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True)

    users = db.relationship('User', backref='role')

    def __repr__(self):
        return '<Roles %r>' % self.name


class User(db.Model):
    __table_name__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(64), unique=True, index=True)

    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))

    def __repr__(self):
        return '<User %r>' % self.username


@app.route('/user', methods=['GET', 'POST'])
def index():
    user = User.query.all()
    print('jhahfhafda', user)
    if user is None:
        user = db.session.query(User.id, User.username).all()
        db.session.add(user)
    print(user)
    resp = dict(user)
    return resp


@app.route('/')
def hello_world():
    s = ['张三', '年龄', '姓名']
    t = {}
    t['data'] = s
    return json.dumps(t)


tasks = [
    {
        'id': 1,
        'title': u'Buy groceries',
        'description': u'Milk, Cheese, Pizza, Fruit, Tylenol',
        'done': False
    },
    {
        'id': 2,
        'title': u'Learn Python',
        'description': u'Need to find a good Python tutorial on the web',
        'done': False
    }
]


# @app.route('/todo/api/tasks', methods=['GET'])
# def get_tasks():
#     return jsonify({'tasks': tasks})


@app.route('/todo/api/tasks/<int:task_id>', methods=['GET'])
def get_task(task_id):
    print(type(filter(lambda t: t['id'] == task_id, tasks)))
    task = list(filter(lambda t: t['id'] == task_id, tasks))
    if len(task) == 0:
        abort(404)
    return jsonify({'task':task[0]})

@app.errorhandler(404)
def not_found(error):
    return make_response(jsonify({'error':'Not found'}),404)


@app.route('/todo/api/tasks',methods=['POST'])
def create_task():
    if not request.json or not 'title' in request.json:
        abort(400)
    task = {
        'id':tasks[-1]['id'] + 1,
        'title':request.json['title'],
        'description':request.json.get('description',""),
        'done': False
    }

    tasks.append(task)
    return jsonify({'task':task}),201

@app.route('/todo/api/tasks/<int:task_id>',methods=['PUT'])
def update_task(task_id):
    task = list(filter(lambda t:t['id'] == task_id,tasks))
    if len(task) == 0:
        abort(404)
    if not request.json:
        abort(400)
    task[0]['title'] = request.json.get('title',task[0]['title'])
    task[0]['description'] = request.json.get('description',task[0]['description'])
    task[0]['done'] = request.json.get('done',task[0]['done'])
    return jsonify({'task':task[0]})


@app.route('/todo/api/tasks/<int:task_id>',methods=['DELETE'])
def delete_task(task_id):
    task = list(filter(lambda t:t['id'] == task_id,tasks))
    if len(task) == 0:
        abort(404)
    tasks.remove(task[0])
    return jsonify({'result':True})

def make_public_task(task):
    new_task = {}
    for field in task:
        if field == 'id':
            new_task['uri'] = url_for('get_task',task_id=task['id'],_external=True)
        else:
            new_task[field] = task[field]

    return new_task

@app.route('/todo/api/tasks',methods=['GET'])
def get_tasks():
    return jsonify({'tasks':list(map(make_public_task,tasks))})



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