def make_sure_rm(self, tg_id: int): msg = 'Вы уверены, что хотите удалить историю: "{}"'.format(self.name) buttons = [ [('Да', tools.make_call_back(RM_PREFIX, {'is_sure': True}))], [('Нет', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, msg, buttons)
def make_sure_rm(self, tg_id: int): msg = 'Вы уверены, что хотите удалить это сообщение ?' buttons = [ [('Да', tools.make_call_back(RM_PREFIX, {'is_sure': True}))], [('Нет', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, msg, buttons)
def show(self, tg_id: int): msg = f'id: {self.id}\n{self.name}' user_context = mem.UserContext(tg_id) user_context.update_context('chapter_id', str(self.id)) if self.start_message: buttons = [[('Все сообщения', None, '{app_url}chapter/{chapter_uid}'.format( app_url=APP_URL, chapter_uid=hashlib.sha224( bytes(f'{self.id}{tg_id}', 'utf-8')).hexdigest(), ))]] else: buttons = [[('Написать первое сообщение', tools.make_call_back(message.MAKE_PREFIX, { 'is_start_chapter_msg': True, }))]] buttons.extend([ [ ('Переименовать', tools.make_call_back(RENAME_PREFIX)), ('Удалить', tools.make_call_back(RM_PREFIX, {'is_sure': False})), ], [ ('Все главы', tools.make_call_back(story.SHOW_CHAPTERS_PREFIX)), ], ]) tools.send_menu_msg(tg_id, msg, buttons)
def show_chapters(self, tg_id: int): msg = 'Главы истории {}.'.format(self.name) buttons = [] for _chapter in self.chapters: buttons.append([ (_chapter.get('name'), tools.make_call_back(chapter.SHOW_PREFIX, {'chapter_id': _chapter.get('id')})), ]) settings_row = [] if int(_chapter.get('number')) != 0: settings_row.append( ('Up', tools.make_call_back(chapter.UP_PREFIX, { 'chapter_id': _chapter.get('id'), }))) if int(_chapter.get('number')) != len(self.chapters) - 1: settings_row.append( ('Down', tools.make_call_back(chapter.DOWN_PREFIX, { 'chapter_id': _chapter.get('id'), }))) buttons.append(settings_row) buttons.extend([ [('Новая глава', tools.make_call_back(chapter.MAKE_PREFIX))], [('Назад', tools.make_call_back(SHOW_PREFIX))], ]) tools.send_menu_msg(tg_id, msg, buttons, 3)
def show_stories(self): msg = 'Ваши истории.' buttons = [] for _story in self.stories: buttons.append([(_story.name, tools.make_call_back(story.SHOW_PREFIX, {'story_id': _story.id}))]) buttons.append([('Создать историю', tools.make_call_back(story.MAKE_PREFIX))]) print(msg, buttons) tools.send_menu_msg(self.telegram_id, msg, buttons)
def make_sure_rm_btn(self, tg_id: int, btn_id: int): btn = list(btn for btn in self.buttons if btn.get('id') == btn_id)[0] msg = 'Вы уверены, что хотите удалить ответ "{}"'.format( btn.get('text')) buttons = [ [('Да', tools.make_call_back(RM_BUTTON_PREFIX, { 'btn_id': btn_id, 'is_sure': True }))], [('Нет', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, msg, buttons)
def show(self, tg_id: int): msg = self.name user_context = mem.UserContext(tg_id) user_context.update_context('story_id', str(self.id)) buttons = [ [('Посмотреть главы', tools.make_call_back(SHOW_CHAPTERS_PREFIX))], [ ('Переименовать', tools.make_call_back(RENAME_PREFIX)), ('Загрузить реакции', tools.make_call_back(UPLOAD_REACTIONS_PREFIX)), ], [ (f'Задержка {self.base_timeout}c', tools.make_call_back(SET_BASE_TIMEOUT)), (f'Скорость набора {self.k_timeout} знак/мин', tools.make_call_back(SET_K_TIMEOUT)), ], [('Удалить', tools.make_call_back(RM_PREFIX, {'is_sure': False}))], [('Все истории', tools.make_call_back(tg_user.SHOW_STORIES_PREFIX)) ], ] if self.is_reactions: buttons[1].append( ('Реакции', tools.make_call_back(DOWNLOAD_REACTIONS_PREFIX))) tools.send_menu_msg(tg_id, msg, buttons)
def get_line(self, tg_id: int, text: str, prefix: str): buttons = [ [('Назад', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, text, buttons) user_context = mem.UserContext(tg_id) user_context.set_status('wait_line') user_context.set_params({'call_to': prefix})
def get_direct_link(self, tg_id: int): text = 'ID сообщения:' buttons = [ [('Назад', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, text, buttons) user_context = mem.UserContext(tg_id) user_context.set_status('wait_line') user_context.set_params({'call_to': ADD_DIRECT_LINK_PREFIX})
def get_timeout(self, tg_id: int): text = 'Новая задержка:' buttons = [ [('Назад', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, text, buttons) user_context = mem.UserContext(tg_id) user_context.set_status('wait_line') user_context.set_params({'call_to': EDIT_TIMEOUT_PREFIX})
def get_new_btn(self, tg_id: int): text = 'Ответ:' buttons = [ [('Назад', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, text, buttons) user_context = mem.UserContext(tg_id) user_context.set_status('wait_line') user_context.set_params({'call_to': ADD_BUTTON_PREFIX})
def get_new_msg(self, tg_id: int): text = 'Новое сообщение:' buttons = [ [('Назад', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, text, buttons) user_context = mem.UserContext(tg_id) user_context.set_status('wait_line') user_context.set_params({'call_to': EDIT_PREFIX})
def show(self, tg_id: int): text = self.message user_context = mem.UserContext(tg_id) user_context.rm_status() user_context.update_context('message_id', str(self.id)) buttons = [] for btn in self.buttons: buttons.append([(btn['text'], tools.make_call_back(ADD_BUTTON_PREFIX))]) buttons.extend([ [ ('Добавить ответ', tools.make_call_back(ADD_BUTTON_PREFIX)), ], [ ('Редактировать', tools.make_call_back(EDIT_PREFIX)), ('К главе', tools.make_call_back(chapter.SHOW_PREFIX)) ], ]) tools.send_menu_msg(tg_id, text, buttons)
def get_new_btn(tg_id: int, is_start_chapter_msg: bool = False): text = 'Ответ:' buttons = [ [('Назад', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, text, buttons) user_context = mem.UserContext(tg_id) user_context.set_status('wait_line') user_context.set_params({'call_to': MAKE_PREFIX})
def change_reaction(self, tg_id: int, reaction_id: int): if reaction_id: req_data = { 'msg_id': self.id, 'chapter_id': self.chapter_id, 'reaction_id': reaction_id, 'tg_id': tg_id, } edit_msg_resp = json.loads( requests.post( DB_URL.format(item='message', cmd='edit'), json=req_data, ).text) if edit_msg_resp.get('error'): msg = edit_msg_resp.get('error') tools.send_menu_msg(tg_id, msg) else: self.wait_reaction = edit_msg_resp.get('wait_reaction') self.show(tg_id) else: msg = 'Выберите реакцию:' story_reactions = json.loads( requests.post( DB_URL.format(item='story', cmd='get-reactions'), json={ 'story_id': self.story_id, 'tg_id': tg_id, }, ).text) buttons = [] for reaction in story_reactions['reactions']: buttons.append([ (reaction.get('name'), tools.make_call_back(CHANGE_REACTION_PREFIX, { 'reaction_id': reaction.get('id'), })) ]) buttons.append([('Назад', tools.make_call_back(SHOW_PREFIX))]) tools.send_menu_msg(tg_id, msg, buttons)
def get_text_btn(self, tg_id: int, btn_id: int): btn = list(btn for btn in self.buttons if btn.get('id') == btn_id)[0] text = f'Текущий ответ: {btn["text"]}\n\n Новый ответ:' buttons = [ [('Назад', tools.make_call_back(SHOW_PREFIX))], ] tools.send_menu_msg(tg_id, text, buttons) user_context = mem.UserContext(tg_id) user_context.set_status('wait_line') user_context.set_params({ 'call_to': EDIT_BUTTON_PREFIX, 'btn_id': btn_id })
def rm(self, tg_id: int): result = json.loads( requests.post( DB_URL.format(item='story', cmd='rm'), json={ 'story_id': self.id, 'tg_id': tg_id, }, ).text) if result.get('result') == 'ok': msg = 'Успешно.' else: msg = result.get('error') buttons = [ [('Мои истории', tools.make_call_back(tg_user.SHOW_STORIES_PREFIX)) ], ] tools.send_menu_msg(tg_id, msg, buttons)
def rm(self, tg_id: int): result = json.loads( requests.post( DB_URL.format(item='chapter', cmd='rm'), json={ 'story_id': self.story_id, 'tg_id': tg_id, 'chapter_id': self.id }, ).text) if result.get('result') == 'ok': msg = 'Успешно.' else: msg = result.get('error') buttons = [ [('Все главы', tools.make_call_back(story.SHOW_CHAPTERS_PREFIX))], ] tools.send_menu_msg(tg_id, msg, buttons)
def show(self, tg_id: int): if self.content_type == 'text': data = self.message elif self.content_type in ['photo', 'voice', 'video_note']: data = {} media = requests.get(DB_URL.format( item='media', cmd='get/{item_id}', ), params={ 'item_id': hashlib.sha224( bytes(f'{self.media["id"]}{self.id}', 'utf-8')).hexdigest() }).content data['media'] = media data['caption'] = self.message user_context = mem.UserContext(tg_id) user_context.set_status('in_menu') user_context.update_context('message_id', str(self.id)) buttons = [] if not self.link: for btn in self.buttons: if btn['next_message_id']: buttons.append([ (f'{btn["text"]} => К сообщению', tools.make_call_back(SHOW_PREFIX, { 'msg_id': btn['next_message_id'], })) ]) else: buttons.append([ (f'{btn["text"]} => Создать сообщение', tools.make_call_back(MAKE_PREFIX, { 'from_msg_id': self.id, 'from_btn_id': btn['id'], })) ]) buttons.append([ ('Delete', tools.make_call_back(RM_BUTTON_PREFIX, {'btn_id': btn['id']})), ('New text', tools.make_call_back(EDIT_BUTTON_PREFIX, {'btn_id': btn['id']})), ('Add link', tools.make_call_back(ADD_BUTTON_LINK_PREFIX, {'btn_id': btn['id']})), ]) if btn['number'] < len(self.buttons) - 1: buttons[-1].append( ('Down', tools.make_call_back(MOVE_BUTTON_PREFIX, { 'btn_id': btn['id'], 'move': 1, }))) if btn['number'] > 0: buttons[-1].append( ('Up', tools.make_call_back(MOVE_BUTTON_PREFIX, { 'btn_id': btn['id'], 'move': -1, }))) buttons.append([ ('Добавить ответ', tools.make_call_back(ADD_BUTTON_PREFIX)), ]) direct_msg_buttons = [] if self.parrent: direct_msg_buttons.append(('<=', tools.make_call_back( SHOW_PREFIX, {'msg_id': self.parrent}, ))) if not self.buttons: direct_msg_buttons.append( ('Add direct link', tools.make_call_back(ADD_DIRECT_LINK_PREFIX, { 'from_msg_id': self.id, }))) if self.link: direct_msg_buttons.append(('=>', tools.make_call_back( SHOW_PREFIX, {'msg_id': self.link}, ))) else: direct_msg_buttons.append( ('Add direct msg', tools.make_call_back(MAKE_PREFIX, { 'from_msg_id': self.id, }))) buttons.append(direct_msg_buttons) back_from_buttons = [] if self.from_buttons: for from_btn in self.from_buttons: back_from_buttons.append(( f'<= {from_btn["text"][:15]}', tools.make_call_back( SHOW_PREFIX, {'msg_id': from_btn['parrent_message_id']}), )) buttons.append(back_from_buttons) buttons.append([ ('Редактировать', tools.make_call_back(EDIT_PREFIX)), (f'Задержка - {self.timeout}с.', tools.make_call_back(EDIT_TIMEOUT_PREFIX)), (f'Реакция - {self.wait_reaction.get("name")}', tools.make_call_back(CHANGE_REACTION_PREFIX)), ]) buttons.append([ (f'Referal block - {self.referal_block}', tools.make_call_back(EDIT_REFERAL_BLOCK_PREFIX)), ('Все сообщения', None, '{app_url}chapter/{chapter_uid}'.format( app_url=APP_URL, chapter_uid=hashlib.sha224( bytes(f'{self.chapter_id}{tg_id}', 'utf-8')).hexdigest(), )), ]) buttons.append([ ('Удалить', tools.make_call_back(RM_PREFIX)), ('К главе', tools.make_call_back(chapter.SHOW_PREFIX)), ]) tools.send_menu_msg(tg_id, data, buttons, content_type=self.content_type)