From c21224269034cbc90244c8e30191df714e2ea639 Mon Sep 17 00:00:00 2001 From: harryghgim Date: Thu, 20 Mar 2025 01:16:47 +0900 Subject: [PATCH] sqlalchemy 2.0 compatible syntax --- src/allocation/adapters/orm.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/allocation/adapters/orm.py b/src/allocation/adapters/orm.py index a87068a5..95c0a285 100644 --- a/src/allocation/adapters/orm.py +++ b/src/allocation/adapters/orm.py @@ -8,7 +8,7 @@ ForeignKey, event, ) -from sqlalchemy.orm import mapper, relationship +from sqlalchemy.orm import relationship, registry from allocation.domain import model @@ -57,10 +57,13 @@ Column("batchref", String(255)), ) +# Create a registry +mapper_registry = registry() + def start_mappers(): - lines_mapper = mapper(model.OrderLine, order_lines) - batches_mapper = mapper( + lines_mapper = mapper_registry.map_imperatively(model.OrderLine, order_lines) + batches_mapper = mapper_registry.map_imperatively( model.Batch, batches, properties={ @@ -71,7 +74,7 @@ def start_mappers(): ) }, ) - mapper( + mapper_registry.map_imperatively( model.Product, products, properties={"batches": relationship(batches_mapper)},