Move main to __init__ and enable python -m jockey.

Author: peterctl

pyproject.toml

@@ -22,7 +22,7 @@ packages = [
 ]
 
 [tool.poetry.scripts]
-juju-jockey = "jockey.__main__:main"
+juju-jockey = "jockey:main"
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.13"
@@ -145,7 +145,7 @@ version = "6.7.0"
 disable-upx = true
 
 [tool.poetry-pyinstaller-plugin.scripts]
-juju-jockey = { source = "src/jockey/__main__.py", type = "onefile", bundle = false }
+juju-jockey = { source = "src/jockey/__init__.py", type = "onefile", bundle = false }
 
 [tool.commitizen]
 tag_format = "v$version"

src/jockey/__init__.py

@@ -13,3 +13,62 @@
 __copyright__ = "Copyright (c) 2024, Connor Chamberlain"
 
 __docformat__ = "numpy"
+
+import logging
+import os
+from pkgutil import get_data
+import sys
+from typing import Optional, Sequence
+
+from rich import print
+from rich.console import Console
+from rich.markdown import Markdown
+
+from jockey.__args__ import parse_args
+from jockey.core import query
+from jockey.log import configure_logging
+
+
+logger = logging.getLogger(__name__)
+
+
+if "SNAP" in os.environ:
+    os.environ["PATH"] += ":" + os.path.join(os.environ["SNAP"], "usr", "juju", "bin")
+
+
+def info() -> Markdown:
+    info_data = get_data("jockey", "info.md")
+    info_decoded = info_data.decode("utf-8") if info_data else ""
+    return Markdown(info_decoded)
+
+
+def print_info(console: Optional[Console] = None) -> None:
+    if not console:
+        console = Console(width=120)
+
+    console.print(info())
+
+
+def main(argv: Optional[Sequence[str]] = None) -> int:
+    # parse command-line arguments and configure logging
+    if argv is None:
+        argv = sys.argv[1:]
+
+    args = parse_args(argv)
+    logger.debug("Parsed command-line arguments:\n%r", args)
+
+    verbosity = args.verbose if "verbose" in args else 0
+    configure_logging(verbosity)
+
+    # check if OBJECT is requesting the informational message
+    if args.object == "info":
+        print_info()
+        return 0
+
+    filters = args.filters if "filters" in args else []
+    print("\n".join(query(object_type=args.object, filter_strings=filters, file=args.file, model=args.model)))
+    return 0
+
+
+if __name__ == "__main__":
+    exit(main(sys.argv[1:]))