import re

from parser import command, argument, call

FAIL = '\033[91m'
SUCCESS = '\033[92m'
RESET   = '\033[0m'


@argument('module', nargs='?', default='')
@command('test', help='runs the tests')
def test(args=None):
    """
    Runs the tests.
    """
    module = getattr(args, 'module', '')
    if module == '':
        module = 'tests'
    else:
        module = 'tests.{}'.format(module)
    return call("DJANGO_SETTINGS_MODULE='tests.settings' coverage run "
                "./manage.py test {}".format(module))


@argument('--plain', action='store_true')
@command('coverage', help='Runs all tests and builds the coverage html files')
def coverage(args=None, plain=None):
    """
    Runs the tests and creates a coverage report.

    By default it creates a html report. With the argument --plain, it creates
    a plain report and fails under a certain amount of untested lines.
    """
    if plain is None:
        plain = getattr(args, 'plain', False)
    if plain:
        return call('coverage report -m --fail-under=80')
    else:
        return call('coverage html')


@command('check', help='Checks for pep8 errors in openslides and tests')
def check(args=None):
    """
    Checks for pep8 and other code styling conventions.
    """
    return call('flake8 --max-line-length=150 --statistics openslides tests')


@command('travis', help='Runs the code that travis does')
def travis(args=None):
    """
    Runs all commands that travis tests.
    """
    return_codes = []
    with open('.travis.yml') as f:
        script_lines = False
        for line in (line.strip() for line in f.readlines()):
            if line == 'script:':
                script_lines = True
                continue
            if not script_lines or not line:
                continue

            match = re.search(r'"(.*)"', line)
            print('Run: %s' % match.group(1))
            return_code = call(match.group(1))
            return_codes.append(return_code)
            if return_code:
                print(FAIL + 'fail!\n' + RESET)
            else:
                print(SUCCESS + 'success!\n' + RESET)

    # Retuns True if one command exited with a different statuscode then 1
    return bool(list(filter(bool, return_codes)))


@argument('-r', '--requirements', nargs='?',
          default='requirements_production.txt')
@command('min_requirements',
         help='Prints a pip line to install the minimum supported versions of '
              'the requirements.')
def min_requirements(args=None):
    """
    Prints a pip install command to install the minimal supported versions of a
    requirement file.

    Uses requirements_production.txt by default.
    """

    import pip

    def get_lowest_versions(requirements_file):
        for line in pip.req.parse_requirements(requirements_file, session=pip.download.PipSession()):
            yield '%s==%s' % (line.req.key, line.req.specs[0][1])

    print('pip install %s' % ' '.join(get_lowest_versions(args.requirements)))


@command('clear',
         help='Deletes unneeded files and folders')
def clean(args=None):
    """
    Deletes all .pyc and .orig files and empty folders.
    """
    call('find -name "*.pyc" -delete')
    call('find -name "*.orig" -delete')
    call('find -type d -empty -delete')


@command('isort',
         help='Sorts all imports in all python files.')
def isort(args=None):
    return call('isort --recursive openslides tests')