From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Mon, 21 Mar 2022 21:47:34 +0000 (-0400)
Subject: use begin() for settting up first PK
X-Git-Tag: rel_2_0_0b1~412
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=257f67b180cadf4ca0df0d1facf27d55c98df676;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

use begin() for settting up first PK

Change-Id: I227bbb46fbcbae1f60d3f5bb4dd2b9f41ca3dd0c
---

diff --git a/examples/sharding/separate_databases.py b/examples/sharding/separate_databases.py
index accbfd79b4..9818656c3c 100644
--- a/examples/sharding/separate_databases.py
+++ b/examples/sharding/separate_databases.py
@@ -56,7 +56,7 @@ ids = Table("ids", Base.metadata, Column("nextid", Integer, nullable=False))
 
 def id_generator(ctx):
     # in reality, might want to use a separate transaction for this.
-    with db1.connect() as conn:
+    with db1.begin() as conn:
         nextid = conn.scalar(ids.select().with_for_update())
         conn.execute(ids.update().values({ids.c.nextid: ids.c.nextid + 1}))
     return nextid
diff --git a/examples/sharding/separate_tables.py b/examples/sharding/separate_tables.py
index 4314e22337..0f6e2ffd83 100644
--- a/examples/sharding/separate_tables.py
+++ b/examples/sharding/separate_tables.py
@@ -70,7 +70,7 @@ ids = Table("ids", Base.metadata, Column("nextid", Integer, nullable=False))
 
 def id_generator(ctx):
     # in reality, might want to use a separate transaction for this.
-    with engine.connect() as conn:
+    with engine.begin() as conn:
         nextid = conn.scalar(ids.select().with_for_update())
         conn.execute(ids.update().values({ids.c.nextid: ids.c.nextid + 1}))
     return nextid