Add some more type hints

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.1.4
+    rev: v0.1.5
     hooks:
       - id: ruff
         args: [--fix, --exit-non-zero-on-fix]
@@ -24,8 +24,26 @@ repos:
       - id: trailing-whitespace
         exclude: tests/data/expected_output.py
 
+  #  - repo: https://github.com/pre-commit/mirrors-mypy
+  #    rev: v1.7.0
+  #    hooks:
+  #      - id: mypy
+  #        args: [--pretty, --show-error-codes]
+  #        additional_dependencies:
+  #          [
+  #            freezegun,
+  #            platformdirs,
+  #            prettytable,
+  #            pytablewriter,
+  #            pytest,
+  #            respx,
+  #            termcolor,
+  #            types-python-dateutil,
+  #            types-python-slugify,
+  #          ]
+
   - repo: https://github.com/tox-dev/pyproject-fmt
-    rev: 1.4.1
+    rev: 1.5.1
     hooks:
       - id: pyproject-fmt
         additional_dependencies: [tox]

scripts/run_command.py

@@ -3,7 +3,7 @@
 import subprocess
 
 
-def run(command: str, with_console: bool = True, line_limit: int = None) -> None:
+def run(command: str, with_console: bool = True, line_limit: int | None = None) -> None:
     output = subprocess.run(command.split(), capture_output=True, text=True)
     print()
     if with_console:

src/norwegianblue/__init__.py

@@ -166,7 +166,10 @@ def _apply_colour(text: str, colour: str, *, is_html: bool = False) -> str:
 
 
 def _tabulate(
-    data: list[dict], format_: str = "markdown", color: str = "yes", title: str = None
+    data: list[dict],
+    format_: str = "markdown",
+    color: str = "yes",
+    title: str | None = None,
 ) -> str:
     """Return data in specified format"""
 
@@ -212,7 +215,7 @@ def _prettytable(
     data: list[dict],
     format_: str,
     color: str = "yes",
-    title: str = None,
+    title: str | None = None,
 ) -> str:
     from prettytable import MARKDOWN, SINGLE_BORDER, PrettyTable
 
@@ -240,7 +243,7 @@ def _prettytable(
 
 
 def _pytablewriter(
-    headers: list[str], data: list[dict], format_: str, title: str = None
+    headers: list[str], data: list[dict], format_: str, title: str | None = None
 ) -> str:
     from pytablewriter import (
         CsvTableWriter,

src/norwegianblue/_cache.py

@@ -21,7 +21,7 @@ def filename(url: str) -> Path:
     return CACHE_DIR / f"{today}-{slug}.json"
 
 
-def load(cache_file):
+def load(cache_file: Path):
     """Load data from cache_file"""
     if not cache_file.exists():
         return {}

src/norwegianblue/_data.py

@@ -8,7 +8,7 @@
         d[chr(i + c)] = chr((i + 47) % 94 + c)
 
 
-def text(s):
+def text(s: str) -> str:
     return "".join([d.get(c, c) for c in s])
 
 

tests/test_norwegianblue.py

@@ -6,6 +6,7 @@
 import json
 import os
 from pathlib import Path
+from typing import Any
 from unittest import mock
 
 import pytest
@@ -42,7 +43,7 @@
 EXPECTED_MD_WITH_TITLE = "## ubuntu\n" + EXPECTED_MD
 
 
-def stub__cache_filename(*args):
+def stub__cache_filename(*args: Any) -> Path:
     return Path("/this/does/not/exist")