Source code for ska.generate_signed_url

import argparse

from .shortcuts import sign_url

__author__ = "Artur Barseghyan <artur.barseghyan@gmail.com>"
__copyright__ = "2013-2023 Artur Barseghyan"
__license__ = "GPL-2.0-only OR LGPL-2.1-or-later"
__all__ = ("main",)




def main():
    """Prints signed URL to console.

    :example:

        python src/ska/generate_signature.py -u http://example.com -au user -sk test

    :example:

        ska-sign-url -u http://example.com -au user -sk test
    """
    parser = argparse.ArgumentParser(
        description="""
    Generates signed URL.
    
    Example:

        ska-sign-url -u http://example.com -au user -sk test
    """
    )

    parser.add_argument(
        "-au",
        "--auth-user",
        dest="auth_user",
        type=str,
        help="`auth_user` value",
        metavar="AUTH_USER",
    )
    parser.add_argument(
        "-sk",
        "--secret-key",
        dest="secret_key",
        type=str,
        help="`secret_key` value",
        metavar="SECRET_KEY",
    )
    parser.add_argument(
        "-vu",
        "--valid-until",
        dest="valid_until",
        type=float,
        help="`valid_until` value",
        metavar="VALID_UNTIL",
    )
    parser.add_argument(
        "-l",
        "--lifetime",
        dest="lifetime",
        type=int,
        help="`lifetime` value",
        metavar="LIFETIME",
    )
    parser.add_argument(
        "-u", "--url", dest="url", type=str, help="URL to sign", metavar="URL"
    )
    parser.add_argument(
        "-s",
        "--suffix",
        dest="suffix",
        type=str,
        help="Added after the `url`.",
        metavar="SUFFIX",
    )
    parser.add_argument(
        "-sp",
        "--signature-param",
        dest="signature_param",
        type=str,
        help="GET param holding the `signature` value",
        metavar="SIGNATURE_PARAM",
    )
    parser.add_argument(
        "-aup",
        "--auth-user-param",
        dest="auth_user_param",
        type=str,
        help="GET param holding the `auth_user` value",
        metavar="AUTH_USER_PARAM",
    )
    parser.add_argument(
        "-vup",
        "--valid-until-param",
        dest="valid_until_param",
        type=str,
        help="GET param holding the `auth_user` value",
        metavar="VALID_UNTIL_PARAM",
    )

    args = parser.parse_args()

    kwargs = {}

    if args.auth_user:
        kwargs.update({"auth_user": args.auth_user})

    if args.secret_key:
        kwargs.update({"secret_key": args.secret_key})

    if args.valid_until:
        kwargs.update({"valid_until": args.valid_until})

    if args.lifetime:
        kwargs.update({"lifetime": args.lifetime})

    if args.url:
        kwargs.update({"url": args.url})

    if args.suffix:
        kwargs.update({"suffix": args.suffix})

    if args.signature_param:
        kwargs.update({"signature_param": args.signature_param})

    if args.auth_user_param:
        kwargs.update({"auth_user_param": args.auth_user_param})

    if args.valid_until_param:
        kwargs.update({"valid_until_param": args.valid_until_param})

    try:
        signed_url = sign_url(**kwargs)
        print(signed_url)
    except Exception as err:
        print(err)