From 131cf245c11a71742d88ce446908233b9a5c0d39 Mon Sep 17 00:00:00 2001 From: Waket Zheng Date: Fri, 10 Jan 2025 10:26:59 +0800 Subject: [PATCH] Apply examples/ tests/ --- examples/fastapi/_tests.py | 4 ++-- examples/fastapi/main.py | 2 +- examples/fastapi/main_custom_timezone.py | 2 +- examples/fastapi/routers.py | 4 +--- examples/signals.py | 4 ++-- tests/test_signals.py | 4 ++-- tests/testmodels.py | 4 ++-- 7 files changed, 11 insertions(+), 13 deletions(-) diff --git a/examples/fastapi/_tests.py b/examples/fastapi/_tests.py index 5257ba3f8..ff229ffdb 100644 --- a/examples/fastapi/_tests.py +++ b/examples/fastapi/_tests.py @@ -2,11 +2,11 @@ # pylint: disable=E0611,E0401 import multiprocessing import os +from collections.abc import AsyncGenerator from concurrent.futures import ProcessPoolExecutor from contextlib import asynccontextmanager from datetime import datetime from pathlib import Path -from typing import AsyncGenerator, Tuple import anyio import pytest @@ -74,7 +74,7 @@ async def create_user(self, async_client: AsyncClient) -> Users: assert user_obj.id == user_id return user_obj - async def user_list(self, async_client: AsyncClient) -> Tuple[datetime, Users, User_Pydantic]: + async def user_list(self, async_client: AsyncClient) -> tuple[datetime, Users, User_Pydantic]: utc_now = datetime.now(pytz.utc) user_obj = await Users.create(username="test") response = await async_client.get("/users") diff --git a/examples/fastapi/main.py b/examples/fastapi/main.py index 4b686e731..7aba0b88b 100644 --- a/examples/fastapi/main.py +++ b/examples/fastapi/main.py @@ -1,7 +1,7 @@ # pylint: disable=E0611,E0401 import os +from collections.abc import AsyncGenerator from contextlib import asynccontextmanager -from typing import AsyncGenerator from fastapi import FastAPI from routers import router as users_router diff --git a/examples/fastapi/main_custom_timezone.py b/examples/fastapi/main_custom_timezone.py index 54abce8d2..b9cc8b855 100644 --- a/examples/fastapi/main_custom_timezone.py +++ b/examples/fastapi/main_custom_timezone.py @@ -1,6 +1,6 @@ # pylint: disable=E0611,E0401 +from collections.abc import AsyncGenerator from contextlib import asynccontextmanager -from typing import AsyncGenerator from config import register_orm from fastapi import FastAPI diff --git a/examples/fastapi/routers.py b/examples/fastapi/routers.py index 3c6587ae9..1530af073 100644 --- a/examples/fastapi/routers.py +++ b/examples/fastapi/routers.py @@ -1,5 +1,3 @@ -from typing import List - from fastapi import APIRouter, HTTPException from models import Users from schemas import Status, User_Pydantic, UserIn_Pydantic @@ -7,7 +5,7 @@ router = APIRouter() -@router.get("/users", response_model=List[User_Pydantic]) +@router.get("/users", response_model=list[User_Pydantic]) async def get_users(): return await User_Pydantic.from_queryset(Users.all()) diff --git a/examples/signals.py b/examples/signals.py index 0abd741e2..929d0f6f6 100644 --- a/examples/signals.py +++ b/examples/signals.py @@ -2,7 +2,7 @@ This example demonstrates model signals usage """ -from typing import List, Optional, Type +from typing import Optional, Type from tortoise import BaseDBAsyncClient, Tortoise, fields, run_async from tortoise.models import Model @@ -33,7 +33,7 @@ async def signal_post_save( instance: Signal, created: bool, using_db: "Optional[BaseDBAsyncClient]", - update_fields: List[str], + update_fields: list[str], ) -> None: print(sender, instance, using_db, created, update_fields) diff --git a/tests/test_signals.py b/tests/test_signals.py index b53d2b053..c75000834 100644 --- a/tests/test_signals.py +++ b/tests/test_signals.py @@ -1,4 +1,4 @@ -from typing import List, Optional, Type +from typing import Optional, Type from tests.testmodels import Signals from tortoise import BaseDBAsyncClient @@ -20,7 +20,7 @@ async def signal_post_save( instance: Signals, created: bool, using_db: "Optional[BaseDBAsyncClient]", - update_fields: List, + update_fields: list, ) -> None: await Signals.filter(name="test2").update(name="test_post-save") await Signals.filter(name="test6").update(name="test_post-save") diff --git a/tests/testmodels.py b/tests/testmodels.py index 1e2c7dcf6..407d4ef8d 100644 --- a/tests/testmodels.py +++ b/tests/testmodels.py @@ -9,7 +9,7 @@ import uuid from decimal import Decimal from enum import Enum, IntEnum -from typing import List, Union +from typing import Union import pytz from pydantic import BaseModel, ConfigDict @@ -959,7 +959,7 @@ class OldStyleModel(Model): def camelize_var(var_name: str): - var_parts: List[str] = var_name.split("_") + var_parts: list[str] = var_name.split("_") return var_parts[0] + "".join([part.title() for part in var_parts[1:]])