"""passbook nexus_upload management command"""
from getpass import getpass

import requests
from django.core.management.base import BaseCommand


class Command(BaseCommand):
    """Upload debian package to nexus repository"""

    url = None
    user = None
    password = None

    def add_arguments(self, parser):
        parser.add_argument(
            '--repo',
            action='store',
            help='Repository to upload to',
            required=True)
        parser.add_argument(
            '--url',
            action='store',
            help='Nexus root URL',
            required=True)
        parser.add_argument(
            '--user',
            action='store',
            help='Username to use for Nexus upload',
            required=True)
        parser.add_argument(
            '--method',
            action='store',
            nargs='?',
            const='post',
            choices=['post', 'put'],
            help=('Method used for uploading files to nexus. '
                  'Apt repositories use post, Helm uses put.'),
            required=True)
        parser.add_argument(
            '--password',
            action='store',
            help=("Password to use for Nexus upload. "
                  "If parameter not given, we'll interactively ask"))
        # Positional arguments
        parser.add_argument('file', nargs='+', type=str)

    def handle(self, *args, **options):
        """Upload debian package to nexus repository"""
        if options.get('password') is None:
            options['password'] = getpass()
        responses = {}
        url = 'https://%(url)s/repository/%(repo)s//' % options
        method = options.get('method')
        exit_code = 0
        for file in options.get('file'):
            if method == 'post':
                responses[file] = requests.post(url, data=open(file, mode='rb'),
                                                auth=(options.get('user'), options.get('password')))
            else:
                responses[file] = requests.put(url+file, data=open(file, mode='rb'),
                                               auth=(options.get('user'), options.get('password')))
        self.stdout.write('Upload results:\n')
        sep = '-' * 60
        self.stdout.write('%s\n' % sep)
        for path, response in responses.items():
            self.stdout.write('%-55s: %d\n' % (path, response.status_code))
            if response.status_code >= 400:
                exit_code = 1
        self.stdout.write('%s\n' % sep)
        exit(exit_code)