Тестирование подключения Postgres-DB

Question

Я хотел бы поместить кнопку в графический интерфейс, если программное обеспечение подключается к определенной Postgre-DB. Я написал небольшую тестовую функцию: если она может подключиться к БД, она возвращает 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