处理Python中的DuplicateKeyError()异常的方法

在Python中，DuplicateKeyError异常通常发生在使用数据库插入或更新操作时，当尝试插入或更新一个已存在的唯一键值时会抛出该异常。处理DuplicateKeyError异常的方法可以通过以下几种方式：

1. 使用try-except语句来捕获和处理异常：

try:
    # 插入或更新操作
except DuplicateKeyError as e:
    # 处理异常的逻辑
    print("已存在相同的键值")

2. 利用if语句提前查询数据库中是否存在相同的键值，避免触发异常：

def insert_record(record):
    # 检查数据库中是否已存在相同的键值
    if not is_duplicate(record):
        # 插入操作
    else:
        print("已存在相同的键值")

3. 使用数据库的特定方法来处理重复键的情况，如使用MongoDB的update_one()方法，并设置upsert参数为True，表示如果不存在则插入，如果存在则更新：

def insert_record(record):
    try:
        # 使用update_one()方法插入或更新操作
        collection.update_one({"_id": record["_id"]}, {"$set": record}, upsert=True)
    except DuplicateKeyError as e:
        print("已存在相同的键值")

4. 在数据库中设置唯一索引，可以在插入数据时避免重复键的情况，以MySQL为例：

CREATE TABLE table_name (
  id INT AUTO_INCREMENT PRIMARY KEY,
  name VARCHAR(100) UNIQUE,
  ...
);

使用try-except语句处理DuplicateKeyError异常的完整例子：

from pymongo import MongoClient
from pymongo.errors import DuplicateKeyError

def insert_record(record):
    client = MongoClient()
    db = client["my_database"]
    collection = db["my_collection"]

    try:
        # 插入操作
        collection.insert_one(record)
    except DuplicateKeyError as e:
        print("已存在相同的键值")

def main():
    record1 = {"_id": 1, "name": "Alice"}
    record2 = {"_id": 2, "name": "Bob"}
    record3 = {"_id": 1, "name": "Chris"}  # 重复键值

    insert_record(record1)
    insert_record(record2)
    insert_record(record3)

if __name__ == "__main__":
    main()

输出结果：

已存在相同的键值