utils/validation.py

import os
from dataclasses import dataclass
from pathlib import Path

from telethon import TelegramClient


@dataclass
class ValidationResult:
    is_valid: bool
    is_error: bool = False
    message: str = ''


class Validator:
    env_vars = ['API_ID', 'API_HASH', 'PHONE_NUMBER']

    @staticmethod
    def validate_env_file(env_filename: str = '.env') -> ValidationResult:
        if not Path(env_filename).is_file():
            log_msg = 'Отсутвует файл .env'
            return ValidationResult(is_valid=False, message=log_msg)
        return ValidationResult(is_valid=True)

    @staticmethod
    def validate_env_var(var_name: str, error_msg: str | None = None) -> ValidationResult:
        if os.getenv(var_name) is None:
            return ValidationResult(is_valid=False, message=error_msg)
        return ValidationResult(is_valid=True)

    @classmethod
    def validate_env_id(cls) -> ValidationResult:
        return cls.validate_env_var('API_ID', 'Отсутствует переменная API_ID')

    @classmethod
    def validate_env_hash(cls) -> ValidationResult:
        return cls.validate_env_var('API_HASH', 'Отсутствует переменная API_HASH')

    @classmethod
    def validate_env_phone_number(cls) -> ValidationResult:
        return cls.validate_env_var('PHONE_NUMBER', 'Отсутствует переменная PHONE_NUMBER')

    @classmethod
    def validate_env_vars(cls) -> ValidationResult:
        if all([cls.validate_env_var(var).is_valid for var in cls.env_vars]):
            return ValidationResult(is_valid=True)
        return ValidationResult(is_valid=False)

    @staticmethod
    async def validate_auth(client: TelegramClient) -> ValidationResult:
        try:
            if not client.is_connected():
                await client.connect()
            is_user_authorized = await client.is_user_authorized()
            if not is_user_authorized:
                log_msg = 'Клиент не авторизован'
                return ValidationResult(is_valid=False, message=log_msg)
            return ValidationResult(is_valid=True)
        except Exception as ex:
            log_msg = f'Ошибка при подключении клиента, код ошибки: {ex}'
            return ValidationResult(is_valid=False, is_error=True, message=log_msg)
        finally:
            if client.is_connected():
                await client.disconnect()