pgsqlasync2fast-fastapi 0.4.0__py3-none-any.whl → 0.4.1__py3-none-any.whl

pgsqlasync2fast_fastapi/__init__.py

@@ -52,6 +52,8 @@ from pgsqlasync2fast_fastapi.seeder import (
     seed_all,
     # Shared idempotent insert-if-missing primitive
     insert_if_missing,
+    # Shared PostgreSQL sequence re-sync primitive
+    sync_table_sequence,
 )
 
 __all__ = [
@@ -90,4 +92,6 @@ __all__ = [
     "seed_all",
     # Shared idempotent insert-if-missing primitive
     "insert_if_missing",
+    # Shared PostgreSQL sequence re-sync primitive
+    "sync_table_sequence",
 ]

pgsqlasync2fast_fastapi/__version__.py

@@ -1 +1 @@
-__version__ = "0.4.0"
+__version__ = "0.4.1"

pgsqlasync2fast_fastapi/seeder.py

@@ -593,6 +593,60 @@ def _resolve_load_order(manifest: dict[str, Any]) -> list[str]:
 # ============================================================================
 
 
+async def sync_table_sequence(
+    session: Any,
+    model_class: type,
+) -> None:
+    """
+    Re-sync a PostgreSQL table identity sequence to its current MAX(id).
+
+    The seeder inserts rows with explicit ``id`` values straight from the
+    manifest JSON. PostgreSQL's plain ``serial``/``sequence``-backed columns do
+    NOT advance the sequence when a row is inserted with an explicit id, so a
+    later sequence-backed insert (e.g. ``insert_if_missing``) can generate an
+    id that collides with an existing explicit-id row (UniqueViolation).
+
+    This issues one ``setval`` to ``MAX(id)`` after the explicit-id inserts so
+    the next auto-generated id continues past the seeded rows. It only runs
+    against PostgreSQL — SQLite assigns ``INTEGER PRIMARY KEY`` as max+1
+    automatically and has no incompatible sequence catalog, so it is a no-op
+    there.
+
+    Args:
+        session: SQLModel AsyncSession bound to the target DB.
+        model_class: The SQLModel/SQLAlchemy table class to re-sync.
+    """
+    from sqlalchemy import text
+
+    try:
+        bind = getattr(session, "bind", None)
+        if bind is None:
+            return
+        dialect = getattr(bind.dialect, "name", None)
+        if dialect != "postgresql":
+            return
+
+        table = model_class.__tablename__
+        seq = f"{table}_id_seq"
+        async with bind.begin() as conn:
+            max_id = await conn.scalar(
+                text(f'SELECT COALESCE(MAX(id), 0) FROM "{table}"')
+            )
+            if max_id:
+                await conn.execute(
+                    text("SELECT setval(:seq, :val)"),
+                    {"seq": seq, "val": max_id},
+                )
+                logger.info(
+                    f"Synced sequence '{seq}' to MAX(id)={max_id} "
+                    f"(table '{table}')"
+                )
+    except Exception as exc:  # pragma: no cover - defensive, non-fatal
+        logger.warning(
+            f"Could not re-sync sequence for '{model_class.__name__}': {exc}"
+        )
+
+
 async def _seed_table_idempotent(
     session: Any,
     table_name: str,
@@ -658,6 +712,9 @@ async def _seed_table_idempotent(
             logger.error(f"Failed to insert row in table '{table_name}': {e}")
             raise
 
+    if model_class is not None:
+        await sync_table_sequence(session, model_class)
+
     return rows_inserted, rows_skipped
 
 
@@ -727,6 +784,8 @@ async def _seed_table_idempotent_generic(
             logger.error(f"Failed to insert row in table '{table_name}': {e}")
             raise
 
+    await sync_table_sequence(session, model_class)
+
     return rows_inserted, rows_skipped
 
 

{pgsqlasync2fast_fastapi-0.4.0.dist-info → pgsqlasync2fast_fastapi-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pgsqlasync2fast-fastapi
-Version: 0.4.0
+Version: 0.4.1
 Summary: Simple and fast PostgreSQL async module for FastAPI with multi-database support
 Author-email: Angel Daniel Sanchez Castillo <angeldaniel.sanchezcastillo@gmail.com>
 License: MIT License

pgsqlasync2fast_fastapi-0.4.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+pgsqlasync2fast_fastapi/__init__.py,sha256=jc45a0ClPL76d2BTkBIM6DvRzJj3O9mv1sgGEn17dmA,2474
+pgsqlasync2fast_fastapi/__version__.py,sha256=pMtTmSUht-XtbR_7Doz6bsQqopJJd8rZ8I8zy2HwwoA,22
+pgsqlasync2fast_fastapi/connection.py,sha256=vyR3SLpLTENYGKWkEoou0ud5IwUOJ9ksfkysUW5tz-I,6168
+pgsqlasync2fast_fastapi/database.py,sha256=t6i8V4Bbec0sPoWYsBPScSqQpde1tVkqGx0CBRnAupk,8856
+pgsqlasync2fast_fastapi/dependencies.py,sha256=jgX14S6h6i4Uk6edMDGh-wiDpxTlB16qvYMlIm1QofQ,4444
+pgsqlasync2fast_fastapi/seeder.py,sha256=b_7fKP19Qh08NaTyY4kA7HL48dqZIE-Re5I-ki3M5a4,33284
+pgsqlasync2fast_fastapi/settings.py,sha256=9SAmEVtotKJYlbTjI4qkKBjEOghTTtaICAFBBrR-eHk,6678
+pgsqlasync2fast_fastapi/skills/SKILL.md,sha256=Yqge3ydNnZXO5hN5Ac4aynCG0hn-WYPCy_nkzToUDsM,6787
+pgsqlasync2fast_fastapi-0.4.1.dist-info/licenses/LICENSE,sha256=CkISX1hNEwxxrPTOXet3IYMEH28Bn7SoUyKniRjg68I,1086
+pgsqlasync2fast_fastapi-0.4.1.dist-info/METADATA,sha256=_giXkOTGxUOtf__J-zPoRXtzyznR0WedKi-42Bgi4Ws,10105
+pgsqlasync2fast_fastapi-0.4.1.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+pgsqlasync2fast_fastapi-0.4.1.dist-info/top_level.txt,sha256=7pllDm9nlFGWKJDtG5zukxqdADqjZddU-GzaQJPLMVc,24
+pgsqlasync2fast_fastapi-0.4.1.dist-info/RECORD,,

pgsqlasync2fast_fastapi-0.4.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-pgsqlasync2fast_fastapi/__init__.py,sha256=rfPB2UJsmImEeCR-FfZ3Gg0cfuVLc4bymp-A1gi5Do4,2320
-pgsqlasync2fast_fastapi/__version__.py,sha256=42STGor_9nKYXumfeV5tiyD_M8VdcddX7CEexmibPBk,22
-pgsqlasync2fast_fastapi/connection.py,sha256=vyR3SLpLTENYGKWkEoou0ud5IwUOJ9ksfkysUW5tz-I,6168
-pgsqlasync2fast_fastapi/database.py,sha256=t6i8V4Bbec0sPoWYsBPScSqQpde1tVkqGx0CBRnAupk,8856
-pgsqlasync2fast_fastapi/dependencies.py,sha256=jgX14S6h6i4Uk6edMDGh-wiDpxTlB16qvYMlIm1QofQ,4444
-pgsqlasync2fast_fastapi/seeder.py,sha256=Pl9xEkgqOcOlZgFAHuTGgPUSCZ-Mt-M7wvYJZ6VLM98,31122
-pgsqlasync2fast_fastapi/settings.py,sha256=9SAmEVtotKJYlbTjI4qkKBjEOghTTtaICAFBBrR-eHk,6678
-pgsqlasync2fast_fastapi/skills/SKILL.md,sha256=Yqge3ydNnZXO5hN5Ac4aynCG0hn-WYPCy_nkzToUDsM,6787
-pgsqlasync2fast_fastapi-0.4.0.dist-info/licenses/LICENSE,sha256=CkISX1hNEwxxrPTOXet3IYMEH28Bn7SoUyKniRjg68I,1086
-pgsqlasync2fast_fastapi-0.4.0.dist-info/METADATA,sha256=rA3ZltgZR83lOfPOidK2yKxpOZxX1KbGL50cP5LqMys,10105
-pgsqlasync2fast_fastapi-0.4.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
-pgsqlasync2fast_fastapi-0.4.0.dist-info/top_level.txt,sha256=7pllDm9nlFGWKJDtG5zukxqdADqjZddU-GzaQJPLMVc,24
-pgsqlasync2fast_fastapi-0.4.0.dist-info/RECORD,,