Source code for ska.generate_signed_url

from __future__ import print_function

import argparse

from .shortcuts import sign_url

__title__ = 'ska.generate_signed_url'
__author__ = 'Artur Barseghyan <artur.barseghyan@gmail.com>'
__copyright__ = '2013-2019 Artur Barseghyan'
__license__ = 'GPL 2.0/LGPL 2.1'
__all__ = ('main',)


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

    :example:

        python src/ska/generate_signature.py -au user -sk test
    """
    parser = argparse.ArgumentParser(description="""
    Generates signed URL.
    """)

    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)


if __name__ == "__main__":
    main()