python钩子函数实现 python 钩子

转载

网络安全专家 2023-12-07 07:03:34

文章标签 python钩子函数实现钩子函数 before_request flask python 文章分类 Python 后端开发

Flask上下文

Flask项目中有两个上下文，一个是应用上下文（app），另外一个是请求上下文（request）。请求上下文request和应用上下文current_app都是一个全局变量。所有请求都共享的。Flask有特殊的机制可以保证每次请求的数据都是隔离的，即A请求所产生的数据不会影响到B请求。所以可以直接导入request对象，也不会被一些脏数据影响了，并且不需要在每个函数中使用request的时候传入request对象。这两个上下文具体的实现方式和原理可以没必要详细了解。只要了解这两个上下文的四个属性就可以了：

request： 请求上下文上的对象。这个对象一般用来保存一些请求的变量。比如method、args、form等。
session： 请求上下文上的对象。这个对象一般用来保存一些会话信息。
current_app： 返回当前的app。
g：应用上下文上的对象。处理请求时用作临时存储的对象。

常用的钩子函数

before_first_request：处理第一次请求之前执行。例如以下代码：

@app.before_first_request
def first_request():
    print 'first time request'

before_request：在每次请求之前执行。通常可以用这个装饰器来给视图函数增加一些变量。例如以下代码：

@app.before_request
def before_request():
    if not hasattr(g,'user'):
        setattr(g,'user','xxxx')

teardown_appcontext：不管是否有异常，注册的函数都会在每次请求之后执行。

@app.teardown_appcontext
def teardown(exc=None):
    if exc is None:
        db.session.commit()
    else:
        db.session.rollback()
        db.session.remove()

template_filter：在使用Jinja2模板的时候自定义过滤器。比如可以增加一个upper的过滤器（当然Jinja2已经存在这个过滤器，本示例只是为了演示作用）：

@app.template_filter
def upper_filter(s):
    return s.upper()

context_processor：上下文处理器。返回的字典中的键key可以在模板上下文中使用。例如：

@app.context_processor
    return {'current_user':'xxx'}

errorhandler： errorhandler接收状态码，可以自定义返回这种状态码的响应的处理方法。例如：

@app.errorhandler(404)
def page_not_found(error):
    return 'This page does not exist',404

示例

before_request

from flask import Flask,request,render_template,session,g,redirect,url_for
import os

app = Flask(__name__)
app.config['SECRET_KEY'] = os.urandom(24)

@app.route('/')
def index():
    return '这是首页！'

@app.route('/login/',methods=['GET','POST'])
def login():
    if request.method == 'GET':
        return render_template('login.html')
    else:
        username = request.form.get('username')
        password = request.form.get('password')
        if username == '李四' and password == '111':
            session['username'] = username
            return '登陆成功！你好：' + username
        else:
            return '登陆失败！'

@app.route('/edit/')
def edit():
    if hasattr(g,'username'):
        return '成功进入编辑页面！'
    else:
        return redirect(url_for('login'))

@app.before_request
def my_before_request():
    if session.get('username'):
        g.username = session.get('username')

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

<form action="{{ url_for('login') }}" method="post">
        <table>
            <tbody>
                <tr>
                    <td>用户名：</td>
                    <td><input type="text" placeholder="请输入用户名：" name="username"></td>
                </tr>
                <tr>
                    <td>密码：</td>
                    <td><input type="password" placeholder="请输入密码：" name="password"></td>
                </tr>
                <tr>
                    <td></td>
                    <td><input type="submit" value="提交"></td>
                </tr>
            </tbody>
        </table>
    </form>

注意： 其中hasattr(g,'username')是用来判断g对象中是否存在username这个属性。

hasattr(object, name) ：判断object对象中是否存在name属性，当然对于python的对象而言，属性包含变量和方法；有则返回True，没有则返回False；需要注意的是username参数是string类型，所以不管是要判断变量还是方法，其名称都以字符串形式传参 ；getattr和setattr也同样；类似的还有getattr和setattr
getattr(object, name[, default]) ：获取object对象的属性的值，如果存在则返回属性值，如果不存在分为两种情况，

一种是没有default参数时，会直接报错；给定了default参数，若对象本身没有name属性，则会返回给定的default值；
如果给定的属性name是对象的方法，则返回的是函数对象，需要调用函数对象来获得函数的返回值；调用的话就是函数对象后面加括号，如func之于func();
另外还需要注意，如果给定的方法func()是实例函数，则不能写getattr(A, ‘func’)()，因为fun()是实例函数的话，是不能用A类对象来调用的，应该写成getattr(A(), ‘func’)()；实例函数和类函数的区别可以简单的理解一下，实例函数定义时，直接def func(self):，这样定义的函数只能是将类实例化后，用类的实例化对象来调用；而类函数定义时，需要用@classmethod来装饰，函数默认的参数一般是cls，类函数可以通过类对象来直接调用，而不需要对类进行实例化；