<reponame>stPhoenix/project_osirius
from linguist.models import Word, GlobalWord, Language, Category
from random import randint, choice
from linguist.utils import LinguistTranslator


class LinguistHQ:
    def __init__(self, student=None):
        self.student = student
        self.langs = Language.objects.all()
        self.global_words = GlobalWord.objects.all()
        if self.student is None:
            raise ValueError("Student can't be None")

    def get_current_language(self):
        return self.langs.get(name=self.student.current_language)

    def add_from_global_word(self, global_word=None, alternative_translation=None):
        global_word = global_word
        word = Word(
            name=global_word.name,
            translation=alternative_translation if alternative_translation is not None else global_word.translation,
            language=self.get_current_language(),
            student=self.student,
            pronunciation=global_word.pronunciation
        )
        word.save()
        word.category_set.set(global_word.category_set.all())

    def search_word(self, word_name=None):
        language = self.get_current_language()
        words = self.global_words.filter(name=word_name, language=language)
        global_word_search = False
        google_translate_search = False
        home_language = self.langs.get(name=self.student.home_language)
        if words.count() == 0:
            translator = LinguistTranslator()
            words = translator.translate(text=word_name, src=language.slug, dest=home_language.slug)
            google_translate_search = True if words.text is not None else False
        else:
            global_word_search = True

        return {'global_word_search': global_word_search, 'google_translate_search': google_translate_search,
                'words': words}

    def add_custom_word(self, **kwargs):
        error = None
        if kwargs['word_name'] is None or kwargs['translation'] is None or kwargs['pronunciation'] is None:
            error = 'You did not choose word or translation or category'
            return error
        category = kwargs['category']
        if kwargs['category'] is None:
            category = Category.objects.get(name="Default")
        word = Word(
            name=kwargs['word_name'],
            translation=kwargs['translation'],
            language=self.get_current_language(),
            student=self.student,
            pronunciation=kwargs['pronunciation']
        )
        word.save()
        word.category_set.add(category)
        word.save()

    def get_student_categories(self):
        categories = []
        for word in self.student.word_set.filter(language=self.get_current_language()):
            for category in word.category_set.all():
                if category.name not in categories:
                    categories.append(category.name)
        return categories

    def get_all_words(self, category):
        return self.student.word_set.filter(category=category, language=self.get_current_language())

    def get_words(self):
        return self.student.word_set.filter(language=self.get_current_language())

    def learn_word(self):
        return self.student.word_set.filter(language=self.get_current_language(), viewed=False)

    def get_learned_words(self, category):
        return self.student.word_set.filter(language=self.get_current_language(), category=category, viewed=True,
                                            played_match=True, played_reversed_match=True, played_typing=True,
                                            played_reversed_typing=True)

    def get_all_learned_words(self):
        return self.student.word_set.filter(language=self.get_current_language(), viewed=True, played_match=True,
                                            played_reversed_match=True, played_typing=True, played_reversed_typing=True)

    def delete_word(self, word):
        word.delete()

    def update_word_translation(self, word, translation):
        word.translation = translation
        word.save()

    def update_viewed_field(self, word, viewed=False):
        word.viewed = viewed
        word.save()

    def update_match_field(self, word, played, reverse=False):
       if reverse is False:
           word.played_match = played
       else:
           word.played_reversed_match = played
       word.save()

    def update_typing_field(self, word, played, reverse=False):
       if reverse is False:
           word.played_typing = played
       else:
           word.played_reversed_typing = played
       word.save()

    def learn_again(self, word):
        self.update_viewed_field(word, False)
        self.update_match_field(word, False, False)
        self.update_match_field(word, False, True)
        self.update_typing_field(word, False, False)
        self.update_typing_field(word, False, True)

    def play_matching(self, reverse=False):
        words = self.student.word_set.filter(language=self.get_current_language())
        words = words.filter(played_match=False) if reverse is False else words.filter(played_reversed_match=False)
        count = words.count()
        if count == 0:
            return 'No words to play matching'
        else:
            return self.create_play_words(words)

    def create_play_words(self, words):
        word = choice(words)
        fake_words = []
        for i in range(0, 3):
            w = choice(words)
            if w not in fake_words and w != word:
                fake_words.append(w)
        length = len(fake_words)
        if length == 0:
           fake_words.append(word)
        else:
            fake_words.insert(randint(0, length), word)
        return {'words': fake_words, 'answer': word}

    def play_typing(self, reverse=False):
        words = self.student.word_set.filter(language=self.get_current_language())
        words = words.filter(played_typing=False) if reverse is False else words.filter(played_reversed_typing=False)
        count = words.count()
        if count == 0:
            word = 'No word to play typing'
        else:
            word = words[randint(0, count-1)]
        return word

from linguist.models import Word, GlobalWord, Language, Category
from random import randint, choice
from linguist.utils import LinguistTranslator


class LinguistHQ:
    def __init__(self, student=None):
        self.student = student
        self.langs = Language.objects.all()
        self.global_words = GlobalWord.objects.all()
        if self.student is None:
            raise ValueError("Student can't be None")

    def get_current_language(self):
        return self.langs.get(name=self.student.current_language)

    def add_from_global_word(self, global_word=None, alternative_translation=None):
        global_word = global_word
        word = Word(
            name=global_word.name,
            translation=alternative_translation if alternative_translation is not None else global_word.translation,
            language=self.get_current_language(),
            student=self.student,
            pronunciation=global_word.pronunciation
        )
        word.save()
        word.category_set.set(global_word.category_set.all())

    def search_word(self, word_name=None):
        language = self.get_current_language()
        words = self.global_words.filter(name=word_name, language=language)
        global_word_search = False
        google_translate_search = False
        home_language = self.langs.get(name=self.student.home_language)
        if words.count() == 0:
            translator = LinguistTranslator()
            words = translator.translate(text=word_name, src=language.slug, dest=home_language.slug)
            google_translate_search = True if words.text is not None else False
        else:
            global_word_search = True

        return {'global_word_search': global_word_search, 'google_translate_search': google_translate_search,
                'words': words}

    def add_custom_word(self, **kwargs):
        error = None
        if kwargs['word_name'] is None or kwargs['translation'] is None or kwargs['pronunciation'] is None:
            error = 'You did not choose word or translation or category'
            return error
        category = kwargs['category']
        if kwargs['category'] is None:
            category = Category.objects.get(name="Default")
        word = Word(
            name=kwargs['word_name'],
            translation=kwargs['translation'],
            language=self.get_current_language(),
            student=self.student,
            pronunciation=kwargs['pronunciation']
        )
        word.save()
        word.category_set.add(category)
        word.save()

    def get_student_categories(self):
        categories = []
        for word in self.student.word_set.filter(language=self.get_current_language()):
            for category in word.category_set.all():
                if category.name not in categories:
                    categories.append(category.name)
        return categories

    def get_all_words(self, category):
        return self.student.word_set.filter(category=category, language=self.get_current_language())

    def get_words(self):
        return self.student.word_set.filter(language=self.get_current_language())

    def learn_word(self):
        return self.student.word_set.filter(language=self.get_current_language(), viewed=False)

    def get_learned_words(self, category):
        return self.student.word_set.filter(language=self.get_current_language(), category=category, viewed=True,
                                            played_match=True, played_reversed_match=True, played_typing=True,
                                            played_reversed_typing=True)

    def get_all_learned_words(self):
        return self.student.word_set.filter(language=self.get_current_language(), viewed=True, played_match=True,
                                            played_reversed_match=True, played_typing=True, played_reversed_typing=True)

    def delete_word(self, word):
        word.delete()

    def update_word_translation(self, word, translation):
        word.translation = translation
        word.save()

    def update_viewed_field(self, word, viewed=False):
        word.viewed = viewed
        word.save()

    def update_match_field(self, word, played, reverse=False):
       if reverse is False:
           word.played_match = played
       else:
           word.played_reversed_match = played
       word.save()

    def update_typing_field(self, word, played, reverse=False):
       if reverse is False:
           word.played_typing = played
       else:
           word.played_reversed_typing = played
       word.save()

    def learn_again(self, word):
        self.update_viewed_field(word, False)
        self.update_match_field(word, False, False)
        self.update_match_field(word, False, True)
        self.update_typing_field(word, False, False)
        self.update_typing_field(word, False, True)

    def play_matching(self, reverse=False):
        words = self.student.word_set.filter(language=self.get_current_language())
        words = words.filter(played_match=False) if reverse is False else words.filter(played_reversed_match=False)
        count = words.count()
        if count == 0:
            return 'No words to play matching'
        else:
            return self.create_play_words(words)

    def create_play_words(self, words):
        word = choice(words)
        fake_words = []
        for i in range(0, 3):
            w = choice(words)
            if w not in fake_words and w != word:
                fake_words.append(w)
        length = len(fake_words)
        if length == 0:
           fake_words.append(word)
        else:
            fake_words.insert(randint(0, length), word)
        return {'words': fake_words, 'answer': word}

    def play_typing(self, reverse=False):
        words = self.student.word_set.filter(language=self.get_current_language())
        words = words.filter(played_typing=False) if reverse is False else words.filter(played_reversed_typing=False)
        count = words.count()
        if count == 0:
            word = 'No word to play typing'
        else:
            word = words[randint(0, count-1)]
        return word