Question

我正在尝试进行一些数据处理，然后一次插入，表中有一些组合键，可用来检查该ID的记录是否存在，然后它应该更新记录而不是创建记录。

在进行数据处理时，样本数据中可能存在相同的ID多次，并且在处理过程中将找不到与db中的ID匹配的记录，因此它将尝试每次创建记录。

有什么方法可以检查高速缓存中是否有复合键匹配？

Answer 1

您可以使用实体实例的get方法通过主键值的组合来查找对象。

get方法同时搜索db_session缓存和数据库，如果未找到对象，则返回None：

from pony import orm

db = orm.Database('sqlite', ':memory:')

class Point(db.Entity):
    x = orm.Required(int)
    y = orm.Required(int)
    description = orm.Optional(str)
    orm.PrimaryKey(x, y)

db.generate_mapping(create_tables=True)

points = [(1, 2, 'foo'), (3, 4, 'bar'), (1, 2, 'baz'), (5, 6, 'qux')]

with orm.db_session:
    for a, b, s in points:
        point = Point.get(x=a, y=b)
        if point is None:
            point = Point(x=a, y=b, description=s)

get方法也可用于辅助composite keys。

您还可以使用square brackets搜索实体实例并捕获异常：

with orm.db_session:
    try:
        point = Point[10, 20]
    except orm.ObjectNotFound:
        point = Point(x=10, y=20)

更新：有关IdentityMap模式使用的说明

PonyORM使用IdentityMap设计模式。这意味着在db_session内加载或创建的同一类的所有对象均由其主键值索引。同一db_session中不可能有具有相同主键的同一类的多个对象。如果使用相同的主键值多次调用get方法，它将返回相同的实例：

with db_session:
    a = Point.get(x=10, y=20)
    b = Point.get(x=10, y=20)
    assert a is b  # the same object!

当您创建对象然后尝试使用相同的主键尝试get对象时，情况也是如此：Pony将返回您刚刚创建的对象：

with db_session:
    a = Point.get(x=10, y=20)
    if a is None:
        a = Point(x=10, y=20)

    # later, in the same db_session:
    b = Point.get(x=10, y=20)
    assert a is b  # the same object, not saved yet

另一方面，如果尝试使用相同的主键创建两个不同的对象，则会收到错误消息：

with db_session:
    a = Point(x=10, y=20)
    b = Point(x=10, y=20)

Traceback (most recent call last):
    ...
pony.orm.core.CacheIndexError: Cannot create Point: instance with primary key 10, 20 already exists

此外，如果您尝试创建数据库中已经存在的对象而不进行检查，则会收到错误消息：

with db_session:
    a = Point(x=30, y=40)

with db_session:
    b = Point(x=30, y=40)

Traceback (most recent call last):
    ...
pony.orm.core.TransactionIntegrityError: Object Point[30, 40] cannot be stored in the database. IntegrityError: UNIQUE constraint failed: Point.x, Point.y

为了避免此错误，您可以在创建新对象之前使用get检查对象是否存在。

如何检查小马orm缓存中是否存在对象？

1 个答案:

更新：有关IdentityMap模式使用的说明