python - 在重复输入时退出Python MySQLdb

Question

我想在重复输入时退出程序，这是我所做的没有成功的事情:

我想处理该错误，但不知道如何并且尚未找到有关它的信息。

def connection():

    global servername, username, password, dbname

    con = mdb.connect(servername, username, password, dbname)

    return con;


def insert_vulnerabilities (CVE,Description,P_Date,U_Date,Score,Type):



    con = connection()



    with con:

        cur = con.cursor()


        try:
            cur.execute("INSERT INTO Vulnerabilities_test(CVE,Description,P_Date,U_Date,Score,Type) VALUES(%s,%s,%s,%s,%s,%s)",(CVE,Description,P_Date,U_Date,Score,Type))
            con.commit()
        except:

            sys.exit(0)

先感谢您。

Answer 1

import sys
...
...
...

    try:
        write_cursor.execute(write_operation)
        db_writer.commit()
    except IntegrityError as ie:
        print("Duplicate key found. Exiting.")
        sys.exit(1)