svc-infra 0.1.182__py3-none-any.whl → 0.1.184__py3-none-any.whl

svc_infra/api/fastapi/db/uniq.py

@@ -0,0 +1,82 @@
+from __future__ import annotations
+
+from typing import Iterable, Optional, Dict, Any, Callable
+from sqlalchemy import func, and_
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from .repository import Repository
+from svc_infra.db.uniq import ColumnSpec, _as_tuple
+
+def make_uniqueness_hooks(
+        *,
+        model: type,
+        repo: Repository,
+        unique_cs: Iterable[ColumnSpec] = (),
+        unique_ci: Iterable[ColumnSpec] = (),
+        tenant_field: Optional[str] = None,
+        duplicate_message: str = "Duplicate resource violates uniqueness policy.",
+) -> tuple[
+    Callable[[Dict[str, Any]], Any],  # async pre_create(data) -> data
+    Callable[[Dict[str, Any]], Any],  # async pre_update(data) -> data
+]:
+    """Return (pre_create, pre_update) hooks that check duplicates *before* insert/update.
+
+    Usage with ServiceWithHooks:
+        pre_c, pre_u = make_uniqueness_hooks(model=User, repo=repo, unique_ci=["email"], tenant_field="tenant_id")
+        svc = ServiceWithHooks(repo, pre_create=pre_c, pre_update=pre_u)
+    """
+    cs_specs = [_as_tuple(s) for s in unique_cs]
+    ci_specs = [_as_tuple(s) for s in unique_ci]
+
+    def _build_wheres(data: Dict[str, Any]):
+        wheres = []
+
+        def col(name: str):
+            return getattr(model, name)
+
+        tenant_val = data.get(tenant_field) if tenant_field else None
+
+        # case-sensitive groups
+        for spec in cs_specs:
+            if not all(k in data for k in spec):
+                continue
+            parts = []
+            if tenant_field:
+                parts.append(col(tenant_field).is_(None) if tenant_val is None else col(tenant_field) == tenant_val)
+            for k in spec:
+                parts.append(col(k) == data[k])
+            wheres.append(and_(*parts))
+
+        # case-insensitive groups
+        for spec in ci_specs:
+            if not all(k in data for k in spec):
+                continue
+            parts = []
+            if tenant_field:
+                parts.append(col(tenant_field).is_(None) if tenant_val is None else col(tenant_field) == tenant_val)
+            for k in spec:
+                # compare lower(db_col) == lower(<value>) safely
+                parts.append(func.lower(col(k)) == func.lower(func.cast(data[k], col(k).type)))
+            wheres.append(and_(*parts))
+
+        return wheres
+
+    async def _deferred_check(session: AsyncSession, data: Dict[str, Any]):
+        wheres = _build_wheres(data)
+        for where in wheres:
+            if await repo.exists(session, where=[where]):
+                from fastapi import HTTPException
+                raise HTTPException(status_code=409, detail=duplicate_message)
+
+    async def _pre_create(data: Dict[str, Any]) -> Dict[str, Any]:
+        # store a coroutine to be awaited when a session is available in Service.create
+        data = dict(data)
+        data.setdefault("__uniq_check__", _deferred_check)
+        return data
+
+    async def _pre_update(data: Dict[str, Any]) -> Dict[str, Any]:
+        data = dict(data)
+        data.setdefault("__uniq_check__", _deferred_check)
+        return data
+
+    return _pre_create, _pre_update

svc_infra/auth/user_default.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict
 from fastapi import HTTPException
 from fastapi_users.password import PasswordHelper
-from sqlalchemy import func, and_, or_
+from sqlalchemy import func
 
 from svc_infra.api.fastapi.db.service_hooks import ServiceWithHooks
 from svc_infra.api.fastapi.db.repository import Repository

svc_infra/db/templates/models_schemas/auth/models.py.tmpl

@@ -12,6 +12,7 @@ from sqlalchemy.ext.mutable import MutableDict, MutableList
 
 from svc_infra.db.base import ModelBase
 from svc_infra.auth.user_default import _pwd
+from svc_infra.db.uniq import make_unique_indexes
 
 
 class User(ModelBase):
@@ -60,23 +61,12 @@ class User(ModelBase):
         return f"<User id={self.id} email={self.email!r}>"
 
 
-# ---------- Uniqueness: case-insensitive email per-tenant ----------
-# Because tenant_id is Optional (nullable), use two partial unique indexes:
-# - When tenant_id IS NULL: email (lowercased) must be globally unique
-# - When tenant_id IS NOT NULL: (tenant_id, lower(email)) must be unique
-Index(
-    "uq_users_email_lower_global",
-    func.lower(User.email),
-    unique=True,
-    postgresql_where=User.tenant_id.is_(None),
-)
-Index(
-    "uq_users_email_lower_per_tenant",
-    func.lower(User.email),
-    User.tenant_id,
-    unique=True,
-    postgresql_where=User.tenant_id.isnot(None),
-)
-
-# Optional helper index for case-insensitive lookups (non-unique)
-Index("ix_users_email_lower", func.lower(User.email))
+# Unique indexes, including case-insensitive and/or tenant-scoped.
+for _ix in make_unique_indexes(
+    User,
+    unique_ci=["email"],            # case-insensitive unique email
+    tenant_field="tenant_id",       # scoped by tenant if provided
+):
+    # Simply iterating keeps them registered with the Table metadata.
+    # (No further action needed if you use Alembic autogenerate or metadata.create_all.)
+    pass

svc_infra/db/templates/models_schemas/entity/models.py.tmpl

@@ -1,29 +1,52 @@
+# models/${table_name}.py
 from __future__ import annotations
 from datetime import datetime
 from typing import Optional
 import uuid
 
-from sqlalchemy import String, Boolean, DateTime, JSON, Text, func, UniqueConstraint, Index
+from sqlalchemy import String, Boolean, DateTime, JSON, Text, func
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.orm import Mapped, mapped_column
 from sqlalchemy.ext.mutable import MutableDict
 
 from svc_infra.db.base import ModelBase
+from svc_infra.db.uniq import make_unique_indexes
 
 
 class ${Entity}(ModelBase):
     __tablename__ = "${table_name}"
 
+    # identity
     id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
-    name: Mapped[str] = mapped_column(String(255), nullable=False)
+
+    # core fields
+    name: Mapped[str] = mapped_column(String(255), nullable=False, index=True)
     description: Mapped[Optional[str]] = mapped_column(Text)
 
-${tenant_field}${soft_delete_field}    extra: Mapped[dict] = mapped_column(MutableDict.as_mutable(JSON), default=dict)
+${tenant_field}\
+${soft_delete_field}\
+    # misc (avoid attr name "metadata" clash)
+    extra: Mapped[dict] = mapped_column(MutableDict.as_mutable(JSON), default=dict)
 
-    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now(), nullable=False)
-    updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now(), nullable=False)
+    # auditing (DB-side timestamps)
+    created_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), server_default=func.now(), nullable=False
+    )
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), server_default=func.now(), onupdate=func.now(), nullable=False
+    )
 
-${constraints}    def __repr__(self) -> str:
+    def __repr__(self) -> str:
         return f"<${Entity} id={self.id} name={self.name!r}>"
 
-${indexes}
+
+# --- Uniqueness policy --------------------------------------------------------
+# Register functional unique indexes (case-insensitive on "name"),
+# optionally scoped by tenant if present.
+for _ix in make_unique_indexes(
+    ${Entity},
+    unique_ci=["name"]${tenant_arg}
+):
+    # Iteration is enough to attach them to the Table metadata
+    # (Alembic autogenerate / metadata.create_all will pick them up)
+    pass

svc_infra/db/uniq.py

@@ -0,0 +1,77 @@
+from __future__ import annotations
+from typing import Iterable, Sequence, Tuple, Union, List, Optional
+from sqlalchemy import Index, func
+
+ColumnSpec = Union[str, Sequence[str]]
+
+def _as_tuple(spec: ColumnSpec) -> Tuple[str, ...]:
+    return (spec,) if isinstance(spec, str) else tuple(spec)
+
+def make_unique_indexes(
+        model: type,
+        *,
+        unique_cs: Iterable[ColumnSpec] = (),
+        unique_ci: Iterable[ColumnSpec] = (),
+        tenant_field: Optional[str] = None,
+        name_prefix: str = "uq",
+) -> List[Index]:
+    """Return SQLAlchemy Index objects that enforce uniqueness.
+
+    - unique_cs: case-sensitive unique specs
+    - unique_ci: case-insensitive unique specs (lower(column))
+    - tenant_field: if provided, create two partial unique indexes:
+        * tenant IS NULL (global bucket)
+        * tenant IS NOT NULL (scoped per-tenant)
+
+    Declare right after your model class; Alembic or metadata.create_all will pick them up.
+    """
+    idxs: List[Index] = []
+
+    def _col(name: str):
+        return getattr(model, name)
+
+    def _to_sa_cols(spec: Tuple[str, ...], *, ci: bool):
+        cols = []
+        for cname in spec:
+            c = _col(cname)
+            cols.append(func.lower(c) if ci else c)
+        return tuple(cols)
+
+    tenant_col = _col(tenant_field) if tenant_field else None
+
+    def _name(ci: bool, spec: Tuple[str, ...], null_bucket: Optional[str] = None):
+        parts = [name_prefix, model.__tablename__]
+        if tenant_field:
+            parts.append(tenant_field)
+        if null_bucket:
+            parts.append(null_bucket)
+        parts.append("ci" if ci else "cs")
+        parts.extend(spec)
+        return "_".join(parts)
+
+    for ci, spec_list in ((False, unique_cs), (True, unique_ci)):
+        for spec in spec_list:
+            spec_t = _as_tuple(spec)
+            cols = _to_sa_cols(spec_t, ci=ci)
+
+            if tenant_col is None:
+                idxs.append(Index(_name(ci, spec_t), *cols, unique=True))
+            else:
+                idxs.append(
+                    Index(
+                        _name(ci, spec_t, "null"),
+                        *cols,
+                        unique=True,
+                        postgresql_where=tenant_col.is_(None),
+                    )
+                )
+                idxs.append(
+                    Index(
+                        _name(ci, spec_t, "notnull"),
+                        tenant_col,
+                        *cols,
+                        unique=True,
+                        postgresql_where=tenant_col.isnot(None),
+                    )
+                )
+    return idxs

{svc_infra-0.1.182.dist-info → svc_infra-0.1.184.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: svc-infra
-Version: 0.1.182
+Version: 0.1.184
 Summary: Infrastructure for building and deploying prod-ready services
 License: MIT
 Keywords: fastapi,sqlalchemy,alembic,auth,infra,async,pydantic

{svc_infra-0.1.182.dist-info → svc_infra-0.1.184.dist-info}/RECORD

@@ -13,6 +13,7 @@ svc_infra/api/fastapi/db/resource.py,sha256=h-kCjp0_Sw_lGthizkEaneKyIZZXX3SCs6LF
 svc_infra/api/fastapi/db/service.py,sha256=w7WFzETndKuk-2wkOdjmGbznY8lJTzUP3781Dg42alE,2033
 svc_infra/api/fastapi/db/service_hooks.py,sha256=KZzkKL-2y7hrPx4bfcDdwItAqPBksyVFUjnx1hGJwxw,655
 svc_infra/api/fastapi/db/session.py,sha256=1ixBxRZLbdE8LXLP6R87u7pDzITZkZRYZ3Un8ugvJNY,1775
+svc_infra/api/fastapi/db/uniq.py,sha256=f96o8-Sz1nZIPa9lKX5bniot8Kv0xP1_zRns8R3p2KY,3132
 svc_infra/api/fastapi/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 svc_infra/api/fastapi/middleware/errors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 svc_infra/api/fastapi/middleware/errors/catchall.py,sha256=Kvu7yrHu6r8wlyEblB8sH4Mh-l8FYPmwAGJeSXugYOM,1713
@@ -31,7 +32,7 @@ svc_infra/auth/integration.py,sha256=L230xUw7vRM0AJIZwNzufRQrfNTPRdczp4BPd_NILeo
 svc_infra/auth/oauth_router.py,sha256=JY3VQ9_0oS_a2gGg418IDG2TbK79sQWcpA9KPiOAfjY,4918
 svc_infra/auth/providers.py,sha256=fiw0ouuGKtwcwMY0Zw7cEv-CXNaUYDnqo_NmiWiB8Lc,3185
 svc_infra/auth/settings.py,sha256=wVCLQnpA9M6zfiWyUpSdn9fo4q-Z4WpVyC3KvkG3HYg,1742
-svc_infra/auth/user_default.py,sha256=PFQSRTWnVo17jK43muF0ZxwSCtKM9VjVP10D_fpv6FE,2092
+svc_infra/auth/user_default.py,sha256=1GKxiJNUFKKghtGiIQKdC_rD1cs4vVye0zUTSPMrrDg,2081
 svc_infra/auth/users.py,sha256=4rlTCELU-po7bF7u6z02Bd8pXLGcLI2Adj0VjA-gg5E,2098
 svc_infra/cli/__init__.py,sha256=g47RSROuFW8LSsB6WFNSus5BnUl9z_DrPK9idYo3O6M,401
 svc_infra/cli/cmds/__init__.py,sha256=263YKSg73Ik-SQeilyGePdc663BYi4RfBrc0wMFxeoU,212
@@ -49,15 +50,16 @@ svc_infra/db/core.py,sha256=3Efm88fajfKJa8gDiiTVkhx2ci_UbrLiR4B8gMQYuwU,10745
 svc_infra/db/scaffold.py,sha256=NBz8BJR8MPvntCGv5HKoqNNR41SMnCkb-DlCCIsu0fo,9878
 svc_infra/db/templates/__init__.py,sha256=3PRa4v05RGyjX6LZXSOf-eMRir8vij7tET95limMips,45
 svc_infra/db/templates/models_schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-svc_infra/db/templates/models_schemas/auth/models.py.tmpl,sha256=fbsI1XNs840fashySKGSoGsojo0yOO8M4yZj7E9tUaE,3020
+svc_infra/db/templates/models_schemas/auth/models.py.tmpl,sha256=CR5TqE_S6xaVgotLF3jSFuigxL8V_aoMwubttnEx2js,2776
 svc_infra/db/templates/models_schemas/auth/schemas.py.tmpl,sha256=w79j8GeRPpywg9OVBmEuoksPVZLf9hUfEcRS4hBbiJQ,1717
-svc_infra/db/templates/models_schemas/entity/models.py.tmpl,sha256=dZk7a3WwpddSylPW81ARTyjbqJ-4rduX8-KAZ3Tlomo,1173
+svc_infra/db/templates/models_schemas/entity/models.py.tmpl,sha256=PXy8S9Zbd8JQWkSDMOdr-ubn_O1wNdZvFJdeOinoPV4,1777
 svc_infra/db/templates/models_schemas/entity/schemas.py.tmpl,sha256=zBhid1jFPwVN86OvkiIJkTFphN7LDcs_Hs9kxxeoNjE,1009
 svc_infra/db/templates/setup/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 svc_infra/db/templates/setup/alembic.ini.tmpl,sha256=QAQ_3-p8GnrpLoX2g-Fpi6aMfrS3m4ZMJ34OTfujnrI,780
 svc_infra/db/templates/setup/env_async.py.tmpl,sha256=05uDe_jiEkJOdXcSIgJovN3TxPG8wm8BjUoSdktsH_0,6224
 svc_infra/db/templates/setup/env_sync.py.tmpl,sha256=vlUKrhnQiBqAGDfbtx0D_CahF1Hjj8uTIQ5lNnx_l0Q,6273
 svc_infra/db/templates/setup/script.py.mako.tmpl,sha256=EgltN1ghYU7rH1zAujsLWLe1NThI-41x9fabByne5II,509
+svc_infra/db/uniq.py,sha256=Pre3tdgXHw6I7HU13xlJPcCe8ymSs8kAdEyAYRfvsto,2614
 svc_infra/db/utils.py,sha256=PxaZ6AesRlDyQk7tOzi9YaeF7Ge4zsZP83wqkKHu5-E,30838
 svc_infra/mcp/__init__.py,sha256=SiJ50LO7J6AneYswZe1kUX3dKXiECPpsjGPjT0SGtuc,1272
 svc_infra/mcp/svc_infra_mcp.py,sha256=VTU8esJ4FH9WNgxplufUWCVqGkxwscaLnBBtj-B5iCQ,1343
@@ -75,7 +77,7 @@ svc_infra/observability/templates/prometheus_rules.yml,sha256=sbVLm1h40FMkGSeWO4
 svc_infra/observability/tracing/__init__.py,sha256=TOs2yCicqBdo4OfOHTMmqeHsn7DBRu5EdvF2L5f31Y0,237
 svc_infra/observability/tracing/setup.py,sha256=21Ob276U4KZOs6M2o1O79wQXFHV0gY6YdMyjeqMrzMU,5042
 svc_infra/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-svc_infra-0.1.182.dist-info/METADATA,sha256=j7UJi0Sz4eQCAC7vrRGcyq4XA-nAi58n5_AN6qUNelY,4981
-svc_infra-0.1.182.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-svc_infra-0.1.182.dist-info/entry_points.txt,sha256=6x_nZOsjvn6hRZsMgZLgTasaCSKCgAjsGhACe_CiP0U,48
-svc_infra-0.1.182.dist-info/RECORD,,
+svc_infra-0.1.184.dist-info/METADATA,sha256=BCIRUX3U8Bva3-_m6s34h04-F3C2yamwc8BPtseu85M,4981
+svc_infra-0.1.184.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+svc_infra-0.1.184.dist-info/entry_points.txt,sha256=6x_nZOsjvn6hRZsMgZLgTasaCSKCgAjsGhACe_CiP0U,48
+svc_infra-0.1.184.dist-info/RECORD,,