سورس ربات منوی کافه و سفارش از میز (بله)

# -*- coding: utf-8 -*-
# ------------------------------------------------------------
#  ربات منوی کافه و سفارش از میز (بله)
#  طراحی و کدنویسی: سروش مقدمی — فیلتور (filtori.com)
# ------------------------------------------------------------
#  مشتری با /start منو رو می‌بینه، با دکمه دسته و آیتم انتخاب می‌کنه،
#  سبد خریدش رو می‌بینه، شمارهٔ میز رو می‌گه و سفارش ثبت می‌شه.
#   - منو کاملاً از روی دیکشنری MENU ساخته می‌شه (فقط همون رو عوض کن)
#   - هر سفارش در orders.csv ذخیره می‌شه
#   - همان لحظه یک پیام به مدیر/بار (ADMIN_CHAT_ID) می‌ره
#
#  آموزش کامل و توضیح خط‌به‌خط:
#     https://filtori.com/bale-bot-source/cafe-menu-bot/
#  نسخهٔ حرفه‌ای با منوی QR، پنل و گزارش فروش:  https://filtori.com/cafe-menu/
#  تماس: 09122874862  |  تلگرام: https://t.me/filtori
# ------------------------------------------------------------

import csv
import time
from datetime import datetime

import requests

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

# chat_id مدیر یا باریستا؛ هر سفارش جدید فوری برای این شماره ارسال می‌شه.
# (ربات رو اجرا کن، /start بزن؛ chat_id تو در ترمینال چاپ می‌شه)
ADMIN_CHAT_ID = 123456789

ORDERS_FILE = "orders.csv"

# ------------------------------------------------------------
#  منوی کافه: دسته → {نام آیتم: قیمت به تومان}
#  فقط همین بخش رو با منوی خودت عوض کن؛ دکمه‌ها خودکار ساخته می‌شن.
# ------------------------------------------------------------
MENU = {
    "☕ قهوه": {
        "اسپرسو": 65000,
        "آمریکانو": 75000,
        "لاته": 95000,
        "کاپوچینو": 95000,
    },
    "🧊 نوشیدنی سرد": {
        "آیس لاته": 105000,
        "لیموناد": 85000,
        "موهیتو": 90000,
    },
    "🍰 کیک و دسر": {
        "چیزکیک": 120000,
        "براونی": 95000,
    },
}

BTN_CART = "🧺 سبد خرید"
BTN_CHECKOUT = "✅ ثبت سفارش"
BTN_CLEAR = "🗑 خالی کردن سبد"
BTN_BACK = "↩️ بازگشت به منو"

# حافظهٔ موقت: برای هر chat_id سبد و مرحلهٔ فعلی رو نگه می‌داریم.
# (در رم است؛ با ری‌استارت پاک می‌شود — نسخهٔ واقعی دیتابیس می‌خواهد)
sessions = {}


def fmt_price(amount):
    """۹۵۰۰۰ → «۹۵٬۰۰۰ تومان» (با ارقام فارسی)"""
    fa = str.maketrans("0123456789,", "۰۱۲۳۴۵۶۷۸۹٬")
    return f"{amount:,}".translate(fa) + " تومان"


def keyboard(rows):
    """ساخت کیبورد دکمه‌ای بله از روی لیست ردیف‌ها."""
    return {
        "keyboard": [[{"text": t} for t in row] for row in rows],
        "resize_keyboard": True,
    }


def main_keyboard():
    """کیبورد صفحهٔ اول: دسته‌ها + سبد و ثبت سفارش."""
    cats = list(MENU.keys())
    rows = [cats[i:i + 2] for i in range(0, len(cats), 2)]
    rows.append([BTN_CART, BTN_CHECKOUT])
    return keyboard(rows)


def category_keyboard(cat):
    """کیبورد آیتم‌های یک دسته + دکمهٔ بازگشت."""
    items = [f"{name} — {fmt_price(p)}" for name, p in MENU[cat].items()]
    rows = [items[i:i + 2] for i in range(0, len(items), 2)]
    rows.append([BTN_CART, BTN_BACK])
    return keyboard(rows)


def send_message(chat_id, text, reply_markup=None):
    url = f"{BASE}/sendMessage"
    data = {"chat_id": chat_id, "text": text}
    if reply_markup is not None:
        data["reply_markup"] = reply_markup
    try:
        requests.post(url, json=data, timeout=15)
    except requests.RequestException as error:
        print("خطا در ارسال پیام:", error)


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


def get_session(chat_id):
    if chat_id not in sessions:
        sessions[chat_id] = {"cart": {}, "step": "menu"}
    return sessions[chat_id]


def find_item(text):
    """از روی متن دکمهٔ زده‌شده، نام و قیمت آیتم رو پیدا می‌کنه."""
    name = text.split(" — ")[0].strip()
    for items in MENU.values():
        if name in items:
            return name, items[name]
    return None, None


def cart_text(cart):
    if not cart:
        return "سبد خریدت خالیه 🧺"
    lines, total = [], 0
    for name, info in cart.items():
        line_total = info["price"] * info["qty"]
        total += line_total
        lines.append(f"• {name} × {info['qty']} = {fmt_price(line_total)}")
    lines.append(f"\nجمع کل: {fmt_price(total)}")
    return "\n".join(lines)


def save_order(chat_id, table, cart):
    """ذخیرهٔ سفارش در CSV (utf-8-sig تا در اکسل فارسی درست باز شود)."""
    total = sum(i["price"] * i["qty"] for i in cart.values())
    items = " + ".join(f"{n}×{i['qty']}" for n, i in cart.items())
    is_new = False
    try:
        with open(ORDERS_FILE, "r", encoding="utf-8-sig"):
            pass
    except FileNotFoundError:
        is_new = True
    with open(ORDERS_FILE, "a", newline="", encoding="utf-8-sig") as f:
        w = csv.writer(f)
        if is_new:
            w.writerow(["تاریخ", "میز", "اقلام", "جمع (تومان)", "chat_id"])
        w.writerow([datetime.now().strftime("%Y-%m-%d %H:%M"), table, items, total, chat_id])
    return total, items


def handle(chat_id, text):
    s = get_session(chat_id)

    if text == "/start" or text == BTN_BACK:
        s["step"] = "menu"
        send_message(chat_id, "سلام ☕ به کافهٔ ما خوش اومدی!\nیک دسته رو انتخاب کن:", main_keyboard())
        return

    # مرحلهٔ گرفتن شمارهٔ میز
    if s["step"] == "table":
        table = text.strip()
        if not table.isdigit():
            send_message(chat_id, "لطفاً فقط شمارهٔ میز رو به عدد بنویس (مثلاً 4):")
            return
        total, items = save_order(chat_id, table, s["cart"])
        send_message(
            chat_id,
            f"سفارشت ثبت شد ✅\n🪑 میز {table}\n{cart_text(s['cart'])}\n\nتا چند دقیقهٔ دیگه آماده می‌شه 🙏",
            main_keyboard(),
        )
        send_message(ADMIN_CHAT_ID, f"🔔 سفارش جدید — میز {table}\n{items}\nجمع: {fmt_price(total)}")
        s["cart"] = {}
        s["step"] = "menu"
        return

    if text == BTN_CART:
        send_message(chat_id, cart_text(s["cart"]), keyboard([[BTN_CHECKOUT, BTN_CLEAR], [BTN_BACK]]))
        return

    if text == BTN_CLEAR:
        s["cart"] = {}
        send_message(chat_id, "سبد خالی شد.", main_keyboard())
        return

    if text == BTN_CHECKOUT:
        if not s["cart"]:
            send_message(chat_id, "اول یه چیزی به سبد اضافه کن 😊", main_keyboard())
            return
        s["step"] = "table"
        send_message(chat_id, "شمارهٔ میزت رو بنویس:", {"remove_keyboard": True})
        return

    # انتخاب دسته
    if text in MENU:
        send_message(chat_id, f"{text}\nیک آیتم انتخاب کن:", category_keyboard(text))
        return

    # انتخاب آیتم
    name, price = find_item(text)
    if name:
        item = s["cart"].setdefault(name, {"price": price, "qty": 0})
        item["qty"] += 1
        send_message(chat_id, f"«{name}» اضافه شد ✅ (تعداد: {item['qty']})", main_keyboard())
        return

    send_message(chat_id, "متوجه نشدم 🤔 از دکمه‌های پایین استفاده کن.", main_keyboard())


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
            message = update.get("message")
            if not message:
                continue
            chat_id = message["chat"]["id"]
            text = message.get("text", "")
            print("پیام از chat_id:", chat_id, "->", text)
            if text:
                handle(chat_id, text)
        time.sleep(0.5)


if __name__ == "__main__":
    main()