launchpad-request-token.py

#!/usr/bin/python3

"""A script to retrieve a Launchpad token."""

import argparse
import os
import sys

from unittest import mock
from launchpadlib.launchpad import Launchpad

parser = argparse.ArgumentParser(
    description = 'Retrieve a Launchpad auth token to store in plain text.',
)
parser.add_argument('-o', '--output', dest='output_file', metavar='FILE',
                    required=True, help='File to store the generated token')
parser.add_argument('-n', '--name', dest='app_name', metavar='APP',
                    required=True,
                    help='Application name to present to Launchpad')
args = parser.parse_args()


def no_credential():
    print("Can't proceed without Launchpad credential.")
    sys.exit()


with mock.patch.dict(os.environ, {'LP_CREDENTIALS_FILE': args.output_file},
                     clear=True):
    launchpad = Launchpad.login_with(args.app_name, 'production',
                                     version='devel',
                                     credential_save_failed=no_credential)
    if os.path.isfile(args.output_file):
        print('Credentials stored in %s' % args.output_file)
    else:
        raise RuntimeError('Credentials file not found: %s' % args.output_file)