yasirarism/MissKatyPyro
1
0
Fork 0
mirror of https://github.com/yasirarism/MissKatyPyro.git synced 2025-12-29 17:44:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add pickle when restart

Browse source
This commit is contained in:
yasir 2023-02-14 12:56:13 +07:00
parent 8df34e6dce
commit f6523c1ff8
2 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
misskaty/__main__.py
View file

@ -8,6 +8,8 @@
import asyncio
import importlib
import re
import os
import pickle
import traceback
from logging import getLogger
@ -68,6 +70,11 @@ async def start_bot():
)
except Exception as e:
LOGGER.error(str(e))
if os.path.exists("restart.pickle"):
with open('restart.pickle', 'rb') as status:
chat_id, message_id = pickle.load(status)
os.remove("restart.pickle")
await app.edit_message_text(chat_id=chat_id, message_id=message_id, text="<b>Bot restarted successfully!</b>")
asyncio.create_task(auto_clean())
await idle()

5
misskaty/plugins/dev.py
View file

@ -2,6 +2,7 @@ import asyncio
import io
import os
import sys
import pickle
import traceback
from inspect import getfullargspec
@ -199,7 +200,9 @@ async def update_restart(_, message):
await message.reply_text(f"<code>{out}</code>")
except Exception as e:
return await message.reply_text(str(e))
await message.reply_text("<b>Updated with default branch, restarting now.</b>")
msg = await message.reply_text("<b>Updated with default branch, restarting now.</b>")
with open('restart.pickle', 'wb') as status:
pickle.dump([message.chat.id, msg.id], status)
os.execvp(sys.executable, [sys.executable, "-m", "misskaty"])