def command_add_step1(m: telebot.types.Message):
cid = m.chat.id
if not m.text:
bot.send_message(cid, msg('ru', 'waiting_for_text'))
return
bot.send_message(cid, msg('ru', 'insert_translate'))
users[cid] = {'step': 2, 'word': m.text}
def command_find_step1(m: telebot.types.Message):
cid = m.chat.id
if not m.text:
bot.send_message(cid, msg('ru', 'waiting_for_text'))
return
results = list(db.get_search_results(cid, m.text))
if not results:
bot.send_message(cid, msg('ru', 'no_results'))
users[cid] = {'step': 0}
return
for result in results:
bot.send_message(cid,
f'*{result["word"]}* - {result["definition"]}',
parse_mode='markdown')
users[cid] = {'step': 0}
def command_add(m: telebot.types.Message):
cid = m.chat.id
word = command_value('add', m.text)
if word:
m.text = word
command_add_step1(m)
return
users[cid] = {'step': 1}
bot.send_message(cid, msg('ru', 'insert_word'), reply_markup=clear_markup)
def command_find(m: telebot.types.Message):
cid = m.chat.id
search_query = command_value('find', m.text)
if search_query:
m.text = search_query
command_find_step1(m)
return
users[cid] = {'step': 3}
bot.send_message(cid,
msg('ru', 'insert_search_query'),
reply_markup=clear_markup)
def command_add_step2(m: telebot.types.Message):
cid = m.chat.id
if not m.text:
bot.send_message(cid, msg('ru', 'waiting_for_text'))
return
db.insert_word(cid, {'word': users[cid]['word'], 'definition': m.text})
bot.send_message(cid,
f'{emoji("open_book")} *{users[cid]["word"]}* - {m.text}',
parse_mode='markdown')
# reply_markup=find_markup)
users[cid] = {'step': 0}
def command_start(m: telebot.types.Message):
cid = m.chat.id
bot.send_message(cid, text=msg('ru', 'help'))
# reply_markup=find_markup)
if cid not in users:
users[cid] = {'step': 0}
from bot.msg import msg, emoji
from bot.db import db
import telebot
from telebot import apihelper, types
import re
if PROXY_ON:
apihelper.proxy = PROXY
bot = telebot.TeleBot(TOKEN)
users = {}
find_markup = types.InlineKeyboardMarkup(row_width=1)
find_markup.add(
types.InlineKeyboardButton(text=msg('ru', 'find_button'),
switch_inline_query_current_chat=''))
add_markup = types.ReplyKeyboardMarkup(row_width=1, resize_keyboard=True)
add_markup.add(types.KeyboardButton(text=msg('ru', 'add_button')))
clear_markup = types.ReplyKeyboardRemove(selective=False)
def step_equals(user_id, step):
user = users.get(user_id)
if user:
return user['step'] == step
def command_value(command, text):