198 lines
9.8 KiB
Python
198 lines
9.8 KiB
Python
from telegram import Update, ReplyKeyboardMarkup, ReplyKeyboardRemove
|
||
from telegram.ext import (
|
||
Application, CommandHandler, MessageHandler, filters, ConversationHandler,
|
||
)
|
||
import os
|
||
import logging
|
||
|
||
# Настройка логирования
|
||
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# Токен бота
|
||
TOKEN = "7693731498:AAG1HUwrk_vSpnaS5ivsqREOBsEP_wC6RV4"
|
||
|
||
# Состояния для диалога
|
||
WELCOME, NAME, TITLE, PHONE, EMAIL, WORK, SUBMITTED = range(7)
|
||
|
||
# Путь к папке на рабочем столе
|
||
DESKTOP_PATH = os.path.join(os.path.expanduser("~"), "Desktop")
|
||
FOLDER_NAME = "Ульяновск.Искусство"
|
||
FOLDER_PATH = os.path.join(DESKTOP_PATH, FOLDER_NAME)
|
||
|
||
# Создание папки на рабочем столе, если она не существует
|
||
if not os.path.exists(FOLDER_PATH):
|
||
os.makedirs(FOLDER_PATH)
|
||
|
||
# Обработчик команды /start
|
||
async def start(update: Update, context):
|
||
logger.info(f"Пользователь {update.effective_user.id} начал диалог.")
|
||
await update.message.reply_text(
|
||
"Привет! 👋\n"
|
||
"Мы рады приветствовать вас в нашем боте, посвящённом городу Ульяновск!\n\n"
|
||
"Ульяновск — это город с богатой историей, красивыми памятниками и уютными улочками. "
|
||
"Мы проводим конкурс на лучшую фотографию нашего города. Присоединяйтесь!\n\n"
|
||
"Чтобы принять участие, нажмите кнопку ниже:",
|
||
reply_markup=ReplyKeyboardMarkup([["Добавить работу"]], one_time_keyboard=True, resize_keyboard=True),
|
||
)
|
||
return WELCOME
|
||
|
||
# Обработчик нажатия на кнопку "Добавить работу"
|
||
async def add_work(update: Update, context):
|
||
logger.info(f"Пользователь {update.effective_user.id} нажал на кнопку 'Добавить работу'.")
|
||
await update.message.reply_text(
|
||
"Отлично! 🎉\n"
|
||
"Для начала введите ваше имя и фамилию:",
|
||
reply_markup=ReplyKeyboardRemove(),
|
||
)
|
||
return NAME
|
||
|
||
# Обработчик ввода имени и фамилии
|
||
async def get_name(update: Update, context):
|
||
name = update.message.text
|
||
context.user_data["name"] = name
|
||
logger.info(f"Пользователь {update.effective_user.id} ввел имя: {name}")
|
||
await update.message.reply_text("Теперь введите название вашей работы:")
|
||
return TITLE
|
||
|
||
# Обработчик ввода названия работы
|
||
async def get_title(update: Update, context):
|
||
title = update.message.text
|
||
context.user_data["title"] = title
|
||
logger.info(f"Пользователь {update.effective_user.id} ввел название работы: {title}")
|
||
await update.message.reply_text("Введите ваш номер телефона:")
|
||
return PHONE
|
||
|
||
# Обработчик ввода номера телефона
|
||
async def get_phone(update: Update, context):
|
||
phone = update.message.text
|
||
context.user_data["phone"] = phone
|
||
logger.info(f"Пользователь {update.effective_user.id} ввел номер телефона: {phone}")
|
||
await update.message.reply_text("Введите вашу электронную почту:")
|
||
return EMAIL
|
||
|
||
# Обработчик ввода электронной почты
|
||
async def get_email(update: Update, context):
|
||
email = update.message.text
|
||
context.user_data["email"] = email
|
||
logger.info(f"Пользователь {update.effective_user.id} ввел email: {email}")
|
||
await update.message.reply_text("Отправьте вашу работу (файл в формате JPEG или PNG):")
|
||
return WORK
|
||
|
||
# Обработчик отправки файла
|
||
async def get_work(update: Update, context):
|
||
try:
|
||
if update.message.photo:
|
||
photo = update.message.photo[-1]
|
||
name = context.user_data["name"]
|
||
title = context.user_data["title"]
|
||
phone = context.user_data["phone"]
|
||
email = context.user_data["email"]
|
||
|
||
# Создаем папку участника
|
||
participant_folder = os.path.join(FOLDER_PATH, name)
|
||
if not os.path.exists(participant_folder):
|
||
os.makedirs(participant_folder)
|
||
|
||
# Сохраняем файл в папку участника
|
||
file_path = os.path.join(participant_folder, f"{title}.jpg")
|
||
file = await photo.get_file()
|
||
await file.download_to_drive(file_path)
|
||
|
||
# Сохраняем контактные данные в текстовый файл
|
||
contact_info = f"Имя: {name}\nНомер телефона: {phone}\nЭлектронная почта: {email}"
|
||
contact_file_path = os.path.join(participant_folder, "contact_info.txt")
|
||
with open(contact_file_path, "w", encoding="utf-8") as file:
|
||
file.write(contact_info)
|
||
|
||
await update.message.reply_text(
|
||
f"Спасибо, {name}! Ваша работа '{title}' сохранена.\n\n"
|
||
"Удачи в конкурсе! 🍀"
|
||
)
|
||
return SUBMITTED
|
||
|
||
elif update.message.document:
|
||
document = update.message.document
|
||
if document.mime_type in ["image/jpeg", "image/png"] and document.file_name.lower().endswith((".jpg", ".jpeg", ".png")):
|
||
name = context.user_data["name"]
|
||
title = context.user_data["title"]
|
||
phone = context.user_data["phone"]
|
||
email = context.user_data["email"]
|
||
|
||
# Создаем папку участника
|
||
participant_folder = os.path.join(FOLDER_PATH, name)
|
||
if not os.path.exists(participant_folder):
|
||
os.makedirs(participant_folder)
|
||
|
||
# Сохраняем файл в папку участника
|
||
file_path = os.path.join(participant_folder, f"{title}.{document.file_name.split('.')[-1]}")
|
||
file = await document.get_file()
|
||
await file.download_to_drive(file_path)
|
||
|
||
# Сохраняем контактные данные в текстовый файл
|
||
contact_info = f"Имя: {name}\nНомер телефона: {phone}\nЭлектронная почта: {email}"
|
||
contact_file_path = os.path.join(participant_folder, "contact_info.txt")
|
||
with open(contact_file_path, "w", encoding="utf-8") as file:
|
||
file.write(contact_info)
|
||
|
||
await update.message.reply_text(
|
||
f"Спасибо, {name}! Ваша работа '{title}' сохранена.\n\n"
|
||
"Удачи в конкурсе! 🍀"
|
||
)
|
||
return SUBMITTED
|
||
else:
|
||
await update.message.reply_text("Пожалуйста, отправьте файл в формате JPEG или PNG.")
|
||
return WORK
|
||
|
||
else:
|
||
await update.message.reply_text("Пожалуйста, отправьте только файл в формате JPEG или PNG.")
|
||
return WORK
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при загрузке файла: {e}")
|
||
await update.message.reply_text("Произошла ошибка при загрузке файла. Пожалуйста, попробуйте снова.")
|
||
return WORK
|
||
|
||
# Обработчик состояния SUBMITTED
|
||
async def submitted(update: Update, context):
|
||
reply_keyboard = [["Посмотреть свою работу", "Изменить свою работу"], ["Удалить работу", "Отказаться от участия"]]
|
||
await update.message.reply_text(
|
||
"Вы уже отправили свою работу. Выберите действие:",
|
||
reply_markup=ReplyKeyboardMarkup(reply_keyboard, one_time_keyboard=True),
|
||
)
|
||
return SUBMITTED
|
||
|
||
# Обработчик команды /cancel
|
||
async def cancel(update: Update, context):
|
||
await update.message.reply_text("Отправка работы отменена. Вы можете начать снова, отправив команду /start.")
|
||
return ConversationHandler.END
|
||
|
||
# Основная функция
|
||
def main():
|
||
application = Application.builder().token(TOKEN).build()
|
||
|
||
# Определяем ConversationHandler
|
||
conv_handler = ConversationHandler(
|
||
entry_points=[CommandHandler("start", start)],
|
||
states={
|
||
WELCOME: [MessageHandler(filters.TEXT & ~filters.COMMAND, add_work)],
|
||
NAME: [MessageHandler(filters.TEXT & ~filters.COMMAND, get_name)],
|
||
TITLE: [MessageHandler(filters.TEXT & ~filters.COMMAND, get_title)],
|
||
PHONE: [MessageHandler(filters.TEXT & ~filters.COMMAND, get_phone)],
|
||
EMAIL: [MessageHandler(filters.TEXT & ~filters.COMMAND, get_email)],
|
||
WORK: [
|
||
MessageHandler(filters.PHOTO | filters.Document.ALL, get_work),
|
||
MessageHandler(filters.TEXT & ~filters.COMMAND, lambda update, context: update.message.reply_text("Пожалуйста, отправьте файл.") or WORK),
|
||
],
|
||
SUBMITTED: [MessageHandler(filters.TEXT & ~filters.COMMAND, submitted)],
|
||
},
|
||
fallbacks=[CommandHandler("cancel", cancel)],
|
||
)
|
||
|
||
# Добавляем обработчик
|
||
application.add_handler(conv_handler)
|
||
|
||
# Запуск бота
|
||
application.run_polling()
|
||
|
||
if __name__ == "__main__":
|
||
main() |