from telegram import ReplyKeyboardMarkup, Bot, Update
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, ConversationHandler, CallbackContext
from credits import bot_token

GENDER = 0
PHOTO = 1
BIO = 2
***** = 3

bot = Bot(token=bot_token)
updater = Updater(token=bot_token)
dispatcher = updater.dispatcher

def start(update, context):
    reply_keyboard = [['Мужчина', 'Женщина']]
    update.message.reply_text('Добрый день! Вас приветствует Крупная Компания! Расскажите о себе, вы мужчина или женщина?', reply_markup=ReplyKeyboardMarkup(reply_keyboard, one_time_keyboard=True))
    return GENDER

def gender(update, context):
    update.message.reply_text('Отлично! Теперь отправьте нам свою фотографию, мы заведем личную карточку, или отправьте /skip если хотите пропустить этот шаг')
    return PHOTO


def photo(update, context):
    user = update.message.from_user
    photo_file = update.message.document.get_file()
    photo_file.download('user_photo.jpg')
    update.message.reply_text('Отлично! Вы сегодня прекрасно выглядите! Теперь отправьте краткое сообщение о себе, или отправьте /skip если хотите пропустить этот шаг')

    return BIO

def skip_photo(update, context):
    update.message.reply_text('Значит без фото! Окей, отправьте краткое сообщение о себе или отправьте /skip.')
    return BIO


def cancel(update, context):
    update.message.reply_text("Хорошего дня!")
    
    return ConversationHandler.END


def bio (update, context):
    update.message.reply_text("Мы получили Ваше резюме!")
    
    return *****

def skip_bio(update, context):
    update.message.reply_text('Значит без резюме! Окей, отправьте видеосообщение или напишите /skip.')
    return *****

def *****(update, context):
    user = update.message.from_user
    video_file = update.message.document.get_file()
    video_file.download(user + "_video.mp4")
    update.message.reply_text('Отлично! Видео принято!')

    return ConversationHandler.END

def skip_video(update, context):
    update.message.reply_text('Значит без видео! Увидимся на очном собеседовании!')
    
    return ConversationHandler.END

start_handler = CommandHandler('start', start)
gender_handler = MessageHandler(Filters.regex('^(Мужчина|Женщина)$'), gender)

photo_handler = MessageHandler(Filters.document.category("image"), photo)
skip_photo_handler = CommandHandler('skip', skip_photo)

bio_handler = MessageHandler(Filters.text & ~Filters.command, bio)
skip_bio_handler = CommandHandler('skip', skip_bio)

video_handler = MessageHandler(Filters.document.category("*****"), *****)
skip_video_handler = CommandHandler('skip', skip_video)

cancel_handler = CommandHandler('cancel', cancel)

conv_handler = ConversationHandler(
    entry_points=[start_handler],
    states={
        GENDER: [gender_handler],
        PHOTO: [photo_handler, skip_photo_handler],
        BIO:   [bio_handler, skip_bio_handler],
        *****: [video_handler, skip_video_handler],
    }, fallbacks=[cancel_handler])

dispatcher.add_handler(conv_handler)

updater.start_polling()
updater.idle()