DR PyPi URL builder

build_pypi_url.py

#!/Users/kayibal/miniconda3/bin/python
import boto3
import click
from urllib.parse import urlparse, urlunparse

def pypi_url(env):
    ssm = boto3.client("ssm")
    url = ssm.get_parameter(Name=f"/CodeBuild/PyPi/{env}/url", WithDecryption=True)[
        "Parameter"
    ]["Value"]
    un = ssm.get_parameter(Name=f"/CodeBuild/PyPi/{env}/un", WithDecryption=True)[
        "Parameter"
    ]["Value"]
    pw = ssm.get_parameter(Name=f"/CodeBuild/PyPi/{env}/pw", WithDecryption=True)[
        "Parameter"
    ]["Value"]

    parts = urlparse(url)
    pypi_url = urlunparse(
        [
            parts.scheme,
            f"{un}:{pw}@" + parts.netloc,
            parts.path,
            parts.params,
            parts.query,
            parts.fragment,
        ]
    )
    return pypi_url

@click.command()
@click.option('--env', default="Dev")
def cli(env):
    print(pypi_url(env))


if __name__ == '__main__':
    cli()