keyboa.button_check

Complexity

Total Complexity

6

Size/Duplication

Total Lines	48
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	ButtonCheck.is_callback_proper_type()	0	8	2
A	ButtonCheck.is_button_data_proper_type()	0	10	2
A	ButtonCheck.is_callback_data_in_limits()	0	8	2

# -*- coding:utf-8 -*-
"""
This module contains all checks for Keyboa Button class parameters
"""
# pylint: disable = C0116

from keyboa.constants import (
    InlineKeyboardButton,
    InlineButtonData,
    callback_data_types,
    MAXIMUM_CBD_LENGTH,
)


class ButtonCheck:
    """
    This class contains all checks for Keyboa Button class parameters
    """

    @staticmethod
    def is_button_data_proper_type(button_data) -> None:
        if not isinstance(button_data, (tuple, dict, str, int)):
            type_error_message = (
                f"Cannot create {InlineKeyboardButton} from {type(button_data)}. "
                f"Please use {InlineButtonData} instead.\n"
                "Probably you specified 'auto_alignment' or 'items_in_line' "
                "parameter for StructuredSequence."
            )
            raise TypeError(type_error_message)

    @staticmethod
    def is_callback_proper_type(callback) -> None:
        if not isinstance(callback, callback_data_types):
            type_error_message = (
                f"Callback cannot be {type(callback)}. "
                f"Only {callback_data_types} allowed."
            )
            raise TypeError(type_error_message)

    @staticmethod
    def is_callback_data_in_limits(callback_data) -> None:
        if len(callback_data.encode()) > MAXIMUM_CBD_LENGTH:
            size_error_message = (
                "The callback data cannot be more than "
                f"64 bytes for one button. Your size is {len(callback_data.encode())}"
            )
            raise ValueError(size_error_message)