def handle_authorization(message):
AUTH_TOKENS[hash(message.chat.id)] = flickr_api.auth.AuthHandler()
url = AUTH_TOKENS[hash(message.chat.id)].get_authorization_url("write")
BOT.send_message(
message.chat.id, "Follow this link, log in and accept the query:" +
url + "\nThen copy everything from the next web-page and send me.")
BOT.register_next_step_handler(message, handle_verifier)
def handle_help(message):
BOT.send_message(
message.chat.id,
"If you want to log in your account through me (for example to upload photos), "
"write /auth.\n"
"If you want to find smth, write /search.\n"
"If you want to upload a photo to your account, write /upload.\n")
def handle_find_person(message):
person = None
try:
person = flickr_api.Person.findByUserName(message.text)
except flickrerrors.FlickrAPIError:
pass
if person is None:
try:
person = flickr_api.Person.findByEmail(message.text)
except flickrerrors.FlickrAPIError:
pass
if person is None:
try:
person = flickr_api.Person.findByUrl(message.text)
except flickrerrors.FlickrAPIError:
BOT.send_message(message.chat.id,
"There is no users with this name/email/URL.")
return
if USERS.get(hash(message.chat.id)) is None:
photos = person.getPublicPhotos()
else:
photos = person.getPhotos()
WALKERS[hash(message.chat.id)] = photos
FIRST_NUMS[hash(message.chat.id)] = 0
handle_print_3_photos_from_walker(message)
def handle_search(message):
markup = types.ReplyKeyboardMarkup(row_width=1, one_time_keyboard=True)
itembtn1 = types.KeyboardButton('Photos in search')
itembtn2 = types.KeyboardButton('User')
markup.add(itembtn1, itembtn2)
BOT.send_message(message.chat.id,
"Choose what you want to find.",
reply_markup=markup)
BOT.register_next_step_handler(message, handle_search_what)
def ask_question(chat_id):
keyboard = telebot.types.InlineKeyboardMarkup()
key_yes = telebot.types.InlineKeyboardButton(text='Да',
callback_data='yes')
keyboard.add(key_yes)
key_no = telebot.types.InlineKeyboardButton(text='Нет', callback_data='no')
keyboard.add(key_no)
BOT.send_message(chat_id,
text='Добавить ещё рецептов?',
reply_markup=keyboard)
def handle_search_what(message):
if message.text == "User":
BOT.send_message(message.chat.id,
"Write a user name, email or URL of the person.")
BOT.register_next_step_handler(message, handle_find_person)
elif message.text == "Photos in search":
BOT.send_message(message.chat.id, "Write a tag or text to search by.")
BOT.register_next_step_handler(message, handle_find_photo)
else:
BOT.send_message(
message.chat.id,
"Wrong input, are you sure you're following my instructions?"
"If you want to continue, do everything from the start.")
def get_ingredients(message):
ingredients = message.text
url = "https://www.food2fork.com/api/search?key={FOOD_TOKEN}&q={ingredients}"
url = url.format(FOOD_TOKEN=FOOD_TOKEN, ingredients=ingredients)
text = requests.get(url).json()
dish_index = 0
if not text["recipes"]:
answer_pattern = 'Извините, не удалось найти не одного рецепта с данными ингридиентами, {name} {surname}'
answer = answer_pattern.format(name=PERSON_INF[(message.chat.id,
'name')],
surname=PERSON_INF[(message.chat.id,
'surname')])
BOT.send_message(message.chat.id, answer)
for item in text["recipes"]:
dish_index += 1
BOT.send_message(message.chat.id, item['source_url'])
if dish_index % 5 == 0:
ask_question(message.chat.id)
global FLAG_ENOUGH
FLAG_ENOUGH = 2
while FLAG_ENOUGH == 2:
time.sleep(1)
if FLAG_ENOUGH:
BOT.send_message(
message.chat.id, 'Рады что смогли помочь, {} {}'.format(
PERSON_INF[(message.chat.id, 'name')],
PERSON_INF[(message.chat.id, 'surname')]))
break
if not FLAG_ENOUGH:
BOT.send_message(
message.chat.id,
'Просим прощения, рецепты закончились, {} {}'.format(
PERSON_INF[(message.chat.id, 'name')],
PERSON_INF[(message.chat.id, 'surname')]))
def handle_set_title(message):
try:
flickr_api.set_auth_handler("./users_data/{}/auth_token.dat".format(
hash(message.chat.id)))
USERS[hash(message.chat.id)] = flickr_api.test.login()
flickr_api.upload(photo_file="Photo",
title=message.text,
photo_file_data=urllib.request.urlopen(URLS[hash(
message.chat.id)]))
except Exception:
BOT.send_message(message.chat.id,
"Oops! Error! Try later or try another photo.")
return
BOT.send_message(message.chat.id, "Photo is uploaded!")
def get_receipe(message):
if PERSON_INF.get((message.chat.id, 'name')) is None:
BOT.send_message(message.chat.id, 'Вы ещё не прошли регистрацию')
BOT.send_message(message.chat.id,
'Вам необходимо зарегестрироваться (напишите /reg)')
else:
BOT.send_message(
message.chat.id,
'Введите интересующие вас ингридиенты через запятую:')
BOT.register_next_step_handler(message, get_ingredients)
def handle_verifier(message):
text = message.text
begin, end = text.find("<oauth_verifier>"), text.find("</oauth_verifier>")
if begin != -1 and end != -1:
AUTH_TOKENS[hash(message.chat.id)].set_verifier(
text[begin + OAUTH_VERIFIER_LENGTH:end])
flickr_api.set_auth_handler(AUTH_TOKENS[hash(message.chat.id)])
USERS[hash(message.chat.id)] = flickr_api.test.login()
os.system("mkdir ./users_data/{}".format(hash(message.chat.id)))
AUTH_TOKENS[hash(message.chat.id)].save(
"./users_data/{}/auth_token.dat".format(hash(message.chat.id)))
BOT.send_message(
message.chat.id, "You're authorised, let's start using me!"
"\n(/help and /info are still available.)")
else:
BOT.send_message(
message.chat.id,
"Wrong input, are you sure you're following my instructions?"
"Let's try from the beginning")
def handle_print_3_photos_from_walker(message):
if message.text == 'Stop':
WALKERS[hash(message.chat.id)] = None
FIRST_NUMS[hash(message.chat.id)] = None
return
required_photo, photo_in_walker = FIRST_NUMS[hash(message.chat.id)], 0
for photo in WALKERS[hash(message.chat.id)]:
if photo_in_walker < required_photo:
photo_in_walker += 1
continue
else:
photo_in_walker += 3
size = None
for size_std in SIZES:
tmp_size = photo.getSizes().get(size_std)
if tmp_size is not None:
if int(tmp_size["width"]) * int(
tmp_size["height"]) < MAX_PICTURE_SIZE:
size = size_std
break
try:
BOT.send_photo(message.chat.id, photo["sizes"][size]["source"])
except Exception:
continue
required_photo += 1
if required_photo == FIRST_NUMS[hash(message.chat.id)] + 3:
break
FIRST_NUMS[hash(message.chat.id)] += 3
markup = types.ReplyKeyboardMarkup(row_width=1, one_time_keyboard=True)
itembtn1 = types.KeyboardButton('Stop')
itembtn2 = types.KeyboardButton('3 more photos')
markup.add(itembtn1, itembtn2)
BOT.send_message(message.chat.id, "More or stop??", reply_markup=markup)
BOT.register_next_step_handler(message, handle_print_3_photos_from_walker)
def handle_help(message):
if PERSON_INF.get((message.chat.id, 'name')) is None:
BOT.send_message(message.chat.id,
'Вам необходимо зарегестрироваться (напишите /reg)')
else:
BOT.send_message(message.chat.id,
'Поздравляем, вы уже успешно зарегестрированы\n')
BOT.send_message(
message.chat.id, 'Вот список доступных вам команд:\n \
/get_recipe - позволяет находить рецепты')
def handle_upload(message):
if USERS.get(hash(message.chat.id)) is None:
try:
flickr_api.set_auth_handler(
"./users_data/{}/auth_token.dat".format(hash(message.chat.id)))
USERS[hash(message.chat.id)] = flickr_api.test.login()
except Exception:
BOT.send_message(
message.chat.id,
"You're not authorized! Do it through /auth command")
return
BOT.send_message(message.chat.id, "Send me your photo.")
BOT.register_next_step_handler(message, handle_upload_photo)
def handle_upload_photo(message):
if not message.photo:
BOT.send_message(
message.chat.id,
"There is no photo here! Try from the beginning and send me a photo!"
)
return
photo = message.photo[-1]
photo_id = photo.file_id
url = "https://api.telegram.org/bot{}/getFile?file_id={}".format(
BOT_TOKEN, photo_id)
photo_path = json.loads(
urllib.request.urlopen(url).read())['result']['file_path']
url = "https://api.telegram.org/file/bot{}/{}".format(
BOT_TOKEN, photo_path)
URLS[hash(message.chat.id)] = url
BOT.send_message(message.chat.id, "Write a title.")
BOT.register_next_step_handler(message, handle_set_title)
def handle_start(message):
BOT.send_message(message.chat.id,
'Привет! Я - бот для еды. Поговори со мной.')
def handle_reg(message):
if PERSON_INF.get((message.chat.id, 'name')) is not None:
BOT.send_message(message.chat.id, 'Вы уже прошли регистрацию')
else:
BOT.send_message(message.chat.id, "Как у тебя имя?")
BOT.register_next_step_handler(message, get_name)
def handle_start(message):
BOT.send_message(
message.chat.id,
"Hi! I'm the FlickrBot and you can contact with Flickr through me.\n"
"To know more write /help or /info")
def handle_text_message(message):
BOT.send_message(message.chat.id, 'Некорректная команда')
def get_surname(message):
PERSON_INF[(message.chat.id, 'surname')] = message.text
BOT.send_message(message.chat.id, 'Спасибо за регистрацию')
def get_name(message):
PERSON_INF[(message.chat.id, 'name')] = message.text
BOT.send_message(message.chat.id, 'Какая у тебя фамилия?')
BOT.register_next_step_handler(message, get_surname)