mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-25 10:42:11 +00:00
silly_girl 逻辑优化 (#74)
Co-authored-by: cdle <50124752+cdle@users.noreply.github.com> Co-authored-by: Sourcery AI <>
This commit is contained in:
sourcery-ai[bot]
GitHub
parent
8df55d835a
commit
309ef7063f
@ -1,3 +1,4 @@
|
||||
|
||||
from asyncio import sleep
|
||||
from pagermaid.listener import listener
|
||||
from pagermaid.enums import Message
|
||||
@ -7,56 +8,53 @@ from pagermaid import bot
|
||||
from pyrogram.enums.chat_type import ChatType
|
||||
from pagermaid.hook import Hook
|
||||
|
||||
import json
|
||||
|
||||
class SillyGirl:
|
||||
address = ""
|
||||
token = ""
|
||||
self_user_id = ""
|
||||
init = False
|
||||
working = False
|
||||
|
||||
def init_connect_info(self, address):
|
||||
sillyGirl.self_user_id = bot.me.id
|
||||
self.init = True
|
||||
self.self_user_id = bot.me.id
|
||||
if address:
|
||||
sqlite["silly_girl_address"] = address
|
||||
else:
|
||||
address = sqlite.get("silly_girl_address", "")
|
||||
try:
|
||||
address = sqlite.get("silly_girl_address")
|
||||
if '@' in address:
|
||||
s1 = address.split("//", 1)
|
||||
s2 = s1[1].split("@", 1)
|
||||
sillyGirl.token = s2[0]
|
||||
self.address = f"{s1[0]}//{s2[1]}"
|
||||
except:
|
||||
self.address = ""
|
||||
address = f"{s1[0]}//{s2[1]}"
|
||||
sillyGirl.address = address
|
||||
|
||||
async def polls(self):
|
||||
while True:
|
||||
if self.address:
|
||||
await self.poll([])
|
||||
else:
|
||||
return
|
||||
|
||||
async def poll(self, data):
|
||||
try:
|
||||
if not self.address:
|
||||
return
|
||||
init = ''
|
||||
if not self.init:
|
||||
if sillyGirl.init == False:
|
||||
init = "&init=true"
|
||||
self.init = True
|
||||
sillyGirl.init = True
|
||||
req_data = await client.post(
|
||||
f"{self.address}/pgm?token={self.token}{init}", json=data
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
print(e,"???====")
|
||||
await sleep(0.1)
|
||||
return
|
||||
if req_data.status_code != 200:
|
||||
await sleep(0.1)
|
||||
return
|
||||
try:
|
||||
replies = req_data.json()
|
||||
replies = json.loads(req_data.text)
|
||||
results = []
|
||||
for reply in replies:
|
||||
if reply["delete"]:
|
||||
@ -82,6 +80,13 @@ class SillyGirl:
|
||||
caption=reply["text"],
|
||||
reply_to_message_id=reply["reply_to"],
|
||||
)
|
||||
elif reply["videos"] and len(reply["videos"]) != 0:
|
||||
message = await bot.send_video(
|
||||
reply["chat_id"],
|
||||
reply["videos"][0],
|
||||
caption=reply["text"],
|
||||
reply_to_message_id=reply["reply_to"],
|
||||
)
|
||||
elif reply["text"] != '':
|
||||
message = await bot.send_message(reply["chat_id"], reply["text"], reply_to_message_id=reply["reply_to"])
|
||||
if message:
|
||||
@ -90,56 +95,71 @@ class SillyGirl:
|
||||
'uuid': reply["uuid"],
|
||||
})
|
||||
if len(results):
|
||||
await self.poll(results)
|
||||
await sillyGirl.poll(results)
|
||||
except Exception as e:
|
||||
print(e)
|
||||
print(e,"???")
|
||||
await sleep(0.1)
|
||||
|
||||
return
|
||||
|
||||
sillyGirl = SillyGirl()
|
||||
|
||||
|
||||
@Hook.on_startup()
|
||||
async def connect_sillyGirl():
|
||||
sillyGirl.init_connect_info("")
|
||||
bot.loop.create_task(sillyGirl.polls())
|
||||
bot.loop.create_task(sillyGirl.polls())
|
||||
|
||||
|
||||
@listener(command="sillyGirl", description="连接到傻妞服务器", parameters="<auth>")
|
||||
async def sillyGirl_connect(message: Message):
|
||||
|
||||
@listener(is_plugin=True,outgoing=True, ignore_edited=True, command="sillyGirl",description="连接到傻妞服务器", parameters="<auth>")
|
||||
async def Connect(message: Message):
|
||||
try:
|
||||
await edit_delete(message, "连接中...")
|
||||
await edit_delete(message,"连接中，建议重启...")
|
||||
sillyGirl.init_connect_info(message.arguments)
|
||||
except Exception as e:
|
||||
print(e,"+++")
|
||||
print(e)
|
||||
|
||||
|
||||
@listener(outgoing=True, ignore_edited=True, incoming=True)
|
||||
@listener(outgoing=True,ignore_edited=True, incoming=True)
|
||||
async def handle_receive(message: Message):
|
||||
try:
|
||||
reply_to = message.id
|
||||
reply = message.reply_to_message
|
||||
reply_to_sender_id = 0
|
||||
chat_id = message.chat.id
|
||||
sender_id = 0
|
||||
if message.from_user:
|
||||
sender_id = message.from_user.id
|
||||
if reply:
|
||||
if reply := message.reply_to_message:
|
||||
reply_to = reply.id
|
||||
if reply.from_user:
|
||||
reply_to_sender_id = reply.from_user.id
|
||||
if not sillyGirl.init:
|
||||
if message.reply_to_message_id:
|
||||
reply_to = message.reply_to_message_id
|
||||
if message.reply_to_message:
|
||||
reply_to = message.reply_to_message.id
|
||||
if message.reply_to_message.from_user:
|
||||
reply_to_sender_id = message.reply_to_message.from_user.id
|
||||
if sillyGirl.init != True:
|
||||
sillyGirl.init_connect_info("")
|
||||
if sillyGirl.self_user_id == sender_id:
|
||||
reply_to = 0
|
||||
await sillyGirl.poll(
|
||||
[{
|
||||
[
|
||||
{
|
||||
'id': message.id,
|
||||
'chat_id': message.chat.id,
|
||||
'chat_id': chat_id,
|
||||
'text': message.text,
|
||||
'sender_id': sender_id,
|
||||
'reply_to': reply_to,
|
||||
'reply_to_sender_id': reply_to_sender_id,
|
||||
'bot_id': sillyGirl.self_user_id,
|
||||
'is_group': message.chat.type != ChatType.PRIVATE,
|
||||
}])
|
||||
'is_group': message.chat.type
|
||||
in [ChatType.SUPERGROUP, ChatType.CHANNEL],
|
||||
}
|
||||
]
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
print(e,"---")
|
||||
return
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user