Skip to content

Commit

Permalink
Add mypy to pre-commit
Browse files Browse the repository at this point in the history
  • Loading branch information
hugovk committed May 2, 2024
1 parent 3a33323 commit 69f023e
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 7 deletions.
19 changes: 19 additions & 0 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,25 @@ repos:
- id: trailing-whitespace
exclude: tests/data/expected_tabulated.py

- repo: https://github.com/pre-commit/mirrors-mypy
rev: v1.10.0
hooks:
- id: mypy
additional_dependencies:
[
freezegun,
platformdirs,
prettytable,
pytablewriter,
pytest,
respx,
termcolor,
types-python-dateutil,
types-python-slugify,
]
args: [--pretty, --show-error-codes, .]
pass_filenames: false

- repo: https://github.com/tox-dev/pyproject-fmt
rev: 1.7.0
hooks:
Expand Down
15 changes: 8 additions & 7 deletions src/pypistats/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
import sys
import warnings
from pathlib import Path
from typing import Any

from platformdirs import user_cache_dir
from slugify import slugify
Expand All @@ -24,13 +25,13 @@
USER_AGENT = f"pypistats/{__version__}"


def _print_verbose(verbose: bool, *args, **kwargs) -> None:
def _print_verbose(verbose: bool, *args, **kwargs: Any) -> None:
"""Print if verbose"""
if verbose:
_print_stderr(*args, **kwargs)


def _print_stderr(*args, **kwargs) -> None:
def _print_stderr(*args, **kwargs: Any) -> None:
"""Print to stderr"""
print(*args, file=sys.stderr, **kwargs)

Expand Down Expand Up @@ -474,38 +475,38 @@ def _paramify(param_name: str, param_value: float | str | None) -> str:
return ""


def recent(package: str, period: str | None = None, **kwargs: str):
def recent(package: str, period: str | None = None, **kwargs: Any):
"""Retrieve the aggregate download quantities for the last day/week/month"""
endpoint = f"packages/{package}/recent"
params = _paramify("period", period)
return pypi_stats_api(endpoint, params, **kwargs)


def overall(package: str, mirrors: bool | str | None = None, **kwargs: str):
def overall(package: str, mirrors: bool | str | None = None, **kwargs: Any):
"""Retrieve the aggregate daily download time series with or without mirror
downloads"""
endpoint = f"packages/{package}/overall"
params = _paramify("mirrors", mirrors)
return pypi_stats_api(endpoint, params, **kwargs)


def python_major(package: str, version: str | None = None, **kwargs: str):
def python_major(package: str, version: str | None = None, **kwargs: Any):
"""Retrieve the aggregate daily download time series by Python major version
number"""
endpoint = f"packages/{package}/python_major"
params = _paramify("version", version)
return pypi_stats_api(endpoint, params, **kwargs)


def python_minor(package: str, version: str | None = None, **kwargs) -> str:
def python_minor(package: str, version: str | None = None, **kwargs: Any) -> str:
"""Retrieve the aggregate daily download time series by Python minor version
number"""
endpoint = f"packages/{package}/python_minor"
params = _paramify("version", version)
return pypi_stats_api(endpoint, params, **kwargs)


def system(package: str, os: str | None = None, **kwargs):
def system(package: str, os: str | None = None, **kwargs: Any):
"""Retrieve the aggregate daily download time series by operating system"""
endpoint = f"packages/{package}/system"
params = _paramify("os", os)
Expand Down

0 comments on commit 69f023e

Please sign in to comment.