我正在创建一个函数，从数据库表User中获取一个特定的用户，如果它不存在，就创建它：

def get_user_id(session, username):
    query = session.query(User).filter(User.username == username)
    try:
        user = query.one()
    except NoResultFound as e:
        user = User(username=username)
        try:
            session.add(user)
            session.flush()
        except IntegrityError as e:
            session.rollback() # ! This will rollback all changes made to the session.
            user = query.one()
    return user.id_user

字符串
如果在事务处理期间创建了相同的用户名，则用户创建部分可能会失败，因为它具有唯一约束。在这种情况下，我将查询现有的id。但是，我需要添加回滚，以便会话在提交时不会引发错误。
我希望能够在同一个会话中多次调用该函数，而无需在调用之间提交。但是，回滚将撤消所有以前添加的用户或在调用该函数之前所做的其他修改。有没有一种方法可以只回滚在当前函数调用中添加的用户？