python MySQLdb:检查连通性的钩子函数

Question

我有一个使用mySQL全局调用数据库连接对象的python脚本。有一个插入函数insertToTable(actionDate, action)，它将一条记录插入到表中。if __name__=="__main__":中脚本的主要部分作为一个无限的while循环运行几个小时/几天。

问题是，mysql连接对象可能会长时间不使用，这会导致MySQL服务器超时连接。我目前在函数中使用try & except来捕获异常。然而，这会导致代码重复，因为我必须在每个函数中包含一个try & except块来处理这个错误。

我想知道python中是否有一种机制可以将insertToTable()连接到另一个函数，比如checkConnectivity()。当调用insertToTable()时，首先执行checkConnectivity()，然后继续执行insertToTable()。

Answer 1

您可以创建一个调用checkConnectivity函数的装饰器。示例：

def checkConnectivity():
    print 'Checking connection'

def CheckConnectionDecorator(fn):
    def new(*args, **kwargs):
        checkConnectivity()
        return fn(*args, **kwargs)
    return new

@CheckConnectionDecorator
def insertToTable():
    print 'Insertion'

@CheckConnectionDecorator
def anotherQueryFunction():
    print 'Another query'