Azure credential helper

azure_token.py

"""
"""
from argparse import ArgumentParser

from artifacts_keyring import CredentialProvider  # type: ignore


def get_artifact_token(organization: str, feed: str) -> str:
    """Get credentials for an azure artifacts feed."""
    provider = CredentialProvider()
    _, token = provider.get_credentials(
        f"https://pkgs.dev.azure.com/{organization}/_packaging/{feed}/pypi/simple/"
    )
    return token


if __name__ == "__main__":
    parser = ArgumentParser(
        prog="azure_token",
        description="Retrieve tokens for Azure Artifact Feeds."
    )
    parser.add_argument(
        "--organization",
        "-o",
        help="The organization to which the Artifact Feed belongs.",
        required=True,
    )
    parser.add_argument(
        "--feed",
        "-f",
        help="The artifact feed for which to retrieve the token.",
        required=True,
    )
    args = parser.parse_args()
    print(get_artifact_credentials(organization=args.organization, feed=args.feed))