style: Run lint

Author: amisadmin

sqlalchemy_database/_abc_async_database.py

@@ -1,12 +1,13 @@
 import abc
 import asyncio
 import functools
-from typing import TypeVar, Callable
+from typing import Callable, TypeVar
 
 try:
     from asyncio import to_thread  # python 3.9+
 except ImportError:
     import contextvars
+
     from typing_extensions import ParamSpec
 
     _P = ParamSpec("_P")
@@ -18,11 +19,11 @@ async def to_thread(func: Callable[_P, _R], *args: _P.args, **kwargs: _P.kwargs)
         func_call = functools.partial(ctx.run, func, *args, **kwargs)
         return await loop.run_in_executor(None, func_call)
 
-class AbcAsyncDatabase(metaclass = abc.ABCMeta):
 
+class AbcAsyncDatabase(metaclass=abc.ABCMeta):  # noqa: B024
     def __init__(self) -> None:
-        for func_name in ['execute', 'scalar', 'scalars_all', 'get', 'delete', 'save', 'run_sync']:
+        for func_name in ["execute", "scalar", "scalars_all", "get", "delete", "save", "run_sync"]:
             func = getattr(self, func_name)
             if not asyncio.iscoroutinefunction(func):
                 func = functools.partial(to_thread, func)
-            setattr(self, f'async_{func_name}', func)
+            setattr(self, f"async_{func_name}", func)

sqlalchemy_database/_abc_async_database.pyi

@@ -1,17 +1,27 @@
 import abc
-from typing import Any, Optional, Mapping, Union, Sequence, Type, List, TypeVar, Callable
+from typing import (
+    Any,
+    Callable,
+    List,
+    Mapping,
+    Optional,
+    Sequence,
+    Type,
+    TypeVar,
+    Union,
+)
 
-from sqlalchemy.engine import Result, Connection
+from sqlalchemy.engine import Connection, Result
 from sqlalchemy.ext.asyncio import AsyncConnection
 from sqlalchemy.sql import Executable
-from typing_extensions import ParamSpec, Concatenate
+from typing_extensions import Concatenate, ParamSpec
 
 try:
     from sqlmodel import Session
     from sqlmodel.ext.asyncio.session import AsyncSession
 except ImportError:
-    from sqlalchemy.orm import Session
     from sqlalchemy.ext.asyncio import AsyncSession
+    from sqlalchemy.orm import Session
 
 _P = ParamSpec("_P")
 _T = TypeVar("_T")
@@ -20,10 +30,8 @@ _R = TypeVar("_R")
 _ExecuteParams = Union[Mapping[Any, Any], Sequence[Mapping[Any, Any]]]
 _ExecuteOptions = Mapping[Any, Any]
 
-class AbcAsyncDatabase(metaclass = abc.ABCMeta):
-    """`sqlalchemy` asynchronous database abstract base class, not directly instantiated
-
-    """
+class AbcAsyncDatabase(metaclass=abc.ABCMeta):
+    """`sqlalchemy` asynchronous database abstract base class, not directly instantiated"""
 
     async def async_execute(
         self,
@@ -38,7 +46,6 @@ class AbcAsyncDatabase(metaclass = abc.ABCMeta):
         executor: Union[Session, Connection, AsyncSession, AsyncConnection, None] = None,
         **kw: Any,
     ) -> Union[Result, _T]: ...
-
     async def async_scalar(
         self,
         statement: Executable,
@@ -49,7 +56,6 @@ class AbcAsyncDatabase(metaclass = abc.ABCMeta):
         session: Union[Session, AsyncSession, None] = None,
         **kw: Any,
     ) -> Any: ...
-
     async def async_scalars_all(
         self,
         statement: Executable,
@@ -59,7 +65,6 @@ class AbcAsyncDatabase(metaclass = abc.ABCMeta):
         session: Union[Session, AsyncSession, None] = None,
         **kw: Any,
     ) -> List[Any]: ...
-
     async def async_get(
         self,
         entity: Type[_T],
@@ -70,18 +75,12 @@ class AbcAsyncDatabase(metaclass = abc.ABCMeta):
         with_for_update: Optional[Any] = None,
         identity_token: Optional[Any] = None,
         execution_options: Optional[_ExecuteOptions] = None,
-        session: Union[Session, AsyncSession, None] = None
+        session: Union[Session, AsyncSession, None] = None,
     ) -> Optional[_T]: ...
-
     async def async_delete(self, instance: Any) -> None: ...
-
     async def async_save(
-        self,
-        *instances: Any,
-        refresh: bool = False,
-        session: Union[Session, AsyncSession, None] = None
+        self, *instances: Any, refresh: bool = False, session: Union[Session, AsyncSession, None] = None
     ) -> None: ...
-
     async def async_run_sync(
         self,
         fn: Callable[[Concatenate[Union[Session, Connection], _P]], _T],
@@ -90,5 +89,5 @@ class AbcAsyncDatabase(metaclass = abc.ABCMeta):
         on_close_pre: Callable[[_T], _R] = None,
         is_session: bool = True,
         executor: Union[Session, Connection, AsyncSession, AsyncConnection, None] = None,
-        **kwargs: _P.kwargs
+        **kwargs: _P.kwargs,
     ) -> Union[_T, _R]: ...