Fix linting, typing

karpetrosyan · Sep 30, 2023 · f11774e · f11774e
1 parent 11214d1
commit f11774e
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 10 deletions.
diff --git a/anysqlite/__init__.py b/anysqlite/__init__.py
@@ -19,4 +19,25 @@
 
 from ._core import Connection, Cursor, connect  # noqa: F401
 
+__all__ = [
+    "Connection",
+    "Cursor",
+    "connect",
+    "DatabaseError",
+    "DataError",
+    "Error",
+    "IntegrityError",
+    "InterfaceError",
+    "InternalError",
+    "NotSupportedError",
+    "OperationalError",
+    "ProgrammingError",
+    "Row",
+    "Warning",
+    "apilevel",
+    "paramstyle",
+    "sqlite_version",
+    "sqlite_version_info",
+    "threadsafety",
+]
 __version__ = "0.0.3"
diff --git a/anysqlite/_core.py b/anysqlite/_core.py
@@ -1,9 +1,10 @@
 import sqlite3
 import typing as tp
-from functools import partial, wraps
+from functools import partial
+from pathlib import Path
 
 from anyio import CapacityLimiter, to_thread
-from pathlib import Path
+
 
 class Connection:
     def __init__(self, _real_connection: sqlite3.Connection) -> None:
@@ -38,7 +39,11 @@ def __init__(self, real_cursor: sqlite3.Cursor, limiter: CapacityLimiter) -> Non
         self._limiter = limiter
 
     @property
-    def description(self) -> tp.Union[tp.Tuple[tp.Tuple[str, None, None, None, None, None, None], ...], tp.Any]:
+    def description(
+        self,
+    ) -> tp.Union[
+        tp.Tuple[tp.Tuple[str, None, None, None, None, None, None], ...], tp.Any
+    ]:
         return self._real_cursor.description
 
     @property
@@ -50,17 +55,17 @@ def arraysize(self) -> int:
         return self._real_cursor.arraysize
 
     async def close(self) -> None:
-        await to_thread.run_sync(
-            self._real_cursor.close, limiter=self._limiter
-        )
+        await to_thread.run_sync(self._real_cursor.close, limiter=self._limiter)
 
     async def execute(self, sql: str, parameters: tp.Iterable[tp.Any] = ()) -> "Cursor":
         real_cursor = await to_thread.run_sync(
             self._real_cursor.execute, sql, parameters, limiter=self._limiter
         )
         return Cursor(real_cursor, self._limiter)
 
-    async def executemany(self, sql: str, seq_of_parameters: tp.Iterable[tp.Iterable[tp.Any]]) -> "Cursor":
+    async def executemany(
+        self, sql: str, seq_of_parameters: tp.Iterable[tp.Iterable[tp.Any]]
+    ) -> "Cursor":
         real_cursor = await to_thread.run_sync(
             self._real_cursor.executemany, sql, seq_of_parameters, limiter=self._limiter
         )
@@ -87,9 +92,10 @@ async def fetchall(self) -> tp.Any:
             self._real_cursor.fetchall, limiter=self._limiter
         )
 
-async def connect(database: tp.Union[str, bytes, Path],
-                  **kwargs: tp.Any) -> "Connection":
-
+
+async def connect(
+    database: tp.Union[str, bytes, Path], **kwargs: tp.Any
+) -> "Connection":
     kwargs["check_same_thread"] = False
     real_connection = await to_thread.run_sync(
         partial(sqlite3.connect, database, **kwargs)