BadHappy/icalingua-python-bot
1
0
Fork 0
forked from SharwOrange/icalingua-python-bot
Code Pull Requests 1 Activity

FXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXK

Browse Source
This commit is contained in:
shenjack 2023-10-15 01:14:00 +08:00
parent a2f5b30e75
commit 88c46f754f
Signed by: shenjack
GPG Key ID: 7B1134A979775551
1 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
connect.py
View File

@ -2,6 +2,7 @@ import time
import random import random
import asyncio import asyncio
import traceback import traceback
import multiprocessing
from typing import Dict, List, Tuple, Any, Optional, Union, Literal from typing import Dict, List, Tuple, Any, Optional, Union, Literal
@ -110,7 +111,7 @@ def update_room(data: Dict[str, Any]):
print(f"{Fore.CYAN}update_room: {data}{Style.RESET_ALL}") print(f"{Fore.CYAN}update_room: {data}{Style.RESET_ALL}")
async def safe_eval(code: str) -> str: def safe_eval(code: str) -> str:
try: try:
# code = code.replace('help', '坏东西！\n') # code = code.replace('help', '坏东西！\n')
# code = code.replace('bytes', '坏东西！\n') # code = code.replace('bytes', '坏东西！\n')
@ -188,7 +189,20 @@ async def add_message(data: Dict[str, Any]):
elif data.get('message').get('content').startswith('=='): elif data.get('message').get('content').startswith('=='):
evals: str = data.get('message').get('content')[2:] evals: str = data.get('message').get('content')[2:]
result = await safe_eval(evals)
quene = multiprocessing.Queue()
def run(quene, evals):
go = safe_eval(evals)
quene.put(go)
process = multiprocessing.Process(target=run, args=(quene, evals))
process.start()
process.join(1)
if quene.empty():
result = '超时'
else:
result = quene.get()
reply = ReplyMessage(id=data['message']['_id']) reply = ReplyMessage(id=data['message']['_id'])
message = Message(content=result, message = Message(content=result,
reply_to=reply, reply_to=reply,
@ -257,12 +271,15 @@ def catch_all(event, data):
async def main(): async def main():
await sio.connect(HOST) async with asyncio.TaskGroup() as task_group:
await sio.connect(HOST)
# task1 = task_group.create_task(sio.wait())
await sio.wait()
await sio.emit('requireAuth', ('', {'version': '', 'protocolVersion': ''})) # await sio.emit('requireAuth', ('', {'version': '', 'protocolVersion': ''}))
await asyncio.sleep(2) # await asyncio.sleep(2)
await sio.wait() # await asyncio.gather(sio.wait(), sio.wait(), sio.wait())
if __name__ == '__main__': if __name__ == '__main__':