# -*- coding: utf-8 -*-
from __future__ import unicode_literals
import datetime
import django
import os
import shutil
import sys
from django.test.utils import get_runner

from base.console_scripts.admin import DJANGO_MAIN_MODULE
from base.tests.utils import mkdtemp


class DjangoEnvironment(object):
    @staticmethod
    def _install_djangoproject(path):
        cmd = 'django-test-admin setup "{}"'.format(path)
        os.system(cmd)

    def __init__(self, path=None, remove_after=True):
        self.path = path

        self._remove_after = remove_after
        self._original_sys_path = None
        self._modified_sys_path = None

    def __enter__(self):
        if self.path is None:
            prefix = 'testrun-{datetime}-'.format(
                datetime=datetime.datetime.now().strftime('%Y%m%d-%H%M')
            )
            self.path = mkdtemp(prefix=prefix)

        self._install_djangoproject(self.path)

        self._original_sys_path = sys.path
        sys.path.append(self.path)
        self._modified_sys_path = sys.path

        os.environ['DJANGO_SETTINGS_MODULE'] = '{}.settings'.format(DJANGO_MAIN_MODULE)
        django.setup()

        from django.conf import settings
        self.settings = settings

        return self

    def __exit__(self, *args):
        if self._modified_sys_path is not None and self._original_sys_path is not None:
            if sys.path == self._modified_sys_path:
                sys.path = self._original_sys_path
        if self._remove_after:
            shutil.rmtree(self.path)


class TestSuite(object):
    @staticmethod
    def run():
        tests = ['base']
        test_tags = None
        exclude_test_tags = None

        with DjangoEnvironment() as env:
            test_runner_class = get_runner(env.settings)
            test_runner = test_runner_class(tags=test_tags, exclude_tags=exclude_test_tags)
            failures = test_runner.run_tests(tests)

        return bool(failures)

    def __call__(self):
        sys.exit(self.run())