340 lines
15 KiB
Python
340 lines
15 KiB
Python
#!/usr/bin/env python3
|
||
"""VPS 到期提醒 Telegram Bot"""
|
||
|
||
import os, json, subprocess
|
||
from datetime import datetime
|
||
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
||
from telegram.ext import Application, CommandHandler, CallbackQueryHandler, MessageHandler, ConversationHandler, filters, ContextTypes
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv()
|
||
TOKEN = os.getenv("BOT_TOKEN")
|
||
ADMIN = int(os.getenv("ADMIN_ID", "0"))
|
||
DATA_FILE = os.path.join(os.path.dirname(__file__), "data.json")
|
||
|
||
def load_data():
|
||
try:
|
||
with open(DATA_FILE) as f: return json.load(f)
|
||
except: return {"vps": [], "remind_days": [1, 3, 7]}
|
||
|
||
def save_data(data):
|
||
with open(DATA_FILE, "w") as f: json.dump(data, f, ensure_ascii=False, indent=2)
|
||
|
||
def parse_date(text):
|
||
"""解析日期简写: 0315 -> 2026-03-15, 3-15 -> 2026-03-15"""
|
||
text = text.strip().replace("/", "-").replace(".", "-")
|
||
year = datetime.now().year
|
||
if len(text) == 4 and text.isdigit():
|
||
return f"{year}-{text[:2]}-{text[2:]}"
|
||
if len(text) in [3,4] and "-" in text:
|
||
parts = text.split("-")
|
||
return f"{year}-{parts[0].zfill(2)}-{parts[1].zfill(2)}"
|
||
if len(text) == 10:
|
||
return text
|
||
return text
|
||
|
||
def days_left(d):
|
||
return (datetime.strptime(d, "%Y-%m-%d") - datetime.now()).days
|
||
|
||
def ping_host(ip):
|
||
try:
|
||
r = subprocess.run(["ping", "-c", "1", "-W", "2", ip], capture_output=True, timeout=5)
|
||
return r.returncode == 0
|
||
except: return False
|
||
|
||
def calc_next_date(date_str, cycle):
|
||
"""根据周期计算下一个到期日"""
|
||
from dateutil.relativedelta import relativedelta
|
||
dt = datetime.strptime(date_str, "%Y-%m-%d")
|
||
c = cycle.lower().strip()
|
||
if c in ("月", "monthly", "month", "1m"):
|
||
return (dt + relativedelta(months=1)).strftime("%Y-%m-%d")
|
||
elif c in ("季", "quarterly", "quarter", "3m"):
|
||
return (dt + relativedelta(months=3)).strftime("%Y-%m-%d")
|
||
elif c in ("半年", "semi", "6m"):
|
||
return (dt + relativedelta(months=6)).strftime("%Y-%m-%d")
|
||
elif c in ("年", "yearly", "annual", "12m"):
|
||
return (dt + relativedelta(years=1)).strftime("%Y-%m-%d")
|
||
return None
|
||
|
||
def main_kb():
|
||
return InlineKeyboardMarkup([
|
||
[InlineKeyboardButton("📋 VPS列表", callback_data="list")],
|
||
[InlineKeyboardButton("➕ 添加", callback_data="add"), InlineKeyboardButton("✏️ 编辑", callback_data="edit"), InlineKeyboardButton("🗑 删除", callback_data="del")],
|
||
[InlineKeyboardButton("📡 Ping全部", callback_data="ping"), InlineKeyboardButton("🔔 测试", callback_data="test"), InlineKeyboardButton("⚙️ 设置", callback_data="settings")]
|
||
])
|
||
|
||
def back_kb():
|
||
return InlineKeyboardMarkup([[InlineKeyboardButton("🔙 返回", callback_data="back")]])
|
||
|
||
async def start(update: Update, ctx: ContextTypes.DEFAULT_TYPE):
|
||
await update.message.reply_text("🖥 *VPS 到期提醒*", parse_mode="Markdown", reply_markup=main_kb())
|
||
|
||
async def button(update: Update, ctx: ContextTypes.DEFAULT_TYPE):
|
||
q = update.callback_query
|
||
await q.answer()
|
||
data = q.data
|
||
|
||
if data == "back":
|
||
await q.edit_message_text("🖥 *VPS 到期提醒*", parse_mode="Markdown", reply_markup=main_kb())
|
||
|
||
elif data == "list":
|
||
d = load_data()
|
||
vps = d.get("vps", [])
|
||
if not vps:
|
||
await q.edit_message_text("📭 暂无VPS", reply_markup=back_kb())
|
||
return
|
||
groups = {}
|
||
for i, v in enumerate(vps):
|
||
p = v.get("provider", "未知")
|
||
if p not in groups: groups[p] = []
|
||
groups[p].append((i, v))
|
||
msg = "📋 *VPS 列表*\n"
|
||
renew_btns = []
|
||
for provider, items in groups.items():
|
||
msg += f"\n🏢 *{provider}*\n"
|
||
for i, v in sorted(items, key=lambda x: days_left(x[1]["date"])):
|
||
dl = days_left(v["date"])
|
||
if dl < 0:
|
||
icon, dl_text = "⚫", f"已过期{-dl}天"
|
||
renew_btns.append(InlineKeyboardButton(f"✅ 续费 {v['name']}", callback_data=f"renew_{i}"))
|
||
elif dl <= 3:
|
||
icon, dl_text = "🔴", f"{dl}天"
|
||
renew_btns.append(InlineKeyboardButton(f"✅ 续费 {v['name']}", callback_data=f"renew_{i}"))
|
||
elif dl <= 7:
|
||
icon, dl_text = "🟡", f"{dl}天"
|
||
else:
|
||
icon, dl_text = "🟢", f"{dl}天"
|
||
msg += f"{icon} {v['name']} - {v['date']} ({dl_text})\n"
|
||
kb = [[b] for b in renew_btns]
|
||
kb.append([InlineKeyboardButton("🔙 返回", callback_data="back")])
|
||
await q.edit_message_text(msg, parse_mode="Markdown", reply_markup=InlineKeyboardMarkup(kb))
|
||
|
||
elif data.startswith("renew_"):
|
||
idx = int(data.split("_")[1])
|
||
d = load_data()
|
||
v = d["vps"][idx]
|
||
cycle = v.get("cycle", "")
|
||
next_date = calc_next_date(v["date"], cycle)
|
||
kb = []
|
||
if next_date:
|
||
kb.append([InlineKeyboardButton(f"🔄 自动续期 → {next_date}", callback_data=f"renewauto_{idx}")])
|
||
kb.append([InlineKeyboardButton("📝 手动输入日期", callback_data=f"renewmanual_{idx}")])
|
||
kb.append([InlineKeyboardButton("🔙 返回", callback_data="list")])
|
||
msg = f"✅ 续费 *{v['name']}*\n\n当前到期: {v['date']}\n周期: {cycle}"
|
||
if next_date:
|
||
msg += f"\n自动续期: {next_date}"
|
||
await q.edit_message_text(msg, parse_mode="Markdown", reply_markup=InlineKeyboardMarkup(kb))
|
||
|
||
elif data.startswith("renewauto_"):
|
||
idx = int(data.split("_")[1])
|
||
d = load_data()
|
||
v = d["vps"][idx]
|
||
old_date = v["date"]
|
||
new_date = calc_next_date(v["date"], v.get("cycle", ""))
|
||
if new_date:
|
||
v["date"] = new_date
|
||
save_data(d)
|
||
await q.edit_message_text(f"✅ *{v['name']}* 已续费\n\n旧日期: {old_date}\n新日期: {new_date}", parse_mode="Markdown", reply_markup=back_kb())
|
||
else:
|
||
await q.edit_message_text("❌ 无法计算，请手动输入", reply_markup=back_kb())
|
||
|
||
elif data.startswith("renewmanual_"):
|
||
idx = int(data.split("_")[1])
|
||
ctx.user_data["renew_idx"] = idx
|
||
ctx.user_data["step"] = "renew_date"
|
||
d = load_data()
|
||
v = d["vps"][idx]
|
||
await q.edit_message_text(f"📝 续费 *{v['name']}*\n\n请输入新的到期日期 (如 0401 或 2026-04-01):", parse_mode="Markdown")
|
||
|
||
elif data == "settings":
|
||
d = load_data()
|
||
days = d.get("remind_days", [1, 3, 7])
|
||
kb = [[InlineKeyboardButton(f"{'✅' if i in days else '⬜'} {i}天", callback_data=f"toggle_{i}") for i in [1, 3, 7]],
|
||
[InlineKeyboardButton(f"{'✅' if i in days else '⬜'} {i}天", callback_data=f"toggle_{i}") for i in [14, 30]],
|
||
[InlineKeyboardButton("🔙 返回", callback_data="back")]]
|
||
await q.edit_message_text(f"⚙️ 提醒天数设置\n当前: {days}", reply_markup=InlineKeyboardMarkup(kb))
|
||
|
||
elif data.startswith("toggle_"):
|
||
day = int(data.split("_")[1])
|
||
d = load_data()
|
||
days = d.get("remind_days", [1, 3, 7])
|
||
if day in days: days.remove(day)
|
||
else: days.append(day)
|
||
days.sort()
|
||
d["remind_days"] = days
|
||
save_data(d)
|
||
kb = [[InlineKeyboardButton(f"{'✅' if i in days else '⬜'} {i}天", callback_data=f"toggle_{i}") for i in [1, 3, 7]],
|
||
[InlineKeyboardButton(f"{'✅' if i in days else '⬜'} {i}天", callback_data=f"toggle_{i}") for i in [14, 30]],
|
||
[InlineKeyboardButton("🔙 返回", callback_data="back")]]
|
||
await q.edit_message_text(f"⚙️ 提醒天数设置\n当前: {days}", reply_markup=InlineKeyboardMarkup(kb))
|
||
|
||
elif data == "edit":
|
||
d = load_data()
|
||
vps = d.get("vps", [])
|
||
if not vps:
|
||
await q.edit_message_text("📭 暂无VPS", reply_markup=back_kb())
|
||
return
|
||
kb = [[InlineKeyboardButton(f"✏️ {v['name']}", callback_data=f"editvps_{i}")] for i, v in enumerate(vps)]
|
||
kb.append([InlineKeyboardButton("🔙 返回", callback_data="back")])
|
||
await q.edit_message_text("选择要编辑的VPS:", reply_markup=InlineKeyboardMarkup(kb))
|
||
|
||
elif data.startswith("editvps_"):
|
||
idx = int(data.split("_")[1])
|
||
ctx.user_data["edit_idx"] = idx
|
||
d = load_data()
|
||
v = d["vps"][idx]
|
||
kb = [
|
||
[InlineKeyboardButton("📝 名称", callback_data="ed_name"), InlineKeyboardButton("🏢 商家", callback_data="ed_provider")],
|
||
[InlineKeyboardButton("🌐 IP", callback_data="ed_ip"), InlineKeyboardButton("📅 日期", callback_data="ed_date")],
|
||
[InlineKeyboardButton("💰 价格", callback_data="ed_price"), InlineKeyboardButton("🔄 周期", callback_data="ed_cycle")],
|
||
[InlineKeyboardButton("🔙 返回", callback_data="edit")]
|
||
]
|
||
msg = f"✏️ 编辑 *{v['name']}*\n\n"
|
||
msg += f"🏢 商家: {v.get('provider','')}\n"
|
||
msg += f"🌐 IP: {v.get('ip','无')}\n"
|
||
msg += f"📅 日期: {v.get('date','')}\n"
|
||
msg += f"💰 价格: {v.get('price','无')}\n"
|
||
msg += f"🔄 周期: {v.get('cycle','')}"
|
||
await q.edit_message_text(msg, parse_mode="Markdown", reply_markup=InlineKeyboardMarkup(kb))
|
||
|
||
elif data.startswith("ed_"):
|
||
field = data.split("_")[1]
|
||
ctx.user_data["edit_field"] = field
|
||
names = {"name":"名称","provider":"商家","ip":"IP","date":"日期","price":"价格","cycle":"周期"}
|
||
await q.edit_message_text(f"请输入新的{names.get(field,field)}:")
|
||
|
||
elif data == "del":
|
||
d = load_data()
|
||
vps = d.get("vps", [])
|
||
if not vps:
|
||
await q.edit_message_text("📭 暂无VPS", reply_markup=back_kb())
|
||
return
|
||
kb = [[InlineKeyboardButton(f"🗑 {v['name']}", callback_data=f"delvps_{i}")] for i, v in enumerate(vps)]
|
||
kb.append([InlineKeyboardButton("🔙 返回", callback_data="back")])
|
||
await q.edit_message_text("选择要删除的VPS:", reply_markup=InlineKeyboardMarkup(kb))
|
||
|
||
elif data.startswith("delvps_"):
|
||
idx = int(data.split("_")[1])
|
||
d = load_data()
|
||
if 0 <= idx < len(d["vps"]):
|
||
removed = d["vps"].pop(idx)
|
||
save_data(d)
|
||
await q.edit_message_text(f"✅ 已删除: {removed['name']}", reply_markup=back_kb())
|
||
|
||
elif data == "test":
|
||
d = load_data()
|
||
vps = d.get("vps", [])
|
||
if not vps:
|
||
await q.edit_message_text("📭 暂无VPS可测试", reply_markup=back_kb())
|
||
return
|
||
v = min(vps, key=lambda x: days_left(x["date"]))
|
||
dl = days_left(v["date"])
|
||
msg = f"🔔 *测试提醒*\n\n{v['name']} ({v['provider']})\n📅 {v['date']} (还剩 {dl} 天)"
|
||
await q.message.reply_text(msg, parse_mode="Markdown")
|
||
await q.edit_message_text("✅ 测试提醒已发送", reply_markup=back_kb())
|
||
|
||
elif data == "ping":
|
||
d = load_data()
|
||
vps = [v for v in d.get("vps", []) if v.get("ip")]
|
||
if not vps:
|
||
await q.edit_message_text("📭 没有可ping的VPS", reply_markup=back_kb())
|
||
return
|
||
await q.edit_message_text("📡 正在检测...")
|
||
msg = "📡 *Ping 结果*\n\n"
|
||
for v in vps:
|
||
ok = ping_host(v["ip"])
|
||
msg += f"{'🟢' if ok else '🔴'} {v['name']} - `{v['ip']}`\n"
|
||
await q.edit_message_text(msg, parse_mode="Markdown", reply_markup=back_kb())
|
||
|
||
elif data == "add":
|
||
ctx.user_data["step"] = "name"
|
||
await q.edit_message_text("📝 请输入 VPS 名称:")
|
||
|
||
async def handle_msg(update: Update, ctx: ContextTypes.DEFAULT_TYPE):
|
||
step = ctx.user_data.get("step")
|
||
text = update.message.text.strip()
|
||
|
||
# 编辑模式
|
||
if ctx.user_data.get("edit_field"):
|
||
idx = ctx.user_data.get("edit_idx", 0)
|
||
field = ctx.user_data["edit_field"]
|
||
d = load_data()
|
||
if field == "date":
|
||
text = parse_date(text)
|
||
d["vps"][idx][field] = text
|
||
save_data(d)
|
||
ctx.user_data.clear()
|
||
await update.message.reply_text("✅ 已更新", reply_markup=main_kb())
|
||
return
|
||
|
||
# 续费模式
|
||
if step == "renew_date":
|
||
idx = ctx.user_data.get("renew_idx", 0)
|
||
d = load_data()
|
||
v = d["vps"][idx]
|
||
old_date = v["date"]
|
||
v["date"] = parse_date(text)
|
||
save_data(d)
|
||
ctx.user_data.clear()
|
||
await update.message.reply_text(f"✅ *{v['name']}* 已续费\n旧日期: {old_date}\n新日期: {v['date']}", parse_mode="Markdown", reply_markup=main_kb())
|
||
return
|
||
return
|
||
|
||
if step == "name":
|
||
ctx.user_data["name"] = text
|
||
ctx.user_data["step"] = "provider"
|
||
await update.message.reply_text("🏢 请输入商家名称:")
|
||
elif step == "provider":
|
||
ctx.user_data["provider"] = text
|
||
ctx.user_data["step"] = "ip"
|
||
await update.message.reply_text("🌐 请输入 IP (没有输入 0):")
|
||
elif step == "ip":
|
||
ctx.user_data["ip"] = None if text == "0" else text
|
||
ctx.user_data["step"] = "cycle"
|
||
await update.message.reply_text("🔄 付款周期 (月/季/年):")
|
||
elif step == "cycle":
|
||
ctx.user_data["cycle"] = text
|
||
ctx.user_data["step"] = "date"
|
||
await update.message.reply_text("📅 到期日期 (YYYY-MM-DD):")
|
||
elif step == "date":
|
||
ctx.user_data["date"] = text
|
||
ctx.user_data["step"] = "price"
|
||
await update.message.reply_text("💰 价格 (如 /月，没有输入 0):")
|
||
elif step == "price":
|
||
ctx.user_data["price"] = None if text == "0" else text
|
||
d = load_data()
|
||
d["vps"].append({
|
||
"name": ctx.user_data["name"],
|
||
"provider": ctx.user_data["provider"],
|
||
"ip": ctx.user_data["ip"],
|
||
"cycle": ctx.user_data["cycle"],
|
||
"date": ctx.user_data["date"],
|
||
"price": ctx.user_data["price"]
|
||
})
|
||
save_data(d)
|
||
ctx.user_data.clear()
|
||
await update.message.reply_text(f"✅ 已添加: {ctx.user_data.get('name', 'VPS')}", reply_markup=main_kb())
|
||
|
||
async def check_expire(ctx: ContextTypes.DEFAULT_TYPE):
|
||
d = load_data()
|
||
remind_days = d.get("remind_days", [1, 3, 7])
|
||
for v in d.get("vps", []):
|
||
dl = days_left(v["date"])
|
||
if dl in remind_days:
|
||
msg = f"⏰ *到期提醒*\n\n{v['name']} ({v['provider']})\n📅 {v['date']} (还剩 {dl} 天)"
|
||
await ctx.bot.send_message(ADMIN, msg, parse_mode="Markdown")
|
||
|
||
def main():
|
||
app = Application.builder().token(TOKEN).build()
|
||
app.add_handler(CommandHandler("start", start))
|
||
app.add_handler(CommandHandler("help", start))
|
||
app.add_handler(CallbackQueryHandler(button))
|
||
app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_msg))
|
||
app.job_queue.run_daily(check_expire, time=datetime.strptime("09:00", "%H:%M").time())
|
||
print("Bot 启动")
|
||
app.run_polling(drop_pending_updates=True)
|
||
|
||
if __name__ == "__main__":
|
||
main()
|