#sora#实验2
昨天还是差不多12点半才睡，今天下午回去上自修肯定会困。。。
稍微试了下把flask应用中的类写在不同的python文件中。
实验目的：把flask应用中的类写成独立的模块，避免在主程序中写太多的class
源码：
flaskrr.py：
__author__ = 'hochikong'
from flask import Flask,request
from flask.ext.restful import Resource,Api,reqparse
from users import Users
app = Flask(__name__)
api = Api(app)
todos = {'task':'get the list of docker'}
parser = reqparse.RequestParser()
parser.add_argument('name',type=str,help='get the name')
parser.add_argument('auth-token',type=str,help='put the token here',location='headers')
class TodoSimple(Resource):
def get(self,todo_id):
args = parser.parse_args()
if args['auth-token'] == 'thisismytoken':
return {todo_id:todos[todo_id]}
else:
return {'error':'token error'},401
def put(self,todo_id):
args = parser.parse_args()
if args['auth-token'] == 'thisismytoken':
todos[todo_id] = request.json.get('data')
return todos,201
else:
return {'status':'error'},401
class GetName(Resource):
def post(self):
args = parser.parse_args()
name = args['name']
#name = {}
#name['ac'] = args['name']
#name = request.json.get('name')
return {'yourame':name}
api.add_resource(TodoSimple,'/todo/<string:todo_id>')
api.add_resource(GetName,'/getname')
api.add_resource(Users,'/users/','/users/<string:user_id>')
if __name__ == '__main__':
app.run(debug=True)
users.py：
__author__ = 'hochikong'
from flask import request
from flask.ext.restful import Resource,reqparse
parser = reqparse.RequestParser()
parser.add_argument('auth-token',type=str,help='put the token here',location='headers')
class Users(Resource):
users = {"admin":"hochikong"}
def get(self):
args = parser.parse_args()
if args['auth-token'] == 'thisismytoken':
return self.users
else:
return {"status":"error","reason":"token error"}
def post(self,user_id):
args = parser.parse_args()
if args['auth-token'] == 'thisismytoken':
self.users[user_id] = request.json.get('username')
return self.users
else:
return {"status":"error","reason":"token error"}
测试的是users.py中的Users类，实现了get和post方法，可以创建或查询用户
设计要求：避免把所有类写在主程序里，代码模块化，提高可读性
代码分析：
每个独立的模块，应该导入并继承flask-restful的Resource类：
from flask.ext.restful import Resource,reqparse
class Users(Resource):
当方法中存在需要reqparse处理的参数，你也要导入reqparse和定义解析参数：
parser = reqparse.RequestParser()
parser.add_argument('auth-token',type=str,help='put the token here',location='headers')
在这里解析发送来的json文档是直接依靠request.json.get()解析的：
self.users[user_id] = request.json.get('username')
在这个例子中，用户数据只是保存在内存中，在sora的实际开发中，还得用sqlalchemy处理。
要使用这个类，只要在主程序中import User类，并添加为资源即可：
from users import Users
api.add_resource(Users,'/users/','/users/<string:user_id>')
这里为Users资源配置了多个url，/users是使用GET获取用户信息时访问的url，而加了user_id变量的是用于创建新的用户。
测试过程：
自然要先启动flaskrr.py
用昨天安装的REST client测试：
先测试get方法：
可以看到数据的正确输出
测试下post方法：
：），用了全蛋哥的名字了
使用错误的token：
基本测试完毕，现在就差实际开发了