bot_processing.py

import logging
from difflib import SequenceMatcher

from redis.client import Redis

from settings import DEFAULT_ANSWER_RATIO_BORDER

logger = logging.getLogger('log.log')


def get_answer_and_status(user_id: str,
                          platform: int,
                          redis_users: Redis,
                          redis_questions: Redis) -> tuple[str, bool]:
    current_question = redis_users.get(
        f'{platform}_{user_id}_current_question'
    )
    if current_question:
        return current_question.decode('utf-8'), False
    else:
        question = redis_questions.randomkey().decode('utf-8')
        redis_users.set(
            f'{platform}_{user_id}_current_question',
            question
        )
        return question, True


def clarify_answer(answer: str,
                   clarify_params: list = (' - ', '. ', ' ('),
                   clean_params: list = ('...', '"')) -> str:
    for param in clean_params:
        answer = ''.join(answer.split(param))
    for param in clarify_params:
        answer = answer.split(param)[0]
    return answer.strip()


def increase_user_score(user_id: str,
                        platform: int,
                        redis_users: Redis) -> None:
    user_score = redis_users.get(f'{platform}_{user_id}_score')
    if not user_score:
        redis_users.set(f'{platform}_{user_id}_score', 1)
    else:
        redis_users.set(
            f'{platform}_{user_id}_score',
            int(user_score.decode('utf-8')) + 1
        )


def is_correct_answer(correct_answer: str,
                      user_answer: str,
                      answer_ratio_border: float = DEFAULT_ANSWER_RATIO_BORDER) -> bool:
    clarified_answer = clarify_answer(correct_answer)
    answer_ratio = SequenceMatcher(
        None,
        clarified_answer.lower(),
        user_answer.lower()
    ).ratio()
    return answer_ratio >= answer_ratio_border