From 282e81362ae34b134abab3d774963d6b2a57a1be Mon Sep 17 00:00:00 2001 From: Simon Willison Date: Wed, 18 Aug 2021 14:55:37 -0700 Subject: [PATCH] Applied Black plus some extra type hints --- setup.py | 8 +++++++- sqlite_utils/db.py | 6 +++--- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 0d3e456..71d9cd4 100644 --- a/setup.py +++ b/setup.py @@ -33,7 +33,13 @@ setup( extras_require={ "test": ["pytest", "black", "hypothesis"], "docs": ["sphinx_rtd_theme", "sphinx-autobuild", "codespell"], - "mypy": ["mypy", "types-click", "types-tabulate", "types-python-dateutil", "data-science-types"], + "mypy": [ + "mypy", + "types-click", + "types-tabulate", + "types-python-dateutil", + "data-science-types", + ], "flake8": ["flake8"], }, entry_points=""" diff --git a/sqlite_utils/db.py b/sqlite_utils/db.py index 4677733..893c3b2 100644 --- a/sqlite_utils/db.py +++ b/sqlite_utils/db.py @@ -277,7 +277,7 @@ class Database: def __init__( self, - filename_or_conn=None, + filename_or_conn: Union[str, pathlib.Path, sqlite3.Connection] = None, memory: bool = False, recreate: bool = False, recursive_triggers: bool = True, @@ -331,7 +331,7 @@ class Database: """ return self.table(table_name) - def __repr__(self): + def __repr__(self) -> str: return "".format(self.conn) def register_function( @@ -1069,7 +1069,7 @@ class Table(Queryable): return next(iter(counts.values())) return self.count_where() - def exists(self): + def exists(self) -> bool: return self.name in self.db.table_names() @property