From f6523c1ff887375e3a22106acce6a9487dbbf459 Mon Sep 17 00:00:00 2001
From: yasir <yasiramunandar@gmail.com>
Date: Tue, 14 Feb 2023 12:56:13 +0700
Subject: [PATCH] Add pickle when restart

---
 misskaty/__main__.py    | 7 +++++++
 misskaty/plugins/dev.py | 5 ++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/misskaty/__main__.py b/misskaty/__main__.py
index 669d0a28..af5628a4 100644
--- a/misskaty/__main__.py
+++ b/misskaty/__main__.py
@@ -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()
 
diff --git a/misskaty/plugins/dev.py b/misskaty/plugins/dev.py
index 4bf254e9..7bb5e397 100644
--- a/misskaty/plugins/dev.py
+++ b/misskaty/plugins/dev.py
@@ -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"])