Flask 给视图函数增加装饰器
在@app.route的情况下增加装饰器的写法: from flask import Flask,request,render_template,redirect,session,url_for,views from flask import render_template app = Flask(__name__) #实例化flask对象 app.debug = True #能够随时更改自动重启,不加的话每次更改代码需要手动重启 app.config['SECRET_KEY'] = '123456' #secret_key,用于给session加密 @app.route('/login',methods=['GET','POST'],endpoint='t1') #endpoint是url的别名,相当于django中Url的name def login(): if request.method == "GET": # res = request.query_string # print(res) 获取通过GET请求url传过来的参数 return render_template('login.html') else: user = request.form.get('user') pwd = request.form.get('pwd') if user == 'tom' and pwd == '123': session['userinfo'] = user #设置session return render_template('hello.html') return render_template('login.html', error='用户名或密码错误') def wapper(func): def inner(*args,**kwargs): user = session.get('user_info') if not user: return redirect('/login') return func(*args,**kwargs) return inner @app.route('/detail/<int:nid>',methods=['GET'],endpoint='n1') @wapper def detail(nid): print(nid) return render_template('hello.html') ''' 如果给一个视图函数增加装饰器,应该加在app.route下面,这样的效果就是, 装饰器将下面的所有内容包裹,然后路由对应到这大的包裹中来。 需要注意endpoint要注明,如果不注明endpoint则默认用函数名来定义, 此时所有的函数名都叫inner了,所以需要注明endpoint,只是为了区分。 ''' if __name__ == '__main__': app.run()
另一种写法: import functools def wapper(func): @functools.wraps(func) def inner(*args,**kwargs): return func(*args,**kwargs) return inner ''' functools.wraps()相当于保留元信息 说白了就是,如果不加这个装饰器,那么你打印detail的__name__它就是inner了, 因为加了装饰器,效果等同于inner=inner(detail()), 如果在装饰器中加了functools这个装饰器,那么相当于给__name__重新赋值,inner.__name__ = func.__name_- 其函数的名字得以保留。 ''' @wapper def detail(): pass print(detail.__name__)
flask的get_flashed_messages,flash from flask import Flask,get_flashed_messages,flash app = Flask(__name__) app.secret_key = 'asdf' @app.route('/get') def get(): data = get_flashed_messages() print(data) return 'Hello world' @app.route('/set') def set(): flash('info info') ''' 闪现效果,相当于set视图函数执行2次,会在一个列表中存储两个flash函数的内容, 当执行get_flashed_messages的时候则会取出该列表,并清空,类似字典的Pop。 具体用处不大。。。 ''' return 'Hello world' if __name__ == '__main__': app.run() flash还可以通过category参数给Flash内容归类,通过不同类别取不同内容。