Mypy checks + fixes (#217)

package.json

@@ -10,7 +10,7 @@
     "prepack": "rm -rf dist && npm build",
     "test": "jest",
     "codegen": "subql codegen",
-    "pylint-checks": "black --check --verbose src/genesis tests && isort --check-only src/genesis tests && flake8 src/genesis tests",
+    "pylint-checks": "black --check --verbose src/genesis tests && isort --check-only src/genesis tests && flake8 src/genesis tests && mypy src/genesis --disable-error-code import && mypy tests --disable-error-code import",
     "pylint": "black src/genesis tests && isort src/genesis tests && flake8 src/genesis tests && mypy src/genesis --disable-error-code import && mypy tests --disable-error-code import && pylint src/genesis tests",
     "start:docker": "docker-compose pull && docker-compose up --remove-orphans",
     "start:tracing": "docker-compose -f docker-compose.tracing.yml pull && docker-compose -f docker-compose.tracing.yml up --remove-orphans",

setup.cfg

@@ -1,12 +1,8 @@
 [flake8]
-paths=cosmpy
+paths=src, tests
 copyright-check = True
 select = B,C,D,E,F,I,W,
-exclude=.md,
-    vulture_whitelist.py
-    cosmpy/protos
-    tests/unit/test_clients/test_cosmos_ledger.py
-    tests/integration/
+exclude=.md
 ignore = D403, E501, W503, F403, E203
 docstring_style=sphinx
 strictness=short
@@ -23,6 +19,7 @@ strictness=short
 python_version = 3.9
 strict_optional = True
 exclude = vulture_whitelist.py
+namespace_packages = False
 
 [mypy-google.*]
 ignore_missing_imports = True

src/genesis/db/table_manager.py

@@ -8,15 +8,19 @@
 
 def table_exists(db_conn: Connection, table: str) -> bool:
     with db_conn.cursor() as db:
-        return db.execute(
+        res_db_execute = db.execute(
             f"""
                 SELECT EXISTS (
                     SELECT FROM pg_tables WHERE
                         schemaname = 'app' AND
                         tablename  = '{table}'
                 )
             """
-        ).fetchone()[0]
+        ).fetchone()
+
+        assert res_db_execute is not None
+
+        return res_db_execute[0]
 
 
 class TableManager:
@@ -29,14 +33,13 @@ def __init__(self, db_conn: Connection):
         self._db_conn = db_conn
 
     @classmethod
-    @property
-    def column_names(cls) -> Generator[str, Any, None]:
+    def get_column_names(cls) -> Generator[str, Any, None]:
         return (name for name, _ in cls._columns)
 
     @classmethod
     def select_query(cls) -> str:
         return f"""
-            SELECT {",".join(cls.column_names)} FROM {cls._table}
+            SELECT {",".join(cls.get_column_names())} FROM {cls._table}
         """
 
     def _ensure_table(self):