140 lines
5.9 KiB
Python
140 lines
5.9 KiB
Python
import datetime
|
||
import re
|
||
import logging
|
||
from sqlalchemy import select
|
||
from app.database import async_session
|
||
from app.models import BotUser, BotConversation, BotMessage
|
||
from app.max_api import max_api
|
||
from app.config_reader import config_reader
|
||
from app.keyboards import contact_keyboard, consent_keyboard, return_to_consent_keyboard
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def _matches_any(text: str, words: list) -> bool:
|
||
text_words = set(text.split())
|
||
for w in words:
|
||
if w in text_words:
|
||
return True
|
||
if len(w) > 3 and w in text:
|
||
return True
|
||
return False
|
||
|
||
|
||
async def handle_consent_callback(user_id: int, conv_id: int, callback_id: str) -> None:
|
||
if callback_id == "consent_yes":
|
||
await handle_consent_yes(user_id, conv_id)
|
||
elif callback_id == "consent_no":
|
||
await handle_consent_no(user_id, conv_id)
|
||
|
||
|
||
async def handle_consent_response(user_id: int, conv_id: int, text: str) -> None:
|
||
text_lower = text.strip().lower()
|
||
consent_words = ["да", "даю", "согласен", "согласна", "yes", "ok", "хорошо", "даю согласие"]
|
||
refuse_words = ["нет", "не даю", "отказ", "no", "не согласен", "не согласна"]
|
||
|
||
if _matches_any(text_lower, consent_words):
|
||
await handle_consent_yes(user_id, conv_id)
|
||
elif _matches_any(text_lower, refuse_words):
|
||
await handle_consent_no(user_id, conv_id)
|
||
else:
|
||
await max_api.send_message(
|
||
user_id,
|
||
"Пожалуйста, дайте чёткий ответ — даёте ли вы согласие на обработку "
|
||
"персональных данных? Это необходимо для обработки вашего запроса.",
|
||
attachments=consent_keyboard(),
|
||
)
|
||
|
||
|
||
async def handle_consent_yes(user_id: int, conv_id: int) -> None:
|
||
async with async_session() as db:
|
||
result = await db.execute(select(BotUser).where(BotUser.id == user_id))
|
||
user = result.scalar_one_or_none()
|
||
if user:
|
||
user.consent_given = True
|
||
user.consent_date = datetime.datetime.utcnow()
|
||
|
||
result = await db.execute(
|
||
select(BotConversation).where(BotConversation.id == conv_id)
|
||
)
|
||
conv = result.scalar_one_or_none()
|
||
if conv:
|
||
conv.state = "awaiting_contact"
|
||
|
||
await db.commit()
|
||
|
||
try:
|
||
await max_api.send_message(
|
||
user_id,
|
||
"Спасибо! Поделитесь, пожалуйста, вашим контактом, чтобы мы могли "
|
||
"связаться с вами.\n\n"
|
||
"Нажмите кнопку «Поделиться контактом» или введите номер телефона / email вручную.",
|
||
attachments=contact_keyboard(),
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Failed to send consent_yes message: {e}")
|
||
|
||
|
||
async def handle_consent_no(user_id: int, conv_id: int) -> None:
|
||
async with async_session() as db:
|
||
result = await db.execute(
|
||
select(BotConversation).where(BotConversation.id == conv_id)
|
||
)
|
||
conv = result.scalar_one_or_none()
|
||
if conv:
|
||
conv.state = "consent_refused"
|
||
await db.commit()
|
||
|
||
phones = config_reader.get_phones()
|
||
email = config_reader.get_site_mail()
|
||
|
||
phone_text = "\n".join(f"📞 {phone}" for phone, _ in phones)
|
||
|
||
try:
|
||
await max_api.send_message(
|
||
user_id,
|
||
f"Без вашего согласия мы не можем обработать обращение.\n\n"
|
||
f"Вы можете позвонить нам:\n{phone_text}\n\n"
|
||
f"📧 {email}\n\n"
|
||
f"Если передумаете и захотите дать согласие — просто напишите об этом.",
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Failed to send consent_no message: {e}")
|
||
|
||
|
||
async def handle_refused_again(user_id: int, conv_id: int, text: str) -> None:
|
||
text_lower = text.strip().lower()
|
||
consent_words = ["да", "даю", "согласен", "согласна", "yes", "ok", "хорошо", "даю согласие", "передумал", "согласен дать"]
|
||
refuse_words = ["нет", "не даю", "не хочу", "отказ", "отмена", "no", "cancel", "я отказался"]
|
||
|
||
if _matches_any(text_lower, consent_words):
|
||
await handle_consent_yes(user_id, conv_id)
|
||
elif _matches_any(text_lower, refuse_words):
|
||
phones = config_reader.get_phones()
|
||
email = config_reader.get_site_mail()
|
||
phone_text = "\n".join(f"📞 {phone}" for phone, _ in phones)
|
||
try:
|
||
await max_api.send_message(
|
||
user_id,
|
||
f"Хорошо, без согласия мы не можем обработать обращение.\n\n"
|
||
f"Вы можете позвонить нам:\n{phone_text}\n\n"
|
||
f"📧 {email}\n\n"
|
||
f"Если передумаете — напишите «Даю согласие».",
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Failed to send refused_again message: {e}")
|
||
else:
|
||
phones = config_reader.get_phones()
|
||
email = config_reader.get_site_mail()
|
||
phone_text = "\n".join(f"📞 {phone}" for phone, _ in phones)
|
||
try:
|
||
await max_api.send_message(
|
||
user_id,
|
||
f"Хорошо, без согласия мы не можем обработать обращение.\n\n"
|
||
f"Вы можете позвонить нам:\n{phone_text}\n\n"
|
||
f"📧 {email}\n\n"
|
||
f"Если передумаете — напишите «Даю согласие».",
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Failed to send refused_again message: {e}")
|