How to get new primary key value after flush session in sqlalchemy

sqlalchemy_utils.py

from sqlalchemy.orm.base import instance_state


def get_instance_identity(instance, single=True):
    """获取实例主键值。"""
    identity = instance_state(instance).identity
    if single:
        return identity[0]

    return identity