SQLAlchemy 是否有相当于 Django 的 get_or_create 的功能

使用 SQLAlchemy 实现类似 Django 中的 get_or_create 功能

python
from sqlalchemy import create_engine, Column, String, Integer
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)
    username = Column(String, unique=True)
# 创建数据库引擎和表
engine = create_engine('sqlite:///:memory:')
Base.metadata.create_all(engine)
# 创建一个会话
Session = sessionmaker(bind=engine)
session = Session()
# 定义 get_or_create 函数
def get_or_create(session, model, defaults=None, kwargs):
    instance = session.query(model).filter_by(kwargs).first()
    if instance:
        return instance, False
    else:
        params = {k: v for k, v in kwargs.items() if not isinstance(v, Column)}
        params.update(defaults or {})
        instance = model(params)
        session.add(instance)
        session.commit()
        return instance, True
# 使用 get_or_create 函数
user, created = get_or_create(session, User, username='john_doe')
# 打印结果
print(f"User: {user.username}, Created: {created}")