PrivacyScanData/Azure/AzureStorage/azure_storage/azure_credentials.py

#!/usr/bin/env python
from azure_storage.methods import create_parent_parser, delete_keyring_credentials, \
    set_account_name, set_connection_string, setup_arguments
from argparse import ArgumentParser, RawTextHelpFormatter
import logging
import sys
import os


def store_credentials(args):
    """
    Run the credentials setting methods
    :param args: type ArgumentParser arguments
    """
    logging.info('Setting Azure storage credentials from system keyring')
    # Set the account name in the keyring
    set_account_name(passphrase=args.passphrase,
                     account_name=args.account_name)
    # Set the connection string in the keyring
    set_connection_string(passphrase=args.passphrase,
                          account_name=args.account_name)


def delete_credentials(args):
    """
    Run the credentials deleting methods
    :param args: type ArgumentParser arguments
    """
    logging.info('Deleting Azure storage credentials from system keyring')
    # Delete the account name in the keyring
    delete_keyring_credentials(passphrase=args.passphrase,
                               account_name=args.passphrase)
    # Delete the connection string in the keyring
    delete_keyring_credentials(passphrase=args.passphrase,
                               account_name=args.account_name)


def cli():
    parser = ArgumentParser(description='Set, modify, or delete Azure storage credentials in the system keyring')
    # Create the parental parser, and the subparser
    subparsers, parent_parser = create_parent_parser(parser=parser,
                                                     container=False)
    # Credentials storing/modifying subparser
    store_subparser = subparsers.add_parser(parents=[parent_parser],
                                            name='store',
                                            description='Store or update Azure storage credentials in the system '
                                                        'keyring',
                                            formatter_class=RawTextHelpFormatter,
                                            help='Store or update Azure storage credentials in the system keyring')
    store_subparser.set_defaults(func=store_credentials)
    # Credentials deleting subparser
    delete_subparser = subparsers.add_parser(parents=[parent_parser],
                                             name='delete',
                                             description='Delete Azure storage credentials in the system keyring',
                                             formatter_class=RawTextHelpFormatter,
                                             help='Delete Azure storage credentials in the system keyring')
    delete_subparser.set_defaults(func=delete_credentials)
    # Set up the arguments, and run the appropriate subparser
    arguments = setup_arguments(parser=parser)
    # Prevent the arguments being printed to the console (they are returned in order for the tests to work)
    sys.stderr = open(os.devnull, 'w')
    return arguments


if __name__ == '__main__':
    cli()