From a37788a97cc43a58e36f581b7479a0cdc27b34e3 Mon Sep 17 00:00:00 2001 From: Evie Litherland-Smith Date: Sun, 24 Mar 2024 08:07:00 +0000 Subject: [PATCH] Add click for CLI, add __version__ to __init__.py --- poetry.lock | 2 +- pyproject.toml | 1 + speedtest_logging/__init__.py | 1 + speedtest_logging/main.py | 16 ++++++++++++++-- 4 files changed, 17 insertions(+), 3 deletions(-) diff --git a/poetry.lock b/poetry.lock index af3d29c..b1aa92d 100644 --- a/poetry.lock +++ b/poetry.lock @@ -560,4 +560,4 @@ test = ["covdefaults (>=2.3)", "coverage (>=7.2.7)", "coverage-enable-subprocess [metadata] lock-version = "2.0" python-versions = "^3.10" -content-hash = "5b5fb95836d5b3dd0fc9d398a46c468246d37709185164edd068c047d7564df2" +content-hash = "b0fa64239e05270bf23d15f90756b11b36793c9b53e02e7e8cf43377163005de" diff --git a/pyproject.toml b/pyproject.toml index a781315..9618459 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,6 +19,7 @@ speedtest-logging = "speedtest_logging.main:main" python = "^3.10" speedtest = "^0.0.1" pandas = "^2.2.1" +click = "^8.1.7" [tool.poetry.group.dev.dependencies] mypy = "^1.9.0" diff --git a/speedtest_logging/__init__.py b/speedtest_logging/__init__.py index e69de29..3dc1f76 100644 --- a/speedtest_logging/__init__.py +++ b/speedtest_logging/__init__.py @@ -0,0 +1 @@ +__version__ = "0.1.0" diff --git a/speedtest_logging/main.py b/speedtest_logging/main.py index 0cbc4d9..d38fa45 100644 --- a/speedtest_logging/main.py +++ b/speedtest_logging/main.py @@ -1,10 +1,22 @@ +from pathlib import Path + +import click import pandas as pd import speedtest +from . import __version__ +CONTEXT_SETTINGS = dict(help_option_names=["-h", "--help"]) + + +@click.command(context_settings=CONTEXT_SETTINGS) +@click.version_option(version=__version__) def main(*args, **kwargs) -> None: - print(pd.__version__) - print(speedtest.__version__) + """ + Entry point to run speedtest logging script + """ + print(f"{pd.__version__ = }") + print(f"{speedtest.__version__ = }") if __name__ == "__main__":