python - 测试 Postgres-DB 的连接

Question

如果软件连接到特定的 Postgre-DB，我想在 GUI 上放一个按钮。我写了一个小测试函数:如果它可以连接到数据库，则返回 True，否则返回 False。

代码有效，但有一个问题:如果没有连接(我只是拔掉互联网电缆，没有其他变化)，它只是需要太多时间。如果没有连接，你能帮我使代码更快吗？

这是我的简单测试函数:

import psycopg2

def postgres_test():

    try:
        conn = psycopg2.connect("dbname='mydb' user='myuser' host='my_ip' password='mypassword'")
        conn.close()
        return True
    except:
        return False

Answer 1

感谢您的评论。是的，它与超时有关。

这是我更快的代码:

import psycopg2

def postgres_test():

    try:
        conn = psycopg2.connect("dbname='mydb' user='myuser' host='my_ip' password='mypassword' connect_timeout=1 ")
        conn.close()
        return True
    except:
        return False