dededdew

2025-11-13 10:31:13 +08:00
parent c6dcf4b0f3
commit 356bba8b20
5 changed files with 223 additions and 176 deletions
--- a/telegram/bot-v3.0.py
+++ b/telegram/bot-v3.0.py
@@ -1,9 +1,10 @@
 import datetime
+import random
+
 import requests
 from telethon import TelegramClient, events, Button
 import sqlite3
 from datetime import date
-import os
 import asyncio

 # ========== 配置区 ==========
@@ -13,7 +14,7 @@ BOT_TOKEN = "8451724418:AAGTGqCmc1JiUr88IABhMiQTHeVLcAcnT5Y"
 DB_PATH = "sign.db"
 SIGN_POINTS = 10

-ALLOWED_GROUPS = [-1003238845008]  # 只允许的群聊
+ALLOWED_GROUPS = [-1003238845008]

 PROXY = {
    'proxy_type': "socks5",
@@ -22,24 +23,44 @@ PROXY = {
    'username': "SyNuejCtrQ",
    'password': "MH8ioL7EXf"
 }
+
+INVITE_LINK = "https://www.websea.my/en/signup?key=77346588"
 # ============================

-# ---------- 命令字典 ----------
+# 命令字典
 COMMANDS = {
    '/sign': '签到，每天签到一次，获取积分',
    '签到': '签到，每天签到一次，获取积分',
    '/daily_rank': '查看今日发言排行',
    '/my_invites': '查看你邀请的人数',
+    # '/btc': '获取 BTC/USDT 最新价格',
    '/help': '显示所有可用命令及说明'
 }

-# ------------命令列表-------------------
+# 命令分组
 data = {
-    "签到": ['/sign', '签到', "haha"],
-    "发言": ["/daily_rank"],
-    "邀请": ['/my_invites'],
+    "签到": ['/sign', '签到',"/tanda"],
+    "发言": ["/daily_rank","发言","/kedudukan_harian"],
+    "邀请": ['/my_invites',"邀请","/daily_position"],
+    "币价": ['/btc', 'btc']
 }

+# 支持币种
+crypto_currencies = {
+    "BTC": {"type": "BTC-USDT", "url": "https://www.websea.com/zh-CN/trade/BTC-USDT"},
+    "ETH": {"type": "ETH-USDT", "url": "https://www.websea.com/zh-CN/trade/ETH-USDT"},
+    "SOL": {"type": "SOL-USDT", "url": "https://www.websea.com/zh-CN/trade/SOL-USDT"},
+    "BNB": {"type": "BNB-USDT", "url": "https://www.websea.com/zh-CN/trade/BNB-USDT"},
+    "WBS": {"type": "WBS-USDT", "url": "https://www.websea.com/zh-CN/trade/WBS-USDT"}
+}
+
+# 主菜单按钮
+main_buttons = [
+    [Button.inline("签到", b"/sign"), Button.inline("今日发言排行", b"/daily_rank")],
+    [Button.inline("我的邀请", b"/my_invites"), Button.inline("币价", b"/btc")],
+    [Button.inline("帮助", b"/help")]
+]
+

 def get_command_list_text():
    msg = "🤖 机器人命令列表：\n"
@@ -48,36 +69,61 @@ def get_command_list_text():
    return msg


-# ---------- 数据库操作 ----------
+# ---------- 数据库 ----------
 def init_db():
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
    cursor.execute('''
-        CREATE TABLE IF NOT EXISTS users
-        (
-            user_id INTEGER PRIMARY KEY,
-            username TEXT,
-            points INTEGER DEFAULT 0,
-            last_sign_date TEXT
-        )
-    ''')
+                   CREATE TABLE IF NOT EXISTS users
+                   (
+                       user_id
+                       INTEGER
+                       PRIMARY
+                       KEY,
+                       username
+                       TEXT,
+                       points
+                       INTEGER
+                       DEFAULT
+                       0,
+                       last_sign_date
+                       TEXT
+                   )
+                   ''')
    cursor.execute('''
-        CREATE TABLE IF NOT EXISTS daily_messages
-        (
-            user_id INTEGER,
-            date TEXT,
-            count INTEGER DEFAULT 0,
-            PRIMARY KEY(user_id, date)
-        )
-    ''')
+                   CREATE TABLE IF NOT EXISTS daily_messages
+                   (
+                       user_id
+                       INTEGER,
+                       date
+                       TEXT,
+                       count
+                       INTEGER
+                       DEFAULT
+                       0,
+                       PRIMARY
+                       KEY
+                   (
+                       user_id,
+                       date
+                   )
+                       )
+                   ''')
    cursor.execute('''
-        CREATE TABLE IF NOT EXISTS invites
-        (
-            inviter_id INTEGER,
-            invitee_id INTEGER,
-            PRIMARY KEY(inviter_id, invitee_id)
-        )
-    ''')
+                   CREATE TABLE IF NOT EXISTS invites
+                   (
+                       inviter_id
+                       INTEGER,
+                       invitee_id
+                       INTEGER,
+                       PRIMARY
+                       KEY
+                   (
+                       inviter_id,
+                       invitee_id
+                   )
+                       )
+                   ''')
    conn.commit()
    conn.close()

@@ -105,12 +151,10 @@ def add_points(user_id, points):
    today = date.today().isoformat()
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
-    cursor.execute("""
-        UPDATE users
-        SET points = points + ?,
-            last_sign_date = ?
-        WHERE user_id = ?
-    """, (points, today, user_id))
+    cursor.execute(
+        "UPDATE users SET points = points + ?, last_sign_date = ? WHERE user_id = ?",
+        (points, today, user_id)
+    )
    conn.commit()
    conn.close()

@@ -119,9 +163,9 @@ def get_points(user_id):
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
    cursor.execute("SELECT points FROM users WHERE user_id=?", (user_id,))
-    points = cursor.fetchone()
+    row = cursor.fetchone()
    conn.close()
-    return points[0] if points else 0
+    return row[0] if row else 0


 # ---------- 发言统计 ----------
@@ -129,11 +173,11 @@ def add_message(user_id):
    today = date.today().isoformat()
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
-    cursor.execute("""
-        INSERT INTO daily_messages (user_id, date, count)
-        VALUES (?, ?, 1) ON CONFLICT(user_id, date)
-        DO UPDATE SET count = count + 1
-    """, (user_id, today))
+    cursor.execute('''
+                   INSERT INTO daily_messages (user_id, date, count)
+                   VALUES (?, ?, 1) ON CONFLICT(user_id, date) DO
+                   UPDATE SET count = count + 1
+                   ''', (user_id, today))
    conn.commit()
    conn.close()

@@ -142,193 +186,199 @@ def get_daily_message_ranking():
    today = date.today().isoformat()
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
-    cursor.execute("""
-        SELECT u.username, d.count
-        FROM daily_messages d
-                 JOIN users u ON d.user_id = u.user_id
-        WHERE d.date = ?
-        ORDER BY d.count DESC
-    """, (today,))
-    results = cursor.fetchall()
+    cursor.execute('''
+                   SELECT u.username, d.count
+                   FROM daily_messages d
+                            JOIN users u ON d.user_id = u.user_id
+                   WHERE d.date = ?
+                   ORDER BY d.count DESC
+                   ''', (today,))
+    rows = cursor.fetchall()
    conn.close()
-    return results
+    return rows


-# ---------- 邀请统计 ----------
-def add_invite(inviter_id, invitee_id):
-    conn = sqlite3.connect(DB_PATH)
-    cursor = conn.cursor()
-    cursor.execute("INSERT OR IGNORE INTO invites (inviter_id, invitee_id) VALUES (?, ?)", (inviter_id, invitee_id))
-    conn.commit()
-    conn.close()
-
-
-def get_invite_count(inviter_id):
-    conn = sqlite3.connect(DB_PATH)
-    cursor = conn.cursor()
-    cursor.execute("SELECT COUNT(*) FROM invites WHERE inviter_id=?", (inviter_id,))
-    count = cursor.fetchone()[0]
-    conn.close()
-    return count
-
-
-# ---------- 获取币价 ----------
-def get_price():
-    headers = {
-        'accept': 'application/json, text/plain, */*',
-        'accept-language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6',
-        'cache-control': 'no-cache',
-        'dnt': '1',
-        'origin': 'https://www.websea.com',
-        'pragma': 'no-cache',
-        'priority': 'u=1, i',
-        'referer': 'https://www.websea.com/',
-        'sec-ch-ua': '"Chromium";v="142", "Microsoft Edge";v="142", "Not_A Brand";v="99"',
-        'sec-ch-ua-mobile': '?0',
-        'sec-ch-ua-platform': '"Windows"',
-        'sec-fetch-dest': 'empty',
-        'sec-fetch-mode': 'cors',
-        'sec-fetch-site': 'same-site',
-        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/142.0.0.0 Safari/537.36 Edg/142.0.0.0',
-    }
-
+# ---------- 币价 ----------
+def get_price(symbol: str):
+    headers = {'user-agent': 'Mozilla/5.0'}
    tz = datetime.timezone(datetime.timedelta(hours=8))
    today = datetime.datetime.now(tz).date()
    start_of_day = datetime.datetime.combine(today, datetime.time.min, tzinfo=tz)
    end_of_day = datetime.datetime.combine(today, datetime.time.max, tzinfo=tz)

    params = {
-        'symbol': 'SOL-USDT',
+        'symbol': symbol,
        'period': '30min',
        'start': int(start_of_day.timestamp()),
        'end': int(end_of_day.timestamp()),
    }

-    response = requests.get('https://eapi.websea.com/webApi/market/getKline', params=params, headers=headers)
-    data = response.json()["result"]["data"]
-
-    if not data:
+    try:
+        response = requests.get('https://eapi.websea.com/webApi/market/getKline', params=params, headers=headers,
+                                timeout=10)
+        data = response.json()["result"]["data"]
+        if not data:
+            return 0, 0, 0, 0
+        current_price = float(data[0]['close'])
+        today_high = max(float(i['high']) for i in data)
+        today_low = min(float(i['low']) for i in data)
+        price_24h_ago = float(data[-1]['open'])
+        change_24h = (current_price - price_24h_ago) / price_24h_ago * 100
+        return current_price, change_24h, today_high, today_low
+    except Exception as e:
+        print("获取币价失败:", e)
        return 0, 0, 0, 0

-    current_price = float(data[0]['close'])
-    today_high = max(float(item['high']) for item in data)
-    today_low = min(float(item['low']) for item in data)
-    price_24h_ago = float(data[-1]['open'])
-    change_24h = (current_price - price_24h_ago) / price_24h_ago * 100

-    return current_price, change_24h, today_high, today_low
+def get_price_msg(symbol: str):
+    current_price, change_24h, today_high, today_low = get_price(symbol)
+    name = symbol.split('-')[0]
+    url = crypto_currencies.get(name, {}).get("url", "#")
+    msg = (
+        f"📊 {name}/USDT 实时行情\n\n"
+        f"💰 当前价格：{current_price:.4f} USDT\n"
+        f"📈 24小时涨跌幅：{change_24h:.2f}%\n"
+        f"⬆️ 最高价：{today_high:.4f} USDT\n"
+        f"⬇️ 最低价：{today_low:.4f} USDT\n\n"
+        f"🔗 交易链接：[点击前往]({url})\n"
+        f"🎁 注册邀请：[立即加入]({INVITE_LINK})"
+    )
+    return msg


-# ---------- 发送币价消息 ----------
-async def send_price_periodically(bot, chat_id):
-    while True:
-        try:
-            current_price, change_24h, today_high, today_low = get_price()
-            msg = (
-                "SOL/USDT 现货数据\n\n"
-                f"💰价格：{current_price:.2f} USDT\n"
-                f"📈24小时涨跌幅：{change_24h:.2f}%\n"
-                f"⬆️最高价：{today_high:.2f} USDT\n"
-                f"⬇️最低价：{today_low:.2f} USDT"
-            )
-            await bot.send_message(chat_id, msg)
-        except Exception as e:
-            print(f"发送价格失败: {e}")
-        await asyncio.sleep(3600)  # 每小时发送一次
+def get_crypto_buttons():
+    """生成币种选择按钮"""
+    buttons = []
+    row = []
+    for name in crypto_currencies.keys():
+        row.append(Button.inline(name, f"price_{name}".encode()))
+        if len(row) == 3:
+            buttons.append(row)
+            row = []
+    if row:
+        buttons.append(row)
+    buttons.append([Button.inline("返回菜单", b"/help")])
+    return buttons


-# ---------- 统一命令处理 ----------
+# ---------- 命令处理 ----------
 async def handle_command(event, cmd):
    user = await event.get_sender()
    user_id = user.id
    username = user.username or user.first_name or "未知用户"

    add_or_update_user(user_id, username)
-
-    reply_text = ""
+    reply = ""

    if cmd in data["签到"]:
        if not can_sign(user_id):
-            reply_text = f"🌞 @{username}，你今天已经签到过啦！"
+            reply = f"🌞 @{username}，你今天已经签到过啦！"
        else:
            add_points(user_id, SIGN_POINTS)
            total = get_points(user_id)
-            reply_text = f"✅ @{username} 签到成功！你获得 {SIGN_POINTS} 积分。\n当前总积分：{total}"
+            reply = f"✅ @{username} 签到成功！获得 {SIGN_POINTS} 积分\n当前总积分：{total}"
    elif cmd in data["发言"]:
        ranking = get_daily_message_ranking()
        if not ranking:
-            reply_text = "📊 今日还没有人发言哦～"
+            reply = "📊 今日还没有人发言哦～"
        else:
-            reply_text = "📊 今日发言排行：\n"
-            for i, (uname, count) in enumerate(ranking, 1):
-                reply_text += f"{i}. {uname}: {count} 条\n"
+            reply = "📊 今日发言排行：\n" + "\n".join(f"{i + 1}. {u}: {c}条" for i, (u, c) in enumerate(ranking))
    elif cmd in data["邀请"]:
-        count = get_invite_count(user_id)
-        reply_text = f"👥 @{username}，你邀请了 {count} 人。"
-    elif cmd == '/help':
-        reply_text = get_command_list_text()
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
+        cursor.execute("SELECT COUNT(*) FROM invites WHERE inviter_id=?", (user_id,))
+        count = cursor.fetchone()[0]
+        conn.close()
+        reply = f"👥 @{username}，你邀请了 {count} 人。"
+    elif cmd in data["币价"]:
+        await event.respond("请选择要查看的币种：", buttons=get_crypto_buttons())
+        return
    else:
-        reply_text = f"⚠️ 未知命令：{cmd}"
+        await event.reply(get_command_list_text(), buttons=main_buttons)
+        return

-    reply_text += "\n\n" + get_command_list_text()
-    await event.reply(reply_text)
+    await event.reply(reply)


-# ---------- Bot 主逻辑 ----------
+# ---------- 价格按钮处理 ----------
+async def handle_price_command(event, symbol_name: str):
+    symbol = crypto_currencies[symbol_name]["type"]
+    msg = get_price_msg(symbol)
+    await event.edit(msg, buttons=get_crypto_buttons(), link_preview=False)
+
+
+# ---------- 主逻辑 ----------
 async def main():
    init_db()
    bot = TelegramClient('bot_session', API_ID, API_HASH, proxy=PROXY)
    await bot.start(bot_token=BOT_TOKEN)

-    # 启动定时任务，每小时发送币价
+    # 定时发送 SOL 币价
    for group_id in ALLOWED_GROUPS:
        asyncio.create_task(send_price_periodically(bot, group_id))

-    # 新用户加入欢迎
+    # 欢迎新成员
    @bot.on(events.ChatAction)
-    async def welcome_new_user(event):
+    async def welcome(event):
        if event.user_added or event.user_joined:
            for user in event.users:
-                username = user.username or user.first_name or "新成员"
-                await event.reply(f"欢迎 @{username} 加入群聊！\n{get_command_list_text()}")
+                name = user.username or user.first_name or "新成员"
+                await event.reply(f"欢迎 @{name} 加入群聊！\n{get_command_list_text()}")

-    # 普通消息处理
+    # 普通消息
    @bot.on(events.NewMessage)
-    async def message_handler(event):
+    async def on_message(event):
        if event.chat_id not in ALLOWED_GROUPS:
            return

-        text = event.raw_text.strip()
-        user = await event.get_sender()
-        user_id = user.id
+        text = event.raw_text.strip().lower()

-        for i in data:
-            if text in data[i]:
-                cmd = text.split('@')[0]
-                await handle_command(event, cmd)
-                break
-        else:
-            add_message(user_id)
+        # ✅ 去掉命令中的 @botname（如 /sign@dockersebot → /sign）
+        if "@" in text:
+            text = text.split("@")[0]

-    # 按钮命令发送
-    @bot.on(events.NewMessage(pattern='/commands'))
-    async def send_buttons(event):
-        buttons = [
-            [Button.inline("签到", b"/sign"), Button.inline("今日发言排行", b"/daily_rank")],
-            [Button.inline("我的邀请", b"/my_invites"), Button.inline("帮助", b"/help")]
-        ]
-        await event.reply("请选择命令：", buttons=buttons)
+        for group in data:
+            if text in [cmd.lower() for cmd in data[group]]:
+                await handle_command(event, text)
+                return

-    # 按钮点击处理
+        add_message((await event.get_sender()).id)
+
+    # # help
+    # @bot.on(events.NewMessage(pattern='/help'))
+    # async def send_help(event):
+
+    # 按钮点击
    @bot.on(events.CallbackQuery)
-    async def callback_handler(event):
+    async def callback(event):
+        await event.answer("正在处理...", alert=False)
        cmd = event.data.decode()
-        await handle_command(event, cmd)
-        await event.answer()
+
+        # ✅ 同样去掉命令中的 @botname
+        if "@" in cmd:
+            cmd = cmd.split("@")[0]
+
+        if cmd.startswith("price_"):
+            symbol = cmd.split("_")[1]
+            await handle_price_command(event, symbol)
+        else:
+            await handle_command(event, cmd)

    print("🤖 机器人已启动，等待群聊消息...")
    await bot.run_until_disconnected()


+# ---------- 定时币价 ----------
+async def send_price_periodically(bot, chat_id):
+    while True:
+        try:
+            random_key = random.choice(list(crypto_currencies.keys()))
+            msg = get_price_msg(crypto_currencies[random_key]["type"])
+            await bot.send_message(chat_id, msg, buttons=get_crypto_buttons(), link_preview=False)
+        except Exception as e:
+            print(f"发送价格失败: {e}")
+        await asyncio.sleep(3600)  # 每小时更新
+
+
 if __name__ == "__main__":
    asyncio.run(main())
--- a/telegram/bot_session.session
+++ b/telegram/bot_session.session
--- a/telegram/bot_session.session-journal
+++ b/telegram/bot_session.session-journal
--- a/telegram/sign.db
+++ b/telegram/sign.db
--- a/test.py
+++ b/test.py
@@ -1,16 +1,13 @@
-import datetime
+import random

-# 获取当前日期（UTC）
-today = datetime.datetime.now(datetime.UTC).date()
+# 定义一个示例字典
+my_dict = {'a': 1, 'b': 2, 'c': 3, 'd': 4}

-# 获取当天开始时间（00:00:00 UTC）
-start_of_day = datetime.datetime.combine(today, datetime.time.min, tzinfo=datetime.timezone.utc)
-# 获取当天结束时间（23:59:59.999999 UTC）
-end_of_day = datetime.datetime.combine(today, datetime.time.max, tzinfo=datetime.timezone.utc)
+# 随机选择一个键
+random_key = random.choice(list(my_dict.keys()))

-# 将时间转换为时间戳
-start_timestamp = start_of_day.timestamp()
-end_timestamp = end_of_day.timestamp()
+# 根据随机选择的键获取对应的值
+random_value = my_dict[random_key]

-print(f"当天开始的 UTC 时间戳: {start_timestamp}")
-print(f"当天结束的 UTC 时间戳: {end_timestamp}")
+# 输出随机选择的键值对
+print(f"随机选择的键: {random_key}, 对应的值: {random_value}")