This commit is contained in:
2024-06-17 10:40:49 +03:00
parent c50fc58257
commit c2255c2166
18 changed files with 1742 additions and 179 deletions

185
old/klining_order.py Normal file
View File

@@ -0,0 +1,185 @@
import logging
import sqlite3
import asyncio
from aiogram import Bot, Dispatcher, types
from aiogram.contrib.middlewares.logging import LoggingMiddleware
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
# Устанавливаем уровень логов
logging.basicConfig(level=logging.INFO)
# Подключаемся к базе данных SQLite
conn = sqlite3.connect("answers.db")
cursor = conn.cursor()
# Создание таблицы для хранения ответов
cursor.execute("""
CREATE TABLE IF NOT EXISTS answers (
user_id INTEGER,
question_id INTEGER,
answer TEXT
)
""")
conn.commit()
# Заменить на свой токен
API_TOKEN = '7472030348:AAGI53nX-ON-WBmEhd_qBC6EnZsHOqp_2kE'
GROUP_ID = '-1001961537659'
# Инициализация бота и диспетчера
bot = Bot(token=API_TOKEN)
dp = Dispatcher(bot)
dp.middleware.setup(LoggingMiddleware())
# Словарь для хранения вопросов и ответов
questions = {
1: "Как вас зовут?",
2: "Укажите номер телефона для связи",
3: "Укажите район, улицу, дом",
4: "Какая уборка нужна, влажная или сухая?",
5: "На какое время?",
6: "Оплата наличными или картой?"
}
answer_map = {
'wet_cleaning': 'Влажная уборка',
'dry_cleaning': 'Сухая уборка',
'morning_time': 'Утро',
'day_time': 'День',
'evening_time': 'Вечер',
'cash_payment': 'Наличные',
'card_payment': 'Карта'
}
# Обработчик команды /start
@dp.message_handler(commands=['start'])
async def start(message: types.Message):
await message.answer("Привет! Я задам тебе 6 вопросов. Давай начнем.")
await ask_question(message.chat.id, 1)
# Функция для задания вопроса
async def ask_question(user_id, question_id):
if question_id in [4, 5, 6]:
if question_id == 4:
keyboard = InlineKeyboardMarkup(inline_keyboard=[
[InlineKeyboardButton(text="Влажная", callback_data="wet_cleaning")],
[InlineKeyboardButton(text="Сухая", callback_data="dry_cleaning")]
])
elif question_id == 5:
keyboard = InlineKeyboardMarkup(inline_keyboard=[
[InlineKeyboardButton(text="Утро", callback_data="morning_time")],
[InlineKeyboardButton(text="День", callback_data="day_time")],
[InlineKeyboardButton(text="Вечер", callback_data="evening_time")]
])
elif question_id == 6:
keyboard = InlineKeyboardMarkup(inline_keyboard=[
[InlineKeyboardButton(text="Наличные", callback_data="cash_payment")],
[InlineKeyboardButton(text="Карта", callback_data="card_payment")]
])
await bot.send_message(user_id, text=questions[question_id], reply_markup=keyboard)
else:
await bot.send_message(user_id, text=questions[question_id])
# Обработчик текстовых ответов на первые три вопроса
@dp.message_handler(lambda message: message.text not in ["Переписать", "Отправить"])
async def handle_text_answer(message: types.Message):
user_id = message.chat.id
cursor.execute("SELECT MAX(question_id) FROM answers WHERE user_id=?", (user_id,))
prev_question_id = cursor.fetchone()[0]
if prev_question_id is None:
question_id = 1
else:
question_id = prev_question_id + 1
answer = message.text
await save_answer(user_id, question_id, answer)
if question_id < 6:
await ask_question(user_id, question_id + 1)
else:
await show_confirmation_options(user_id)
# Обработчик inline-кнопок
@dp.callback_query_handler(lambda query: query.data in answer_map.keys())
async def handle_callback_answer(query: types.CallbackQuery):
user_id = query.from_user.id
cursor.execute("SELECT MAX(question_id) FROM answers WHERE user_id=?", (user_id,))
prev_question_id = cursor.fetchone()[0]
if prev_question_id is None:
question_id = 1
else:
question_id = prev_question_id + 1
await save_answer(user_id, question_id, answer_map[query.data])
await query.answer()
if question_id < 6:
await ask_question(user_id, question_id + 1)
else:
await show_confirmation_options(user_id)
# Функция для сохранения ответа в базе данных
async def save_answer(user_id, question_id, answer):
cursor.execute("INSERT INTO answers (user_id, question_id, answer) VALUES (?, ?, ?)", (user_id, question_id, answer))
conn.commit()
# Функция для отображения опций подтверждения
async def show_confirmation_options(user_id):
cursor.execute("SELECT answer FROM answers WHERE user_id=? ORDER BY question_id", (user_id,))
answers = cursor.fetchall()
answer_text = "\n".join([f"{questions[i+1]}: {answers[i][0]}" for i in range(6)])
await bot.send_message(user_id, text=f"Ваши ответы:\n\n{answer_text}")
keyboard = InlineKeyboardMarkup(inline_keyboard=[
[InlineKeyboardButton(text="Отправить", callback_data="send_answers")],
[InlineKeyboardButton(text="Переписать", callback_data="rewrite_answers")]
])
await bot.send_message(user_id, "Проверьте свои ответы. Выберите действие:", reply_markup=keyboard)
# Обработчик выбора действия подтверждения
@dp.callback_query_handler(lambda query: query.data in ['send_answers', 'rewrite_answers'])
async def handle_confirmation(query: types.CallbackQuery):
user_id = query.from_user.id
if query.data == "send_answers":
await send_answers_to_group(user_id)
elif query.data == "rewrite_answers":
await rewrite_answers(user_id)
# Функция для отправки ответов в группу
async def send_answers_to_group(user_id):
cursor.execute("SELECT * FROM answers WHERE user_id=?", (user_id,))
answers = cursor.fetchall()
if answers:
answer_text = '\n'.join([f"{questions[ans[1]]}: {ans[2]}" for ans in answers])
# Очищаем таблицу с ответами
cursor.execute("DELETE FROM answers WHERE user_id=?", (user_id,))
conn.commit()
# Отправляем ответы в группу
await bot.send_message(GROUP_ID, answer_text)
await bot.send_message(user_id, "Ваши ответы отправлены.")
else:
await bot.send_message(user_id, "Ответов нет")
# Функция для переписывания ответов
async def rewrite_answers(user_id):
# Очищаем таблицу с ответами
cursor.execute("DELETE FROM answers WHERE user_id=?", (user_id,))
conn.commit()
await bot.send_message(user_id, "Ваши ответы удалены. Начнем сначала.")
await ask_question(user_id, 1)
if __name__ == '__main__':
loop = asyncio.get_event_loop()
try:
loop.create_task(dp.start_polling())
loop.run_forever()
except KeyboardInterrupt:
pass
finally:
conn.close()