config.model_telegram.TelegramWord.keyboard_data() - Code Metrics - Inspection of "Extract sub-functions from keyboard_cpx()" - torrua/loglan_bot - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( 18233e...9bad50 )

by torrua

created 2023-03-13 05:55 UTC

config.model_telegram.TelegramWord.keyboard_data() A

↳ Parent: config.model_telegram

Complexity

Conditions

Size

Total Lines	7
Code Lines	7

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
eloc	7
dl	0
loc	7
rs	10
c	0
b	0
f	0
cc	1
nop	1

# -*- coding: utf-8 -*-
"""Model of LOD database for Telegram"""

from typing import List
from collections import defaultdict

from callbaker import callback_from_info
from keyboa import Keyboa
from loglan_core.word import BaseWord
from loglan_core.definition import BaseDefinition
from loglan_core.connect_tables import t_connect_keys
from loglan_core.key import BaseKey
from loglan_core.addons.word_getter import AddonWordGetter
from variables import t, cbd, \
    mark_action, mark_entity, mark_record_id, mark_slice_start, \
    action_predy_send_card, entity_predy, action_predy_kb_cpx_show, action_predy_kb_cpx_hide


class TelegramDefinition(BaseDefinition):
    """Definition class extensions for Telegram"""

    def export(self):
        """
        Convert definition's data to str for sending as a telegram messages
        :return: Adopted for posting in telegram string
        """
        d_usage = f"<b>{self.usage.replace('%', '—')}</b> " if self.usage else ""
        d_grammar = f"({self.slots if self.slots is not None else ''}{self.grammar_code}) "
        d_body = self.body \
            .replace('<', '&#60;').replace('>', '&#62;') \
            .replace('«', '<i>').replace('»', '</i>') \
            .replace('{', '<code>').replace('}', '</code>') \
            .replace('....', '….').replace('...', '…')

        d_case_tags = f" [{self.case_tags}]" if self.case_tags else ""
        return f"{d_usage}{d_grammar}{d_body}{d_case_tags}"


class TelegramWord(BaseWord, AddonWordGetter):
    """Word class extensions for Telegram"""

    def export(self, session) -> str:
        """
        Convert word's data to str for sending as a telegram messages
        :return: List of str with technical info, definitions, used_in part
        """

        # Word
        list_of_afx = ["" + w.name for w in self.affixes]
        w_affixes = f" ({' '.join(list_of_afx)})" if list_of_afx else ""
        w_match = self.match + " " if self.match else ""
        w_year = "'" + str(self.year.year)[-2:] + " "
        w_origin_x = " = " + self.origin_x if self.origin_x else ""
        w_orig = "\n<i>&#60;" + self.origin + w_origin_x + "&#62;</i>" \
            if self.origin or w_origin_x else ""
        w_authors = '/'.join([a.abbreviation for a in self.authors]) + " "
        w_type = self.type.type + " "
        w_rank = self.rank + " " if self.rank else ""
        word_str = f"<b>{self.name}</b>{w_affixes}," \
                   f"\n{w_match}{w_type}{w_authors}{w_year}{w_rank}{w_orig}"

        # Definitions TODO maybe extract from method
        definitions_str = "\n\n".join([d.export() for d in self.get_definitions(session=session)])
        return f"{word_str}\n\n{definitions_str}"

    def get_definitions(self, session) -> List[TelegramDefinition]:
        """
        Get all definitions of the word
        :param session: Session
        :return: List of Definition objects ordered by position
        """
        return session.query(TelegramDefinition).filter(BaseDefinition.word_id == self.id)\
            .order_by(BaseDefinition.position.asc()).all()

    @classmethod
    def translation_by_key(cls, session, request: str, language: str = None) -> str:
        """
        We get information about loglan words by key in a foreign language
        :param session: Session
        :param request: Requested string
        :param language: Key language
        :return: Search results string formatted for sending to Telegram
        """
        words = session.query(cls.name, TelegramDefinition).\
            join(BaseDefinition).\
            join(t_connect_keys).\
            join(BaseKey).\
            filter(BaseKey.word==request).\
            filter(BaseKey.language==language).\
            order_by(cls.id, BaseDefinition.position).all()

        result = defaultdict(list)

        for word in words:
            name, definition = word
            result[name].append(definition.export())

        new = '\n'
        word_items = [f"/{word_name},\n{new.join(definitions)}\n"
                         for word_name, definitions in result.items()]
        return new.join(word_items).strip()


    def keyboard_navi(self, index_start, index_end, delimiter):

            text_arrow_back = "\U0000276E" * 2
            text_arrow_forward = "\U0000276F" * 2
            button_back, button_forward = None, None

            common_data = {
                mark_entity: entity_predy,
                mark_action: action_predy_kb_cpx_show,
                mark_record_id: self.id,
            }

            if index_start != 0:
                cbd_predy_kb_cpx_back = {
                    **common_data, mark_slice_start: index_start - delimiter, }
                button_back = {
                    t: text_arrow_back,
                    cbd: callback_from_info(cbd_predy_kb_cpx_back)}

            if index_end != len(self.complexes):
                cbd_predy_kb_cpx_forward = {
                    **common_data, mark_slice_start: index_end, }
                button_forward = {
                    t: text_arrow_forward,
                    cbd: callback_from_info(cbd_predy_kb_cpx_forward)}

            nav_row = [b for b in [button_back, button_forward] if b]
            return Keyboa(nav_row, items_in_row=2)()

    def keyboard_hide(self, total_number_of_complexes):
            text_cpx_hide = f"Hide Complex{'es' if total_number_of_complexes > 1 else ''}"
            cbd_predy_kb_cpx_hide = {
                mark_entity: entity_predy,
                mark_action: action_predy_kb_cpx_hide,
                mark_record_id: self.id, }
            button_predy_kb_cpx_hide = [{
                t: text_cpx_hide, cbd: callback_from_info(cbd_predy_kb_cpx_hide)}, ]
            return Keyboa(button_predy_kb_cpx_hide)()

    def keyboard_show(self, total_number_of_complexes: int):
            text_cpx_show = f"Show Complex{'es' if total_number_of_complexes > 1 else ''} ({total_number_of_complexes})"
            cbd_predy_kb_cpx_show = {
                mark_entity: entity_predy,
                mark_action: action_predy_kb_cpx_show,
                mark_record_id: self.id, }
            button_show = [{
                t: text_cpx_show, cbd: callback_from_info(cbd_predy_kb_cpx_show)}, ]
            return Keyboa.combine((Keyboa(button_show)(), kb_close()))

    @staticmethod
    def get_delimiter(total_number_of_complexes: int):
            from bot import MIN_NUMBER_OF_BUTTONS
            allowed_range = list(range(MIN_NUMBER_OF_BUTTONS, MIN_NUMBER_OF_BUTTONS + 11))
            lst = [(total_number_of_complexes % i, i) for i in allowed_range]
            delimiter = min(lst, key=lambda x: abs(x[0] - MIN_NUMBER_OF_BUTTONS))[1]
            for i in lst:
                if i[0] == 0:
                    delimiter = i[1]
                    break
            return delimiter
    @staticmethod
    def keyboard_data(current_complexes):
        cpx_items = [{t: cpx.name, cbd: callback_from_info({
            mark_entity: entity_predy,
            mark_action: action_predy_send_card,
            mark_record_id: cpx.id, })} for cpx in current_complexes]
        return Keyboa(items=cpx_items, alignment=True, items_in_row=4)()

    def keyboard_cpx(self, show_list: bool = False, slice_start: int = 0):
        """
        :return:
        """

        total_num_of_cpx = len(self.complexes)

        if not total_num_of_cpx:
            return kb_close()

        if not show_list:
            return self.keyboard_show(total_num_of_cpx)

        current_delimiter = self.get_delimiter(total_num_of_cpx)

        kb_cpx_hide = self.keyboard_hide(total_num_of_cpx)

        last_allowed_element = slice_start + current_delimiter
        slice_end = last_allowed_element if last_allowed_element < total_num_of_cpx else total_num_of_cpx

        current_cpx_set = self.complexes[slice_start:slice_end]
        kb_cpx_data = self.keyboard_data(current_cpx_set)

        kb_cpx_nav = None

        if total_num_of_cpx > current_delimiter:
            kb_cpx_nav = self.keyboard_navi(slice_start, slice_end, current_delimiter)

        kb_combo = (kb_cpx_hide, kb_cpx_data, kb_cpx_nav, kb_close())

        return Keyboa.combine(kb_combo)

    def send_card_to_user(self, session, bot, user_id):
        bot.send_message(
            chat_id=user_id,
            text=self.export(session),
            reply_markup=self.keyboard_cpx())

    @classmethod
    def by_request(cls, session, request) -> list:
        return [cls.get_by_id(session, request), ] if isinstance(request, int) else cls.by_name(session, request).all()

def kb_close():
    return Keyboa({t: "Close", cbd: "close"})()


1			# -- coding: utf-8 --
2			"""Model of LOD database for Telegram"""
3
4			from typing import List
5			from collections import defaultdict
6
7			from callbaker import callback_from_info
8			from keyboa import Keyboa
9			from loglan_core.word import BaseWord
10			from loglan_core.definition import BaseDefinition
11			from loglan_core.connect_tables import t_connect_keys
12			from loglan_core.key import BaseKey
13			from loglan_core.addons.word_getter import AddonWordGetter
14			from variables import t, cbd, \
15			mark_action, mark_entity, mark_record_id, mark_slice_start, \
16			action_predy_send_card, entity_predy, action_predy_kb_cpx_show, action_predy_kb_cpx_hide
17
18
19			class TelegramDefinition(BaseDefinition):
20			"""Definition class extensions for Telegram"""
21
22			def export(self):
23			"""
24			Convert definition's data to str for sending as a telegram messages
25			:return: Adopted for posting in telegram string
26			"""
27			d_usage = f"<b>{self.usage.replace('%', '—')}</b> " if self.usage else ""
28			d_grammar = f"({self.slots if self.slots is not None else ''}{self.grammar_code}) "
29			d_body = self.body \
30			.replace('<', '<').replace('>', '>') \
31			.replace('«', '<i>').replace('»', '</i>') \
32			.replace('{', '<code>').replace('}', '</code>') \
33			.replace('....', '….').replace('...', '…')
34
35			d_case_tags = f" [{self.case_tags}]" if self.case_tags else ""
36			return f"{d_usage}{d_grammar}{d_body}{d_case_tags}"
37
38
39			class TelegramWord(BaseWord, AddonWordGetter):
40			"""Word class extensions for Telegram"""
41
42			def export(self, session) -> str:
43			"""
44			Convert word's data to str for sending as a telegram messages
45			:return: List of str with technical info, definitions, used_in part
46			"""
47
48			# Word
49			list_of_afx = ["" + w.name for w in self.affixes]
50			w_affixes = f" ({' '.join(list_of_afx)})" if list_of_afx else ""
51			w_match = self.match + " " if self.match else ""
52			w_year = "'" + str(self.year.year)[-2:] + " "
53			w_origin_x = " = " + self.origin_x if self.origin_x else ""
54			w_orig = "\n<i><" + self.origin + w_origin_x + "></i>" \
55			if self.origin or w_origin_x else ""
56			w_authors = '/'.join([a.abbreviation for a in self.authors]) + " "
57			w_type = self.type.type + " "
58			w_rank = self.rank + " " if self.rank else ""
59			word_str = f"<b>{self.name}</b>{w_affixes}," \
60			f"\n{w_match}{w_type}{w_authors}{w_year}{w_rank}{w_orig}"
61
62			# Definitions TODO maybe extract from method
63			definitions_str = "\n\n".join([d.export() for d in self.get_definitions(session=session)])
64			return f"{word_str}\n\n{definitions_str}"
65
66			def get_definitions(self, session) -> List[TelegramDefinition]:
67			"""
68			Get all definitions of the word
69			:param session: Session
70			:return: List of Definition objects ordered by position
71			"""
72			return session.query(TelegramDefinition).filter(BaseDefinition.word_id == self.id)\
73			.order_by(BaseDefinition.position.asc()).all()
74
75			@classmethod
76			def translation_by_key(cls, session, request: str, language: str = None) -> str:
77			"""
78			We get information about loglan words by key in a foreign language
79			:param session: Session
80			:param request: Requested string
81			:param language: Key language
82			:return: Search results string formatted for sending to Telegram
83			"""
84			words = session.query(cls.name, TelegramDefinition).\
85			join(BaseDefinition).\
86			join(t_connect_keys).\
87			join(BaseKey).\
88			filter(BaseKey.word==request).\
89			filter(BaseKey.language==language).\
90			order_by(cls.id, BaseDefinition.position).all()
91
92			result = defaultdict(list)
93
94			for word in words:
95			name, definition = word
96			result[name].append(definition.export())
97
98			new = '\n'
99			word_items = [f"/{word_name},\n{new.join(definitions)}\n"
100			for word_name, definitions in result.items()]
101			return new.join(word_items).strip()
102
103
104			def keyboard_navi(self, index_start, index_end, delimiter):
105
106			text_arrow_back = "\U0000276E" * 2
107			text_arrow_forward = "\U0000276F" * 2
108			button_back, button_forward = None, None
109
110			common_data = {
111			mark_entity: entity_predy,
112			mark_action: action_predy_kb_cpx_show,
113			mark_record_id: self.id,
114			}
115
116			if index_start != 0:
117			cbd_predy_kb_cpx_back = {
118			**common_data, mark_slice_start: index_start - delimiter, }
119			button_back = {
120			t: text_arrow_back,
121			cbd: callback_from_info(cbd_predy_kb_cpx_back)}
122
123			if index_end != len(self.complexes):
124			cbd_predy_kb_cpx_forward = {
125			**common_data, mark_slice_start: index_end, }
126			button_forward = {
127			t: text_arrow_forward,
128			cbd: callback_from_info(cbd_predy_kb_cpx_forward)}
129
130			nav_row = [b for b in [button_back, button_forward] if b]
131			return Keyboa(nav_row, items_in_row=2)()
132
133			def keyboard_hide(self, total_number_of_complexes):
134			text_cpx_hide = f"Hide Complex{'es' if total_number_of_complexes > 1 else ''}"
135			cbd_predy_kb_cpx_hide = {
136			mark_entity: entity_predy,
137			mark_action: action_predy_kb_cpx_hide,
138			mark_record_id: self.id, }
139			button_predy_kb_cpx_hide = [{
140			t: text_cpx_hide, cbd: callback_from_info(cbd_predy_kb_cpx_hide)}, ]
141			return Keyboa(button_predy_kb_cpx_hide)()
142
143			def keyboard_show(self, total_number_of_complexes: int):
144			text_cpx_show = f"Show Complex{'es' if total_number_of_complexes > 1 else ''} ({total_number_of_complexes})"
145			cbd_predy_kb_cpx_show = {
146			mark_entity: entity_predy,
147			mark_action: action_predy_kb_cpx_show,
148			mark_record_id: self.id, }
149			button_show = [{
150			t: text_cpx_show, cbd: callback_from_info(cbd_predy_kb_cpx_show)}, ]
151			return Keyboa.combine((Keyboa(button_show)(), kb_close()))
152
153			@staticmethod
154			def get_delimiter(total_number_of_complexes: int):
155			from bot import MIN_NUMBER_OF_BUTTONS
156			allowed_range = list(range(MIN_NUMBER_OF_BUTTONS, MIN_NUMBER_OF_BUTTONS + 11))
157			lst = [(total_number_of_complexes % i, i) for i in allowed_range]
158			delimiter = min(lst, key=lambda x: abs(x[0] - MIN_NUMBER_OF_BUTTONS))[1]
159			for i in lst:
160			if i[0] == 0:
161			delimiter = i[1]
162			break
163			return delimiter
164			@staticmethod
165			def keyboard_data(current_complexes):
166			cpx_items = [{t: cpx.name, cbd: callback_from_info({
167			mark_entity: entity_predy,
168			mark_action: action_predy_send_card,
169			mark_record_id: cpx.id, })} for cpx in current_complexes]
170			return Keyboa(items=cpx_items, alignment=True, items_in_row=4)()
171
172			def keyboard_cpx(self, show_list: bool = False, slice_start: int = 0):
173			"""
174			:return:
175			"""
176
177			total_num_of_cpx = len(self.complexes)
178
179			if not total_num_of_cpx:
180			return kb_close()
181
182			if not show_list:
183			return self.keyboard_show(total_num_of_cpx)
184
185			current_delimiter = self.get_delimiter(total_num_of_cpx)
186
187			kb_cpx_hide = self.keyboard_hide(total_num_of_cpx)
188
189			last_allowed_element = slice_start + current_delimiter
190			slice_end = last_allowed_element if last_allowed_element < total_num_of_cpx else total_num_of_cpx
191
192			current_cpx_set = self.complexes[slice_start:slice_end]
193			kb_cpx_data = self.keyboard_data(current_cpx_set)
194
195			kb_cpx_nav = None
196
197			if total_num_of_cpx > current_delimiter:
198			kb_cpx_nav = self.keyboard_navi(slice_start, slice_end, current_delimiter)
199
200			kb_combo = (kb_cpx_hide, kb_cpx_data, kb_cpx_nav, kb_close())
201
202			return Keyboa.combine(kb_combo)
203
204			def send_card_to_user(self, session, bot, user_id):
205			bot.send_message(
206			chat_id=user_id,
207			text=self.export(session),
208			reply_markup=self.keyboard_cpx())
209
210			@classmethod
211			def by_request(cls, session, request) -> list:
212			return [cls.get_by_id(session, request), ] if isinstance(request, int) else cls.by_name(session, request).all()
213
214			def kb_close():
215			return Keyboa({t: "Close", cbd: "close"})()
216

torrua / loglan_bot

Push — master ( 18233e...9bad50 )

config.model_telegram.TelegramWord.keyboard_data() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like