nlbone 0.11.0__py3-none-any.whl → 0.11.3__py3-none-any.whl

nlbone/adapters/db/postgres/query_builder.py

@@ -1,7 +1,8 @@
 from typing import Any, Callable, List, Optional, Sequence, Type, Union
 
-from sqlalchemy import and_, asc, case, desc, literal, or_
+from sqlalchemy import Select, and_, asc, case, desc, func, literal, or_, select
 from sqlalchemy.dialects.postgresql import ENUM as PGEnum
+from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.orm import Query, Session, aliased
 from sqlalchemy.orm.attributes import InstrumentedAttribute
 from sqlalchemy.orm.interfaces import LoaderOption
@@ -384,6 +385,19 @@ def _apply_order(pagination: PaginateRequest, entity, query):
     return query
 
 
+def apply_pagination_async(pagination: PaginateRequest, entity, stmt: Select = None, limit: bool = True) -> Select:
+    if stmt is None:
+        stmt = select(entity)
+
+    stmt = _apply_filters(pagination, entity, stmt)
+    stmt = _apply_order(pagination, entity, stmt)
+
+    if limit:
+        stmt = stmt.limit(pagination.limit).offset(pagination.offset)
+
+    return stmt
+
+
 def apply_pagination(pagination: PaginateRequest, entity, session: Session, limit=True, query=None) -> Query:
     if not query:
         query = session.query(entity)
@@ -437,6 +451,43 @@ def _serialize_item(item: Any, output_cls: OutputType) -> Any:
         return item
 
 
+async def get_paginated_response_async(
+    pagination,
+    entity,
+    session: AsyncSession,
+    *,
+    with_count: bool = True,
+    output_cls: Optional[Type] = None,
+    eager_options: Optional[Sequence[LoaderOption]] = None,
+    query: Optional[Select] = None,
+) -> dict:
+    stmt = query if query is not None else select(entity)
+
+    if eager_options:
+        stmt = stmt.options(*eager_options)
+
+    filtered_stmt = apply_pagination_async(pagination, entity, stmt=stmt, limit=False)
+
+    total_count = None
+    if with_count:
+        subquery = filtered_stmt.subquery()
+        count_stmt = select(func.count()).select_from(subquery)
+        total_count = await session.scalar(count_stmt)
+
+    final_stmt = filtered_stmt.limit(pagination.limit).offset(pagination.offset)
+    result = await session.execute(final_stmt)
+    rows = result.scalars().all()
+
+    if output_cls is not None:
+        data = [output_cls.model_validate(r, from_attributes=True).model_dump() for r in rows]
+    else:
+        data = rows
+
+    return PaginateResponse(
+        total_count=total_count, data=data, limit=pagination.limit, offset=pagination.offset
+    ).to_dict()
+
+
 def get_paginated_response(
     pagination,
     entity,

nlbone/adapters/db/postgres/repository.py

@@ -236,6 +236,7 @@ class SQLAlchemyAsyncRepository(AsyncRepository, ABC):
         if await self.exists(getattr(entity, "id")):
             raise ValueError(f"Entity with id={getattr(entity, 'id')!r} already exists")
         self.session.add(entity)
+        await self.session.flush()
         if self.autocommit:
             await self.session.commit()
         return entity

nlbone/interfaces/api/additional_filed/assembler.py

@@ -2,11 +2,61 @@ import inspect
 from typing import Any, Dict
 
 from pydantic import BaseModel
+from sqlalchemy.ext.asyncio import AsyncSession
 
 from nlbone.container import Container
 from nlbone.interfaces.api.additional_filed.field_registry import FieldRule, ResourceRegistry
 
 
+async def assemble_response_async(
+    obj: Any,
+    reg: ResourceRegistry,
+    selected_rules: Dict[str, FieldRule],
+    session: AsyncSession,
+    base_schema: type[BaseModel] | None,
+    scope_map: dict[str, set[str]] = None,
+    **kwargs,
+) -> Dict[str, Any]:
+    """
+    Async version of assemble_response.
+    Awaits loaders if they are coroutines.
+    """
+    base = {f: getattr(obj, f, None) for f in reg.default_fields - set(reg.rules.keys())}
+    if base_schema:
+        base = base_schema.model_validate(base).model_dump()
+
+    ctx = {
+        "file_service": Container.afiles_service(),
+        "entity": obj,
+        "db": session,
+        "pricing_service": Container.async_pricing_service(),
+        **kwargs,
+    }
+
+    roots = {name.split(".", 1)[0] for name in selected_rules.keys()}
+
+    for root in roots:
+        rule = reg.rules.get(root)
+        if not rule:
+            continue
+
+        if rule.loader:
+            dependencies = ctx | {"scope": scope_map.get(root, {""})} if scope_map else ctx
+
+            result = inject_dependencies(rule.loader, dependencies=dependencies)
+
+            if inspect.iscoroutine(result):
+                value = await result
+            else:
+                value = result
+        else:
+            value = _get_nested_attr(obj, root)
+
+        _put_nested_key(base, root, value)
+
+    return base
+
+
 def assemble_response(
     obj: Any,
     reg: ResourceRegistry,

nlbone/interfaces/api/additional_filed/field_registry.py

@@ -1,10 +1,11 @@
 from dataclasses import dataclass, field
 from enum import Enum
-from typing import Any, Callable, Dict, Iterable, Optional, Set, Tuple, Type
+from typing import Any, Awaitable, Callable, Dict, Iterable, Optional, Set, Tuple, Type
 
 from pydantic import BaseModel
 
 PermissionChecker = Callable[[Any, str], bool]
+AsyncPermissionChecker = Callable[[str], Awaitable[bool]]
 Loader = Callable
 
 

nlbone/interfaces/api/additional_filed/resolver.py

@@ -2,11 +2,15 @@ from collections import defaultdict
 from typing import Dict, List, Set, Tuple
 
 from nlbone.interfaces.api.additional_filed.field_registry import (
+    AsyncPermissionChecker,
     DefaultsMergeMode,
     FieldRule,
     PermissionChecker,
     ResourceRegistry,
 )
+from nlbone.interfaces.api.dependencies.async_auth import (
+    client_or_user_has_access_func as async_client_or_user_has_access_func,
+)
 from nlbone.interfaces.api.dependencies.auth import client_or_user_has_access_func
 from nlbone.interfaces.api.exceptions import BadRequestException, InternalServerException
 
@@ -36,6 +40,81 @@ class AdditionalFieldsRequest:
         return [x.strip() for x in s.split(",") if x.strip()]
 
 
+async def resolve_requested_fields_async(
+    reg: ResourceRegistry,
+    additional_fields: AdditionalFieldsRequest = None,
+    can: AsyncPermissionChecker = None,
+    mode: DefaultsMergeMode = DefaultsMergeMode.UNION,
+) -> Tuple[Set[str], Dict[str, FieldRule]]:
+    if not additional_fields:
+        additional_fields = AdditionalFieldsRequest()
+    if not can:
+        can = async_client_or_user_has_access_func
+
+    reg.finalize_defaults(mode=mode)
+
+    expanded_bundles = _expand_bundles(reg, set(additional_fields.bundles))
+    requested = set(reg.default_fields) | set(additional_fields.fields) | expanded_bundles
+
+    # Validation
+    unknown = {f for f in requested if (f not in reg.default_fields and f not in reg.rules)}
+    if unknown:
+        raise BadRequestException(f"unknown_fields: {sorted(unknown)}")
+
+    selected_rules: Dict[str, FieldRule] = {}
+
+    for f in requested:
+        rule = reg.rules.get(f)
+
+        if f in reg.default_fields and f not in reg.rules:
+            continue
+
+        if not rule:
+            continue
+
+        # Check Permission
+        if rule.permission:
+            try:
+                if not await can(rule.permission):
+                    continue
+            except Exception:
+                continue
+
+        is_explicit = f in additional_fields.fields
+        if not rule.default and not is_explicit:
+            continue
+
+        selected_rules[f] = rule
+
+    # Dependencies
+    final_set = set(requested)
+
+    def add_deps_recursive(name: str):
+        _rule = reg.rules.get(name)
+        if not _rule or not _rule.deps:
+            return
+        for d in _rule.deps:
+            if d not in final_set:
+                final_set.add(d)
+                add_deps_recursive(d)
+
+    for field_with_rule in list(selected_rules.keys()):
+        add_deps_recursive(field_with_rule)
+
+    parents = {f.split(".", 1)[0] for f in final_set if "." in f}
+    for p in parents:
+        if p in reg.rules:
+            final_set.add(p)
+            if p not in selected_rules:
+                selected_rules[p] = reg.rules[p]
+
+    missing_deps = (final_set - set(reg.default_fields)) - set(reg.rules.keys())
+    if missing_deps:
+        raise InternalServerException(f"registry_missing_rules_for: {sorted(missing_deps)}")
+
+    return final_set, selected_rules
+
+
 def resolve_requested_fields(
     reg: ResourceRegistry,
     additional_fields: AdditionalFieldsRequest = None,

nlbone/interfaces/api/dependencies/async_auth.py

@@ -92,7 +92,7 @@ def client_or_user_has_access(*, permissions=None, client_permissions=None):
     return decorator
 
 
-async def is_permitted_user(permissions=None):
+def is_permitted_user(permissions=None):
     async def check_permissions():
         try:
             if bypass_authz():

{nlbone-0.11.0.dist-info → nlbone-0.11.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nlbone
-Version: 0.11.0
+Version: 0.11.3
 Summary: Backbone package for interfaces and infrastructure in Python projects
 Author-email: Amir Hosein Kahkbazzadeh <a.khakbazzadeh@gmail.com>
 License: MIT

{nlbone-0.11.0.dist-info → nlbone-0.11.3.dist-info}/RECORD

@@ -19,8 +19,8 @@ nlbone/adapters/db/postgres/__init__.py,sha256=tvCpHOdZbpQ57o7k-plq7L0e1uZe5_Frb
 nlbone/adapters/db/postgres/audit.py,sha256=IuWkPitr70UyQ6-GkAedckp8U-Z4cTgzFbdt_bQv1VQ,4800
 nlbone/adapters/db/postgres/base.py,sha256=I89PsEeR9ADEScG8D5pVSncPrPRBmf-KQQkjajl7Koo,132
 nlbone/adapters/db/postgres/engine.py,sha256=8bA5qbDN9kcbI2uFitxbD8bseRlI_wQRQQSgfRTH6l8,3812
-nlbone/adapters/db/postgres/query_builder.py,sha256=UTjo83NH09g_TIW1Lg_wpTUp62Xvtk4ihP-bqMzfaNI,15839
-nlbone/adapters/db/postgres/repository.py,sha256=n01TAzdKd-UbOhirE6KMosuvRdJG2l1cszwVHjTM-Ks,10345
+nlbone/adapters/db/postgres/query_builder.py,sha256=sxl_K_WIDjfi2ISTZxWdY6JLXwj6vjkLLM3Jzp305kw,17440
+nlbone/adapters/db/postgres/repository.py,sha256=SJbrkUcY_MoweBjSQRx4yE4BwqFl0GhTkZVW6VPkHyU,10380
 nlbone/adapters/db/postgres/schema.py,sha256=NlE7Rr8uXypsw4oWkdZhZwcIBHQEPIpoHLxcUo98i6s,1039
 nlbone/adapters/db/postgres/types.py,sha256=0SVuIKokog6_ByrYUsYAoIypVM2-uKJhUTeDPtm0qhs,602
 nlbone/adapters/db/postgres/uow.py,sha256=I4RVeIbGEVhVGcuzhdEUJuX11RhEHTn0egrkCbcsz24,3852
@@ -78,13 +78,13 @@ nlbone/interfaces/api/exceptions.py,sha256=IggZxV9q6l4jqw-G7SWEmuyXnWgbNXJJT-rmn
 nlbone/interfaces/api/routers.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nlbone/interfaces/api/schemas.py,sha256=34Tz2EeXyf12rFL9iyYWaB2ftuaXUebQQQxSO9ouV94,133
 nlbone/interfaces/api/additional_filed/__init__.py,sha256=BWemliLSQV9iq1vdUaF733q0FOSipSWBOQk9eYj732Q,318
-nlbone/interfaces/api/additional_filed/assembler.py,sha256=E_TGxorIQ-hC5FiCup5WQ0MmbSGfNaypEf5Plc-rjJs,1988
-nlbone/interfaces/api/additional_filed/field_registry.py,sha256=IhIvzHWOMtKv8iTdFu7LzQboi4SFTLOyJqKdYPY8xFE,5418
-nlbone/interfaces/api/additional_filed/resolver.py,sha256=jv1TIBBHN4LBIMwHGipcy4iq0uP0r6udyaqvhRzb8Bk,4655
+nlbone/interfaces/api/additional_filed/assembler.py,sha256=YsQiui7dr9IAEM5X9KKGDSFeDqA5Zqaj4egoPOWM_ug,3394
+nlbone/interfaces/api/additional_filed/field_registry.py,sha256=bnazgHZ7aO6ZTSaqZKjE8xJOOnbOLi3N19IyR2sV_SA,5487
+nlbone/interfaces/api/additional_filed/resolver.py,sha256=UGZc3hkh6wYjLgzT7lGeGrw8qmaEzouK6Sqk4uSvi34,7125
 nlbone/interfaces/api/additional_filed/default_field_rules/__init__.py,sha256=LUSAOO3xRUt5ptlraIx7H-7dSkdr1D-WprmnqXRB16g,48
 nlbone/interfaces/api/additional_filed/default_field_rules/image_field_rules.py,sha256=ecKqPeXZ-YiF14RK9PmK7ln3PCzpCUc18S5zm5IF3fw,339
 nlbone/interfaces/api/dependencies/__init__.py,sha256=nrmQftdCfKlqSE44R6PkcxtkwCdNpZgI8yLlHyeiACA,274
-nlbone/interfaces/api/dependencies/async_auth.py,sha256=c6PohIprT35konFbHQht0y0MJHBouhKqIK_XFQ9Rcbg,3465
+nlbone/interfaces/api/dependencies/async_auth.py,sha256=Suo2rD-zY5MOLnFBaxlqKxp1XI2f3o3W8ULG-4LF8Ok,3459
 nlbone/interfaces/api/dependencies/auth.py,sha256=4L-hspOyv9HpaCO-rAi7rk52PlZTgMClEG2LA2tMriM,3836
 nlbone/interfaces/api/dependencies/client_credential.py,sha256=Bo4dYx75Qw0JzTKD9ZfV5EXDEOuwndJk2D-V37K2ePg,1293
 nlbone/interfaces/api/dependencies/db.py,sha256=-UD39J_86UU7ZJs2ZncpdND0yhAG0NeeeALrgSDuuFw,466
@@ -119,8 +119,8 @@ nlbone/utils/normalize_mobile.py,sha256=sGH4tV9gX-6eVKozviNWJhm1DN1J28Nj-ERldCYk
 nlbone/utils/read_files.py,sha256=mx8dfvtaaARQFRp_U7OOiERg-GT62h09_lpTzIQsVhs,291
 nlbone/utils/redactor.py,sha256=-V4HrHmHwPi3Kez587Ek1uJlgK35qGSrwBOvcbw8Jas,1279
 nlbone/utils/time.py,sha256=DjjyQ9GLsfXoT6NK8RDW2rOlJg3e6sF04Jw6PBUrSvg,1268
-nlbone-0.11.0.dist-info/METADATA,sha256=tCSglsRxWG-yPYxEmv_M9TMoZqsq_ixuNCgXXK7jJBM,2295
-nlbone-0.11.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nlbone-0.11.0.dist-info/entry_points.txt,sha256=CpIL45t5nbhl1dGQPhfIIDfqqak3teK0SxPGBBr7YCk,59
-nlbone-0.11.0.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nlbone-0.11.0.dist-info/RECORD,,
+nlbone-0.11.3.dist-info/METADATA,sha256=L0Q66tOht43nYTe90NA2zf_e88iV1ucwgmXac8Us_XM,2295
+nlbone-0.11.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nlbone-0.11.3.dist-info/entry_points.txt,sha256=CpIL45t5nbhl1dGQPhfIIDfqqak3teK0SxPGBBr7YCk,59
+nlbone-0.11.3.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nlbone-0.11.3.dist-info/RECORD,,