在python中實現try / except語句時無法連接數據庫錯誤

Question

我下面有一個python腳本，試圖建立與sql db的連接。 數據庫出現了一些問題，我希望腳本嘗試重新連接到數據庫，因此我添加了try / except。 但是現在我收到錯誤消息“無法連接到數據庫”。 如果我刪除try / except語句，腳本將完美運行。 如果有人可以幫助我使try / except語句正常工作，我將不勝感激！

在Try內使用連接更新了代碼

from __future__ import print_function

try:
    import psycopg2
except ImportError:
    raise ImportError('\n\033[33mpsycopg2 library missing. pip install psycopg2\033[1;m\n')
    sys.exit(1)

import re
import sys
import json
import pprint

outfilepath = "crtsh_output/crtsh_flat_file"

DB_HOST = 'crt.sh'
DB_NAME = 'certwatch'
DB_USER = 'guest'

#conn = psycopg2.connect("dbname={0} user={1} host={2}".format(DB_NAME, DB_USER, DB_HOST))
#cursor = conn.cursor()

def connect_to_db():
    filepath = 'forager.txt'
    conn = psycopg2.connect("dbname={0} user={1} host={2}".format(DB_NAME, DB_USER, DB_HOST))
    cursor = conn.cursor()
    with open(filepath) as fp:
        unique_domains = ''
        while True:
            try:
                for cnt, domain_name in enumerate(fp):
                    print("Line {}: {}".format(cnt, domain_name))
                    print(domain_name)
                    domain_name = domain_name.rstrip()

                    cursor.execute('''SELECT c.id, x509_commonName(c.certificate), x509_issuerName(c.certificate), x509_notBefore(c.certificate), x509_notAfter(c.certificate), x509_issuerName(c.certificate)$
                    FROM certificate c, certificate_identity ci WHERE
                    c.id= ci.certificate_id AND ci.name_type = 'dNSName' AND lower(ci.name_value) =
                    lower(%s) AND x509_notAfter(c.certificate) > statement_timestamp()''', (domain_name,))


                    unique_domains = cursor.fetchall()

                    pprint.pprint(unique_domains)

                    outfilepath = "crtsh1" + ".json"
                    with open(outfilepath, 'a') as outfile:
                            outfile.write(json.dumps(unique_domains, sort_keys=True, indent=4, default=str, ensure_ascii = False))
                    break
            except:
                pass
                #print("\n\033[1;31m[!] Unable to connect to the database\n\033[1;m")

if __name__ == "__main__":
    connect_to_db()

Answer 1

您需要做一些事情：

移動此代碼：

 conn = psycopg2.connect("dbname={0} user={1} host={2}".format(DB_NAME, DB_USER, DB_HOST)) 
 cursor = conn.cursor()

進入try ，以這種方式在死時循環時會重新建立連接。 像這樣：

    try:
        conn = psycopg2.connect("dbname={0} user={1} host={2}".format(DB_NAME, DB_USER, DB_HOST)) 
        cursor = conn.cursor()
        for cnt, domain_name in enumerate(fp):
            print("Line {}: {}".format(cnt, domain_name))
            print(domain_name)
            domain_name = domain_name.rstrip()

我會在連接之間停頓一下。 您的DBA可能具有在一段時間內檢測到太多連接的代碼，如下所示：

#at the top of your code
import time

#in your loop add:
time.sleep(1)