سورس ربات نظرسنجی و رضایت مشتری بله

# -*- coding: utf-8 -*-
# ------------------------------------------------------------
#  ربات نظرسنجی و رضایت مشتری بله
#  طراحی و کدنویسی: سروش مقدمی — فیلتور (filtori.com)
# ------------------------------------------------------------
#  بعد از خرید یا دریافت خدمت، لینک ربات رو برای مشتری بفرست:
#   ۱) مشتری با دکمه‌های شیشه‌ای از ۱ تا ۵ ستاره امتیاز می‌ده
#   ۲) ربات (اختیاری) نظر متنی‌اش رو هم می‌پرسه
#   ۳) همه‌چیز در feedback.csv ذخیره می‌شه (با اکسل باز می‌شه)
#   ۴) امتیازهای پایین (۱ و ۲) همون لحظه برای مدیر ارسال می‌شن
#      تا قبل از اینکه مشتری ناراضی بره، باهاش تماس بگیری
#   ۵) مدیر با دستور /stats میانگین و تعداد امتیازها رو می‌بینه
#
#  آموزش کامل:
#     https://filtori.com/bale-bot-source/feedback-bot/
#  سوال یا راهنمایی:  https://filtori.com/contact/
#  نسخهٔ اختصاصی با داشبورد و گزارش ماهانه؟  09122874862
# ------------------------------------------------------------

import csv
import os
import time
from datetime import datetime

import requests

TOKEN = "توکن_ربات_خودت_را_اینجا_بگذار"
BASE = f"https://tapi.bale.ai/bot{TOKEN}"

# chat_id مدیر؛ ربات رو اجرا کن و /start بزن تا در ترمینال چاپ بشه.
ADMIN_CHAT_ID = 123456789  # <-- این عدد رو با chat_id خودت عوض کن

FEEDBACK_FILE = "feedback.csv"
FIELDS = ["date", "chat_id", "name", "rating", "comment"]
ALERT_RATING = 2      # امتیاز مساوی یا کمتر از این عدد فوری به مدیر اطلاع داده می‌شه
SKIP = "رد کردن"

sessions = {}  # chat_id -> امتیازی که منتظر نظر متنی‌اش هستیم


def api(method, data):
    try:
        return requests.post(f"{BASE}/{method}", json=data, timeout=15).json()
    except (requests.RequestException, ValueError) as error:
        print(f"خطا در {method}:", error)
        return {"ok": False}


def get_updates(offset):
    params = {"offset": offset, "timeout": 30}
    try:
        return requests.get(f"{BASE}/getUpdates", params=params, timeout=40).json()
    except (requests.RequestException, ValueError) as error:
        print("خطا در دریافت پیام‌ها:", error)
        return {"ok": False, "result": []}


def send(chat_id, text, markup=None):
    data = {"chat_id": chat_id, "text": text}
    if markup is not None:
        data["reply_markup"] = markup
    api("sendMessage", data)


def stars(n):
    return "⭐" * n


def rating_keyboard():
    """۵ دکمهٔ شیشه‌ای امتیاز؛ از ۵ ستاره تا ۱ ستاره."""
    return {"inline_keyboard": [[{"text": stars(n), "callback_data": f"rate:{n}"}] for n in range(5, 0, -1)]}


def save_feedback(chat_id, name, rating, comment):
    is_new = not os.path.exists(FEEDBACK_FILE)
    with open(FEEDBACK_FILE, "a", newline="", encoding="utf-8-sig") as file:
        writer = csv.DictWriter(file, fieldnames=FIELDS)
        if is_new:
            writer.writeheader()
        writer.writerow({
            "date": datetime.now().strftime("%Y-%m-%d %H:%M"),
            "chat_id": chat_id, "name": name, "rating": rating, "comment": comment,
        })


def stats_text():
    if not os.path.exists(FEEDBACK_FILE):
        return "هنوز هیچ نظری ثبت نشده."
    with open(FEEDBACK_FILE, "r", newline="", encoding="utf-8-sig") as file:
        ratings = [int(row["rating"]) for row in csv.DictReader(file)]
    if not ratings:
        return "هنوز هیچ نظری ثبت نشده."
    average = sum(ratings) / len(ratings)
    lines = [f"📊 تعداد نظرها: {len(ratings)}", f"میانگین امتیاز: {average:.1f} از ۵", ""]
    for n in range(5, 0, -1):
        lines.append(f"{stars(n)}  {ratings.count(n)}")
    return "\n".join(lines)


def finish(chat_id, name, comment):
    """ثبت نهایی امتیاز و نظر، و هشدار به مدیر در صورت امتیاز پایین."""
    rating = sessions.pop(chat_id)
    save_feedback(chat_id, name, rating, comment)
    send(chat_id, "ممنون که وقت گذاشتی 🙏 نظرت ثبت شد.", {"remove_keyboard": True})
    if rating <= ALERT_RATING:
        send(ADMIN_CHAT_ID, f"⚠️ مشتری ناراضی ({stars(rating)})\n👤 {name} (chat_id: {chat_id})\n💬 {comment or '—'}")


def handle_message(message):
    chat_id = message["chat"]["id"]
    text = message.get("text", "")
    user = message.get("from", {})
    name = (user.get("first_name", "") + " " + user.get("last_name", "")).strip() or "مشتری"

    if chat_id == ADMIN_CHAT_ID and text == "/stats":
        send(chat_id, stats_text())
        return

    if text == "/start":
        sessions.pop(chat_id, None)
        send(chat_id, f"سلام {name} 👋\nتجربه‌ات از خرید یا خدمت ما چطور بود؟ یک گزینه رو بزن:", rating_keyboard())
        return

    if chat_id in sessions:  # منتظر نظر متنی هستیم
        finish(chat_id, name, "" if text == SKIP else text)
        return

    send(chat_id, "برای ثبت امتیاز، دستور /start را بفرستید.")


def handle_callback(callback):
    data = callback.get("data", "")
    message = callback.get("message")
    if not message or not data.startswith("rate:"):
        return
    chat_id = message["chat"]["id"]
    rating = int(data.split(":", 1)[1])
    if rating not in range(1, 6):
        return

    sessions[chat_id] = rating
    # پیام دکمه‌ها رو ویرایش می‌کنیم تا دوباره زده نشن
    api("editMessageText", {"chat_id": chat_id, "message_id": message["message_id"],
                            "text": f"امتیاز تو: {stars(rating)}"})
    question = ("متأسفیم 😔 بگو چه چیزی باعث نارضایتی شد تا درستش کنیم:"
                if rating <= 3 else "عالیه 🌟 اگر نکته‌ای داری بنویس (اختیاری):")
    send(chat_id, question, {"keyboard": [[{"text": SKIP}]], "resize_keyboard": True})


def main():
    print("ربات نظرسنجی فیلتور روشن شد... (Ctrl+C برای خاموش‌کردن)")
    offset = 0
    while True:
        updates = get_updates(offset)
        if not updates.get("ok"):
            time.sleep(2)
            continue

        for update in updates.get("result", []):
            offset = update["update_id"] + 1
            if "callback_query" in update:
                handle_callback(update["callback_query"])
            elif "message" in update:
                print("پیام از chat_id:", update["message"]["chat"]["id"])
                handle_message(update["message"])

        time.sleep(0.5)


if __name__ == "__main__":
    main()