Merge pull request #5 from mixxorz/chore/pyright

Add pyright type checking

.pre-commit-config.yaml

@@ -11,3 +11,11 @@ repos:
     rev: 22.6.0
     hooks:
       - id: black
+  - repo: local
+    hooks:
+      - id: pyright
+        name: pyright
+        entry: poetry run npx [email protected]
+        language: node
+        pass_filenames: false
+        types: [python]

dslr/operations.py

@@ -81,6 +81,9 @@ def get_snapshots() -> List[Snapshot]:
     # Find the snapshot databases
     result = exec_sql("SELECT datname FROM pg_database WHERE datname LIKE 'dslr_%'")
 
+    if not result:
+        raise RuntimeError("Did not get results from database.")
+
     snapshot_dbnames = sorted([row[0] for row in result])
 
     # Parse the name into a Snapshot

dslr/pg_client.py

@@ -1,6 +1,7 @@
 from typing import Any, List, Optional, Tuple
 
 import psycopg2
+import psycopg2.extensions
 
 from dslr.console import console
 

dslr/runner.py

@@ -54,7 +54,7 @@ def exec_shell(*cmd: str) -> Result:
 
 
 def exec_sql(
-    sql: Union[sql.SQL, str], data: Optional[List[Any]] = None
+    sql: Union[sql.Composed, str], data: Optional[List[Any]] = None
 ) -> Optional[List[Tuple[Any, ...]]]:
     """
     Executes a SQL query.

tox.ini

@@ -4,6 +4,7 @@ envlist =
     flake8
     black
     isort
+    pyright
     py37
     py38
     py39
@@ -14,7 +15,7 @@ python =
     3.7: py37
     3.8: py38
     3.9: py39
-    3.10: py310, flake8, black, isort
+    3.10: py310, flake8, black, isort, pyright
 
 [testenv]
 deps = psycopg2-binary
@@ -29,6 +30,10 @@ commands = flake8
 deps = black >=22.6.0, <23.0.0
 commands = black --check ./
 
+[testenv:pyright]
+deps = pyright == 1.1.266
+commands = pyright
+
 [testenv:isort]
 deps = isort >= 5.10.1, <6.0.0
 commands = isort dslr tests --check-only --diff