xtex/akari-bot
Archived
1
0
Fork 0
Code Issues Pull requests Activity

Update wiki.py

Browse source
This commit is contained in:
yzhh 2024-01-02 16:16:27 +08:00
parent 857c566d7f
commit 46d911c035
1 changed files with 19 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
modules/wiki/wiki.py
View file

@ -233,29 +233,27 @@ async def query_pages(session: Union[Bot.MessageSession, QueryInfo], title: Unio
if plain_slice:
msg_list.append(Plain('\n'.join(plain_slice)))
if r.invalid_section and r.info.in_allowlist:
if isinstance(session, Bot.MessageSession):
if isinstance(session, Bot.MessageSession) and session.Feature.image and r.sections:
i_msg_lst = []
session_data = [[str(i + 1), r.sections[i]] for i in range(len(r.sections))]
i_msg_lst.append(Plain(session.locale.t('wiki.message.invalid_section')))
i_msg_lst.append(Image(await
image_table_render(
ImageTable(session_data,
['ID',
session.locale.t('wiki.message.section')]))))
if session.Feature.image:
i_msg_lst = []
session_data = [[str(i + 1), r.sections[i]] for i in range(len(r.sections))]
i_msg_lst.append(Plain(session.locale.t('wiki.message.invalid_section')))
i_msg_lst.append(Image(await
image_table_render(
ImageTable(session_data,
['ID',
session.locale.t('wiki.message.section')]))))
async def _callback(msg: Bot.MessageSession):
display = msg.as_display(text_only=True)
if display.isdigit():
display = int(display)
if display <= len(r.sections):
r.selected_section = display - 1
await query_pages(session, title=r.title + '#' + r.sections[display - 1])
async def _callback(msg: Bot.MessageSession):
display = msg.as_display(text_only=True)
if display.isdigit():
display = int(display)
if display <= len(r.sections):
r.selected_section = display - 1
await query_pages(session, title=r.title + '#' + r.sections[display - 1])
await session.send_message(i_msg_lst, callback=_callback)
else:
msg_list.append(Plain(session.locale.t('wiki.message.invalid_section.prompt')))
await session.send_message(i_msg_lst, callback=_callback)
else:
msg_list.append(Plain(session.locale.t('wiki.message.invalid_section.prompt')))
else:
plain_slice = []
wait_plain_slice = []