From fb7df81a949b569000407a8b7238de1bf5a31baf Mon Sep 17 00:00:00 2001 From: Ryan Reed Date: Sun, 3 Jul 2022 12:06:04 -0400 Subject: [PATCH] Adding argparse to the default console.py --- .../{{cookiecutter.package_name}}/console.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/{{cookiecutter.package_name}}/src/{{cookiecutter.package_name}}/console.py b/{{cookiecutter.package_name}}/src/{{cookiecutter.package_name}}/console.py index 38c0a10..4ba4e0f 100644 --- a/{{cookiecutter.package_name}}/src/{{cookiecutter.package_name}}/console.py +++ b/{{cookiecutter.package_name}}/src/{{cookiecutter.package_name}}/console.py @@ -1,8 +1,12 @@ +import argparse + from pydantic import ( BaseModel, BaseSettings, ) +from {{cookiecutter.package_name}} import version + class SubModel(BaseModel): foo: str = "bar" @@ -18,7 +22,21 @@ class Config(BaseSettings): env_nested_delimiter = "__" +def parse_arguments(): + parser = argparse.ArgumentParser( + description=""" + {{cookiecutter.package_description}} + """, + ) + parser.add_argument( + "--version", action="version", version=f"package_name {version}" + ) + + return parser.parse_args() + + def entry_point(): + args = parse_arguments() config = Config()