async def end_session(bot, other_user_record, admin_record):
"""End talking session between user and admin.
Cancel session in database, so it will not be loaded anymore.
Send a notification both to admin and user, clear custom parsers
and return.
"""
with bot.db as db:
db['talking_sessions'].update(
dict(admin=admin_record['id'], cancelled=1), ['admin'])
await bot.send_message(chat_id=other_user_record['telegram_id'],
text=bot.get_message('talk',
'user_session_ended',
user_record=other_user_record,
u=get_user(admin_record)))
await bot.send_message(
chat_id=admin_record['telegram_id'],
text=bot.get_message('talk',
'admin_session_ended',
user_record=admin_record,
u=get_user(other_user_record)),
)
for record in (
admin_record,
other_user_record,
):
bot.remove_individual_text_message_handler(record['telegram_id'])
return
async def start_session(bot, other_user_record, admin_record):
"""Start talking session between user and admin.
Register session in database, so it gets loaded before message_loop starts.
Send a notification both to admin and user, set custom parsers and return.
"""
with bot.db as db:
db['talking_sessions'].insert(
dict(user=other_user_record['id'],
admin=admin_record['id'],
cancelled=0))
await bot.send_message(chat_id=other_user_record['telegram_id'],
text=bot.get_message('talk',
'user_warning',
user_record=other_user_record,
u=get_user(admin_record)))
await bot.send_message(chat_id=admin_record['telegram_id'],
text=bot.get_message('talk',
'admin_warning',
user_record=admin_record,
u=get_user(other_user_record)),
reply_markup=make_inline_keyboard([
make_button(
bot.get_message('talk',
'stop',
user_record=admin_record),
prefix='talk:///',
data=['stop', other_user_record['id']])
]))
bot.set_individual_text_message_handler(
await async_wrapper(_forward_to,
sender=other_user_record['telegram_id'],
addressee=admin_record['telegram_id'],
is_admin=False), other_user_record['telegram_id'])
bot.set_individual_text_message_handler(
await async_wrapper(_forward_to,
sender=admin_record['telegram_id'],
addressee=other_user_record['telegram_id'],
is_admin=True), admin_record['telegram_id'])
return
def get_talk_panel(bot, update, user_record=None, text=''):
"""Return text and reply markup of talk panel.
`text` may be:
- `user_id` as string
- `username` as string
- `''` (empty string) for main menu (default)
"""
users = []
if len(text):
with bot.db as db:
if text.isnumeric():
users = list(db['users'].find(id=int(text)))
else:
users = list(
db.query("SELECT * "
"FROM users "
"WHERE COALESCE( "
" first_name || last_name || username, "
" last_name || username, "
" first_name || username, "
" username, "
" first_name || last_name, "
" last_name, "
" first_name "
f") LIKE '%{text}%' "
"ORDER BY LOWER( "
" COALESCE( "
" first_name || last_name || username, "
" last_name || username, "
" first_name || username, "
" username, "
" first_name || last_name, "
" last_name, "
" first_name "
" ) "
") "
"LIMIT 26"))
if len(text) == 0:
text = (bot.get_message('talk',
'help_text',
update=update,
user_record=user_record,
q=escape_html_chars(remove_html_tags(text))))
reply_markup = make_inline_keyboard([
make_button(bot.get_message('talk',
'search_button',
update=update,
user_record=user_record),
prefix='talk:///',
data=['search'])
], 1)
elif len(users) == 0:
text = (bot.get_message('talk',
'user_not_found',
update=update,
user_record=user_record,
q=escape_html_chars(remove_html_tags(text))))
reply_markup = make_inline_keyboard([
make_button(bot.get_message('talk',
'search_button',
update=update,
user_record=user_record),
prefix='talk:///',
data=['search'])
], 1)
else:
text = "{header}\n\n{u}{etc}".format(
header=bot.get_message('talk',
'select_user',
update=update,
user_record=user_record),
u=line_drawing_unordered_list(
[get_user(user) for user in users[:25]]),
etc=('\n\n[...]' if len(users) > 25 else ''))
reply_markup = make_inline_keyboard([
make_button('👤 {u}'.format(u=get_user({
key: val
for key, val in user.items()
if key in ('first_name', 'last_name', 'username')
})),
prefix='talk:///',
data=['select', user['id']]) for user in users[:25]
], 2)
return text, reply_markup