198 lines
7.6 KiB
Python
198 lines
7.6 KiB
Python
|
import asyncio
|
|||
|
|
import random
|
|||
|
|
import re
|
|||
|
|
import sqlite3
|
|||
|
|
import traceback
|
|||
|
|
|
|||
|
|
from pyrogram import Client
|
|||
|
|
from pyrogram.types import Message
|
|||
|
|
|
|||
|
|
from defs.db import deal_ck, selectDB, OpenPush, CheckDB, connectDB
|
|||
|
|
from defs.event import generate_event
|
|||
|
|
from defs.mys2 import award, sign, daily
|
|||
|
|
|
|||
|
|
from ci import scheduler, app
|
|||
|
|
|
|||
|
|
SUPERUSERS = []
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def mys2_msg(client: Client, message: Message):
|
|||
|
|
text = message.text.replace("米游社", "")
|
|||
|
|
userid = message.from_user.id
|
|||
|
|
if '添加 ' in text:
|
|||
|
|
try:
|
|||
|
|
mes = text.replace('添加', '').strip()
|
|||
|
|
if not mes:
|
|||
|
|
return await message.reply_text("获取 Cookie 请参考：[link](https://github.com/Womsxd/AutoMihoyoBBS/"
|
|||
|
|
"#%E8%8E%B7%E5%8F%96%E7%B1%B3%E6%B8%B8%E7%A4%BEcookie)", quote=True)
|
|||
|
|
await deal_ck(mes, userid)
|
|||
|
|
await message.reply(f'添加Cookies成功！\n'
|
|||
|
|
f'Cookies属于个人重要信息，如果你是在不知情的情况下添加，'
|
|||
|
|
f'请马上修改米游社账户密码，保护个人隐私！\n'
|
|||
|
|
f'<code>=============</code>\n'
|
|||
|
|
f'如果需要【开启自动签到】和【开启推送】还需要使用命令 '
|
|||
|
|
f'<code>米游社绑定uid</code>绑定你的uid。\n'
|
|||
|
|
f'例如：<code>米游社绑定uid123456789</code>。')
|
|||
|
|
except Exception as e:
|
|||
|
|
traceback.print_exc()
|
|||
|
|
await message.reply(f'校验失败！请输入正确的Cookies！', quote=True)
|
|||
|
|
elif '推送' in text:
|
|||
|
|
try:
|
|||
|
|
uid = await selectDB(userid, mode="uid")
|
|||
|
|
if '开启' in text:
|
|||
|
|
im = await OpenPush(int(uid[0]), userid, "on", "StatusA")
|
|||
|
|
await message.reply(im, quote=True)
|
|||
|
|
else:
|
|||
|
|
im = await OpenPush(int(uid[0]), userid, "off", "StatusA")
|
|||
|
|
await message.reply(im, quote=True)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(e.with_traceback)
|
|||
|
|
await message.reply("未找到uid绑定记录。", quote=True)
|
|||
|
|
elif '自动签到' in text:
|
|||
|
|
try:
|
|||
|
|
uid = await selectDB(userid, mode="uid")
|
|||
|
|
if '开启' in text:
|
|||
|
|
im = await OpenPush(int(uid[0]), userid, "on", "StatusB")
|
|||
|
|
await message.reply(im, quote=True)
|
|||
|
|
else:
|
|||
|
|
im = await OpenPush(int(uid[0]), userid, "off", "StatusA")
|
|||
|
|
await message.reply(im, quote=True)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(e.with_traceback)
|
|||
|
|
await message.reply("未找到uid绑定记录。", quote=True)
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def mys2_qun_msg(client: Client, message: Message):
|
|||
|
|
text = message.text.replace("米游社", "")
|
|||
|
|
|
|||
|
|
qid = message.from_user.id
|
|||
|
|
at = message.reply_to_message
|
|||
|
|
if "自动签到" in text:
|
|||
|
|
try:
|
|||
|
|
if at and qid in SUPERUSERS:
|
|||
|
|
qid = at.from_user.id
|
|||
|
|
elif at and qid not in SUPERUSERS:
|
|||
|
|
return await message.reply("你没有权限。")
|
|||
|
|
gid = message.chat.id
|
|||
|
|
uid = await selectDB(qid, mode="uid")
|
|||
|
|
if "开启" in text:
|
|||
|
|
im = await OpenPush(int(uid[0]), message.from_user.id, str(gid), "StatusB")
|
|||
|
|
await message.reply(im, quote=True)
|
|||
|
|
elif "关闭" in text:
|
|||
|
|
im = await OpenPush(int(uid[0]), message.from_user.id, "off", "StatusB")
|
|||
|
|
await message.reply(im)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(e.with_traceback)
|
|||
|
|
await message.reply("未绑定uid信息！")
|
|||
|
|
elif "推送" in text:
|
|||
|
|
try:
|
|||
|
|
if at and qid in SUPERUSERS:
|
|||
|
|
qid = at.from_user.id
|
|||
|
|
elif at and qid not in SUPERUSERS:
|
|||
|
|
return await message.reply("你没有权限。")
|
|||
|
|
gid = message.chat.id
|
|||
|
|
uid = await selectDB(qid, mode="uid")
|
|||
|
|
if "开启" in text:
|
|||
|
|
im = await OpenPush(int(uid[0]), message.from_user.id, str(gid), "StatusA")
|
|||
|
|
await message.reply(im, quote=True)
|
|||
|
|
elif "关闭" in text:
|
|||
|
|
im = await OpenPush(int(uid[0]), message.from_user.id, "off", "StatusA")
|
|||
|
|
await message.reply(im)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(e.with_traceback)
|
|||
|
|
await message.reply("未绑定uid信息！")
|
|||
|
|
elif "每月统计" in text:
|
|||
|
|
try:
|
|||
|
|
uid = await selectDB(message.from_user.id, mode="uid")
|
|||
|
|
uid = uid[0]
|
|||
|
|
im = await award(uid)
|
|||
|
|
await message.reply(im)
|
|||
|
|
except Exception as e:
|
|||
|
|
traceback.print_exc()
|
|||
|
|
await message.reply('未找到绑定信息')
|
|||
|
|
elif "签到" in text:
|
|||
|
|
try:
|
|||
|
|
uid = await selectDB(message.from_user.id, mode="uid")
|
|||
|
|
uid = uid[0]
|
|||
|
|
im = await sign(uid)
|
|||
|
|
await message.reply(im)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(e.with_traceback)
|
|||
|
|
await message.reply('未找到绑定信息')
|
|||
|
|
elif "效验全部" in text:
|
|||
|
|
im = await CheckDB()
|
|||
|
|
await message.reply(im)
|
|||
|
|
elif "当前状态" in text:
|
|||
|
|
try:
|
|||
|
|
uid = await selectDB(message.from_user.id, mode="uid")
|
|||
|
|
uid = uid[0]
|
|||
|
|
mes = await daily("ask", uid)
|
|||
|
|
im = mes[0]['message']
|
|||
|
|
except Exception as e:
|
|||
|
|
print(e.with_traceback)
|
|||
|
|
im = "没有找到绑定信息。"
|
|||
|
|
await message.reply(im)
|
|||
|
|
elif "绑定uid" in text:
|
|||
|
|
uid = text.replace("绑定uid", "") # str
|
|||
|
|
await connectDB(message.from_user.id, uid)
|
|||
|
|
await message.reply('绑定uid成功！')
|
|||
|
|
elif "绑定mys" in text:
|
|||
|
|
mys = text.replace("绑定mys", "") # str
|
|||
|
|
await connectDB(message.from_user.id, None, mys)
|
|||
|
|
await message.reply('绑定米游社id成功！')
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 每隔一小时检测树脂是否超过设定值
|
|||
|
|
@scheduler.scheduled_job('interval', hours=1)
|
|||
|
|
async def push():
|
|||
|
|
daily_data = await daily()
|
|||
|
|
if daily_data is not None:
|
|||
|
|
for i in daily_data:
|
|||
|
|
if i['gid'] == "on":
|
|||
|
|
await app.send_message(int(i['qid']), i['message'])
|
|||
|
|
else:
|
|||
|
|
await app.send_message(int(i['gid']), f"[NOTICE](tg://user?id={i['qid']})" + "\n" + i['message'])
|
|||
|
|
else:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 每日零点半进行米游社签到
|
|||
|
|
@scheduler.scheduled_job('cron', hour='0', minute="30")
|
|||
|
|
async def daily_sign():
|
|||
|
|
conn = sqlite3.connect('ID_DATA.db')
|
|||
|
|
c = conn.cursor()
|
|||
|
|
cursor = c.execute(
|
|||
|
|
"SELECT * FROM NewCookiesTable WHERE StatusB != ?", ("off",))
|
|||
|
|
c_data = cursor.fetchall()
|
|||
|
|
temp_list = []
|
|||
|
|
|
|||
|
|
for row in c_data:
|
|||
|
|
if row[4] == "on":
|
|||
|
|
try:
|
|||
|
|
im = await sign(str(row[0]))
|
|||
|
|
await app.send_message(int(row[2]), im)
|
|||
|
|
except Exception as e:
|
|||
|
|
traceback.print_exc()
|
|||
|
|
else:
|
|||
|
|
im = await sign(str(row[0]))
|
|||
|
|
message = f"[CQ:at,qq={row[2]}]\n{im}"
|
|||
|
|
for i in temp_list:
|
|||
|
|
if row[4] == i["push_group"]:
|
|||
|
|
i["push_message"] = i["push_message"] + "\n" + message
|
|||
|
|
break
|
|||
|
|
else:
|
|||
|
|
temp_list.append({"push_group": row[4], "push_message": message})
|
|||
|
|
await asyncio.sleep(6 + random.randint(0, 2))
|
|||
|
|
|
|||
|
|
for i in temp_list:
|
|||
|
|
try:
|
|||
|
|
await app.send_message(int(i["push_group"]), i["push_message"])
|
|||
|
|
except Exception as e:
|
|||
|
|
traceback.print_exc()
|
|||
|
|
await asyncio.sleep(3 + random.randint(0, 2))
|
|||
|
|
|
|||
|
|
|
|||
|
|
@scheduler.scheduled_job('cron', hour='2')
|
|||
|
|
async def delete():
|
|||
|
|
await generate_event()
|