python - 如何将我的两个 @listens_for 事件回调重构为一个？

Question

我有两个在 before_insert 和 before_update 上触发的回调。执行的代码是完全相同的代码，我想做的是重构它，以便 @listens_for 执行相同的方法。

这是我的:

@listens_for(Employee, 'before_insert')
def change_employee_path(mapper, connection, target):
    if target.path:
        os.rename(os.path.join(app.config['UPLOAD_FOLDER'], target.path), 'static/images/' + remove_blackduck_address(target.email))
        target.path = remove_blackduck_address(target.email)

@listens_for(Employee, 'before_update')
def change_employee_path(mapper, connection, target):
    if target.path:
        os.rename(os.path.join(app.config['UPLOAD_FOLDER'], target.path), 'static/images/' + remove_blackduck_address(target.email))
        target.path = remove_blackduck_address(target.email)

这就是我想要的(顺便说一句不起作用):

def change_employee_path(mapper,connection, target):
    if target.path:
        os.rename(os.path.join(app.config['UPLOAD_FOLDER'], target.path), 'static/images/' + remove_blackduck_address(target.email))
        target.path = remove_blackduck_address(target.email)

@listens_for(Employee, 'before_insert')
change_employee_path(mapper, connection, target)

@listens_for(Employee, 'before_update')
change_employee_path(mapper, connection, target)

有什么办法可以实现吗？

Answer 1

您需要使用 sqlalchemy.event.listen 函数并在声明您的 change_employee_path 函数后调用它:

listen(Employee, 'before_insert', change_employee_path)
listen(Employee, 'before_update', change_employee_path)