SQLAlchemy -大容量插入忽略重复/唯一

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from sqlalchemy.dialects.postgresql import insert

# 定义数据表模型
Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    email = Column(String, unique=True)

# 创建数据库连接
engine = create_engine('postgresql://username:password@localhost/mydatabase')

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

# 插入数据
data = [
    {'name': 'John', 'email': 'john@example.com'},
    {'name': 'Alice', 'email': 'alice@example.com'},
    {'name': 'Bob', 'email': 'bob@example.com'},
    {'name': 'John', 'email': 'john@example.com'}  # 重复数据
]

stmt = insert(User).values(data)
stmt = stmt.on_conflict_do_nothing(index_elements=[User.email])  # 忽略email字段的重复
session.execute(stmt)
session.commit()