upsert sqlalchemy

UPSERT，这是 SQLAlchemy 中一种独特而强大的数据操作方式。它允许我们在单个 SQL 查询中，既插入新数据，又更新现有数据，这在传统 SELECT 或 INSERT 语句中是难以实现的。UPSERT 的出现，大大提高了数据处理的效率，尤其在处理有唯一约束条件的数据时，它的优势更是明显。

创建自定义模型类

在使用 UPSERT 操作之前，我们首先需要创建一个自定义的模型类。这个类需要继承自 SQLAlchemy 的基类 Base。例如，假设我们要对一个用户表（users）进行 UPSERT 操作，我们可以创建一个继承自 Base 的用户模型类，包含 id、name 和 age 三个字段：

from sqlalchemy import Column, Integer, String, DateTime
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

Base = declarative_base()

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

重写 __repr__ 方法

为了实现 UPSERT 的功能，我们还需要重写模型的 __repr__ 方法。这个方法的作用是返回一个代表该对象的字符串，通常我们会使用这个字符串来查看或打印对象的信息。

例如，我们可以重写用户的 __repr__ 方法，使其实现 UPSERT 的功能：

def __repr__(self):
    return f"User(id={self.id}, name='{self.name}', age={self.age})"

创建会话工厂

在完成模型类的定义之后，我们需要创建一个会话工厂来执行实际的 UPSERT 操作。我们可以使用 create_engine 函数来创建一个数据库引擎，然后使用 sessionmaker 函数来创建一个会话工厂。

例如，我们可以创建一个会话工厂，并在获取到数据库会话后，使用 session 对象的 begin_transaction 和 commit 方法来进行 UPSERT 操作：

DATABASE_URL = "sqlite:///example.db"
engine = create_engine(DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
session = SessionLocal()

def upsert_user(name, age):
    user = User(name=name, age=age)
    session.add(user)
    session.commit()

使用 UPSERT

现在我们已经完成了模型类的定义、会话工厂的创建，就可以开始使用 UPSERT 操作了。

例如，我们可以调用 upsert_user 函数来插入或更新用户信息：

upsert_user("张三", 20)

上述代码会执行以下操作：先创建一个名为 User 的对象，其中 name 属性设为 "张三"，age 属性设为 20。然后，将这个对象添加到会话中，并提交事务。

如果 User 对象在会话中已经存在，那么 age 属性的值将被更新为 20。否则，一个新的 User 对象将被创建，并添加到会话中。

这就是 UPSERT 在 SQLAlchemy 中的基本用法。通过重写模型类的 __repr__ 方法和创建会话工厂，我们可以灵活地实现 UPSERT 操作，满足各种复杂的数据处理需求。