python - 如何在 cron 作业中进行 flask API 调用时关闭数据库连接？

Question

cronjob 使用管理 session 使用不同的 API 调用连接到不同的数据库。每次通话后，我想确保连接已关闭。在正常的 API 调用的情况下，我可以在每次 API 调用后使用 before_request 来打开和 after_request 装饰器来关闭，但在 cronjob 中如何调用这些函数。列出其中一个数据库的队列的 cron 作业中的代码如下:

def get_queue_list(session, access_token):
    queue_list = []
    headers={'Authorization': "Bearer " + access_token}
    r = session.get(queue_api, headers=headers)
    queue_list = json.loads(r.content)['queue_list']
    r.connection.close()

    return queue_list

在上面的代码片段中，语句 r.connection.close() 是否确保数据库连接关闭？我如何检查它？我在同一台服务器上安装了很多。这个 cronjob 为不同数据库的不同安装调用不同的 API 来定期处理。请知会

主函数调用:

if __name__ == "__main__":
    print __doc__
    # base_url is read from a csv file
    queue_api = base_url + '/rt/api/v1.0/queues'

Answer 1

如果您已经建立了与数据库的连接，我相信您需要做的就是r.close()。我尝试使用游标和连接器函数调用查询(即 select * from table)。如果出现错误，我知道它没有连接。

我还建议您在调用数据库连接函数后关闭连接。

def get_queue_list(session, access_token):
    queue_list = []
    headers={'Authorization': "Bearer " + access_token}
    r = session.get(queue_api, headers=headers)
    queue_list = json.loads(r.content)['queue_list']

return queue_list, r

if __name__ == "__main__":
    print __doc__
    # base_url is read from a csv file
    queue_api = base_url + '/rt/api/v1.0/queues'
    r.close()