mirror of
https://github.com/creyD/creyPY.git
synced 2026-04-13 11:50:31 +02:00
Compare commits
5 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 2ad7700f72 | |||
| 1d7b767623 | |||
|
|
f1f29e84c2 | ||
| dcb9afb8f2 | |||
|
|
10c1ea5411 |
@@ -1,27 +1,26 @@
|
|||||||
|
from contextlib import suppress
|
||||||
from math import ceil
|
from math import ceil
|
||||||
from typing import Any, Generic, Optional, Self, Sequence, TypeVar, Union, overload
|
from typing import Any, Generic, Optional, Self, Sequence, TypeVar, Union, overload
|
||||||
from contextlib import suppress
|
|
||||||
from pydantic import BaseModel
|
from fastapi import Query
|
||||||
from fastapi_pagination import Params
|
from fastapi_pagination.api import apply_items_transformer, create_page
|
||||||
from fastapi_pagination.bases import AbstractPage, AbstractParams
|
from fastapi_pagination.bases import AbstractPage, AbstractParams, RawParams
|
||||||
|
from fastapi_pagination.ext.sqlalchemy import create_paginate_query
|
||||||
from fastapi_pagination.types import (
|
from fastapi_pagination.types import (
|
||||||
|
AdditionalData,
|
||||||
|
AsyncItemsTransformer,
|
||||||
GreaterEqualOne,
|
GreaterEqualOne,
|
||||||
GreaterEqualZero,
|
GreaterEqualZero,
|
||||||
AdditionalData,
|
|
||||||
SyncItemsTransformer,
|
|
||||||
AsyncItemsTransformer,
|
|
||||||
ItemsTransformer,
|
ItemsTransformer,
|
||||||
|
SyncItemsTransformer,
|
||||||
)
|
)
|
||||||
from fastapi_pagination.api import create_page, apply_items_transformer
|
|
||||||
from fastapi_pagination.utils import verify_params
|
from fastapi_pagination.utils import verify_params
|
||||||
from fastapi_pagination.ext.sqlalchemy import create_paginate_query
|
from pydantic import BaseModel
|
||||||
from fastapi_pagination.bases import AbstractParams, RawParams
|
|
||||||
from pydantic.json_schema import SkipJsonSchema
|
from pydantic.json_schema import SkipJsonSchema
|
||||||
from sqlalchemy.sql.selectable import Select
|
from sqlalchemy import func, select
|
||||||
from sqlalchemy.orm.session import Session
|
|
||||||
from sqlalchemy import select, func
|
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession, async_scoped_session
|
from sqlalchemy.ext.asyncio import AsyncSession, async_scoped_session
|
||||||
from fastapi import Query
|
from sqlalchemy.orm.session import Session
|
||||||
|
from sqlalchemy.sql.selectable import Select
|
||||||
from sqlalchemy.util import await_only, greenlet_spawn
|
from sqlalchemy.util import await_only, greenlet_spawn
|
||||||
|
|
||||||
T = TypeVar("T")
|
T = TypeVar("T")
|
||||||
@@ -29,7 +28,7 @@ T = TypeVar("T")
|
|||||||
|
|
||||||
class PaginationParams(BaseModel, AbstractParams):
|
class PaginationParams(BaseModel, AbstractParams):
|
||||||
page: int = Query(1, ge=1, description="Page number")
|
page: int = Query(1, ge=1, description="Page number")
|
||||||
size: int = Query(50, ge=1, le=100, description="Page size")
|
size: int = Query(50, ge=1, description="Page size")
|
||||||
pagination: bool = Query(True, description="Toggle pagination")
|
pagination: bool = Query(True, description="Toggle pagination")
|
||||||
|
|
||||||
def to_raw_params(self) -> RawParams:
|
def to_raw_params(self) -> RawParams:
|
||||||
@@ -62,7 +61,7 @@ class Page(AbstractPage[T], Generic[T]):
|
|||||||
total: Optional[int] = None,
|
total: Optional[int] = None,
|
||||||
**kwargs: Any,
|
**kwargs: Any,
|
||||||
) -> Self:
|
) -> Self:
|
||||||
if not isinstance(params, Params):
|
if not isinstance(params, PaginationParams):
|
||||||
raise TypeError("Page should be used with Params")
|
raise TypeError("Page should be used with Params")
|
||||||
|
|
||||||
size = params.size or total or len(items)
|
size = params.size or total or len(items)
|
||||||
@@ -170,9 +169,9 @@ def _paginate(
|
|||||||
total = connection.scalar(count_query)
|
total = connection.scalar(count_query)
|
||||||
|
|
||||||
if params.pagination is False and total > 0:
|
if params.pagination is False and total > 0:
|
||||||
params = Params(page=1, size=total)
|
params = PaginationParams(page=1, size=total)
|
||||||
else:
|
else:
|
||||||
params = Params(page=params.page, size=params.size)
|
params = PaginationParams(page=params.page, size=params.size)
|
||||||
|
|
||||||
query = create_paginate_query(query, params)
|
query = create_paginate_query(query, params)
|
||||||
items = connection.execute(query).all()
|
items = connection.execute(query).all()
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import unittest
|
|||||||
from typing import Type
|
from typing import Type
|
||||||
|
|
||||||
from httpx import ASGITransport, AsyncClient, Response
|
from httpx import ASGITransport, AsyncClient, Response
|
||||||
from sqlalchemy import create_engine
|
from sqlalchemy import create_engine, text
|
||||||
from sqlalchemy.ext.asyncio import create_async_engine
|
from sqlalchemy.ext.asyncio import create_async_engine
|
||||||
from sqlalchemy_utils import create_database, database_exists, drop_database
|
from sqlalchemy_utils import create_database, database_exists, drop_database
|
||||||
|
|
||||||
@@ -23,7 +23,9 @@ class AbstractTestAPI(unittest.IsolatedAsyncioTestCase):
|
|||||||
print("setting up abstract")
|
print("setting up abstract")
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def setup_database(cls, sync_db_url: str, async_db_url: str, base: Type[Base]):
|
def setup_database(
|
||||||
|
cls, sync_db_url: str, async_db_url: str, base: Type[Base], btree_gist: bool = False
|
||||||
|
):
|
||||||
cls.engine_s = create_engine(
|
cls.engine_s = create_engine(
|
||||||
sync_db_url,
|
sync_db_url,
|
||||||
echo=False,
|
echo=False,
|
||||||
@@ -34,6 +36,10 @@ class AbstractTestAPI(unittest.IsolatedAsyncioTestCase):
|
|||||||
drop_database(cls.engine_s.url)
|
drop_database(cls.engine_s.url)
|
||||||
create_database(cls.engine_s.url)
|
create_database(cls.engine_s.url)
|
||||||
|
|
||||||
|
if btree_gist:
|
||||||
|
with cls.engine_s.begin() as conn:
|
||||||
|
conn.execute(text("CREATE EXTENSION IF NOT EXISTS btree_gist"))
|
||||||
|
|
||||||
# Migrate
|
# Migrate
|
||||||
base.metadata.create_all(cls.engine_s)
|
base.metadata.create_all(cls.engine_s)
|
||||||
|
|
||||||
|
|||||||
@@ -1 +1 @@
|
|||||||
stripe==10.12.0 # Stripe
|
stripe==11.4.1 # Stripe
|
||||||
|
|||||||
Reference in New Issue
Block a user