diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 7477740..471af7f 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -89,7 +89,7 @@ repos: hooks: - id: python_file_name_check name: "Python: File name check" - entry: "python ./src/precommithooks/__main__.py" + entry: "python ./src/precommithooks/checknames.py" language: python pass_filenames: true types: [file, python] diff --git a/requirements-dev.txt b/requirements-dev.txt index 1325143..392e18d 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -89,7 +89,7 @@ pyproject-api==1.6.1 # via tox pyproject-hooks==1.0.0 # via build -pytest==8.1.2 +pytest==8.2.0 # via # pytest-cov # pytest-pretty diff --git a/src/precommithooks/__init__.py b/src/precommithooks/__init__.py index 235ca74..1cc1778 100644 --- a/src/precommithooks/__init__.py +++ b/src/precommithooks/__init__.py @@ -1,4 +1,5 @@ """Precommithooks package.""" -from .__main__ import Mode as Mode -from .__main__ import check_names as check_names +from .checknames import Mode as Mode +from .checknames import check_names as check_names +from .checknames import main as main diff --git a/src/precommithooks/__main__.py b/src/precommithooks/checknames.py similarity index 92% rename from src/precommithooks/__main__.py rename to src/precommithooks/checknames.py index dc59067..8028136 100644 --- a/src/precommithooks/__main__.py +++ b/src/precommithooks/checknames.py @@ -86,7 +86,7 @@ def check_names( # noqa: C901, PLR0912 return ExitCode.OK -def main(argv: Sequence[str] | None = None) -> int: +def main() -> int: """Get files from passed argument.""" parser = argparse.ArgumentParser() parser.add_argument( @@ -112,7 +112,7 @@ def main(argv: Sequence[str] | None = None) -> int: action="store_true", help="Ignore test files", ) - args = parser.parse_args(argv) + args = parser.parse_args() mode = Mode.NON_STRICT short_name_limit = ( int(args.short_name_limit) @@ -121,14 +121,12 @@ def main(argv: Sequence[str] | None = None) -> int: ) if args.strict: mode = Mode.STRICT - if argv is not None: - return check_names( - args.filenames, - short_name_limit=short_name_limit, - ignore_test_files=args.ignore_test_files, - mode=mode, - ) - return ExitCode.OK + return check_names( + args.filenames, + short_name_limit=short_name_limit, + ignore_test_files=args.ignore_test_files, + mode=mode, + ) if __name__ == "__main__":