From 7d73852f0b83d49850f547dbbc4ec6605b80be77 Mon Sep 17 00:00:00 2001
From: Danish <danish@ultroid.tech>
Date: Sun, 20 Jun 2021 23:04:12 +0530
Subject: [PATCH] Fixed Pmpermit

Co-authored-by: Amit Sharma <48654350+buddhhu@users.noreply.github.com>
---
 plugins/pmpermit.py | 103 ++++++++++++++++++++++++++++++--------------
 1 file changed, 70 insertions(+), 33 deletions(-)

diff --git a/plugins/pmpermit.py b/plugins/pmpermit.py
index f6f6afd..6d96d01 100644
--- a/plugins/pmpermit.py
+++ b/plugins/pmpermit.py
@@ -402,15 +402,25 @@ if sett == "True":
                     await message.delete()
                 async for message in apprvpm.client.iter_messages(user.id, search=UNS):
                     await message.delete()
-                await asst.edit_message(
-                    int(udB.get("LOG_CHANNEL")),
-                    _not_approved[uid],
-                    f"#APPROVED\n\n`User: `[{name0}](tg://user?id={uid})",
-                    buttons=[
-                        Button.inline("Disapprove PM", data=f"disapprove_{uid}"),
-                        Button.inline("Block", data=f"block_{uid}"),
-                    ],
-                )
+                try:
+                    await asst.edit_message(
+                        int(udB.get("LOG_CHANNEL")),
+                        _not_approved[uid],
+                        f"#APPROVED\n\n`User: `[{name0}](tg://user?id={uid})",
+                        buttons=[
+                            Button.inline("Disapprove PM", data=f"disapprove_{uid}"),
+                            Button.inline("Block", data=f"block_{uid}"),
+                        ],
+                    )
+                except KeyError:
+                    _not_approved[uid] = await asst.send_message(
+                        int(udB.get("LOG_CHANNEL")),
+                        f"#APPROVED\n\n`User: `[{name0}](tg://user?id={uid})",
+                        buttons=[
+                            Button.inline("Disapprove PM", data=f"disapprove_{uid}"),
+                            Button.inline("Block", data=f"block_{uid}"),
+                        ],
+                    )
             else:
                 await eod(apprvpm, "`User may already be approved.`")
         else:
@@ -466,15 +476,25 @@ if sett == "True":
                 await e.edit(f"[{name0}](tg://user?id={bbb.id}) `Disaproved to PM!`")
                 await asyncio.sleep(5)
                 await e.delete()
-                await asst.edit_message(
-                    int(udB.get("LOG_CHANNEL")),
-                    _not_approved[bbb.id],
-                    f"#DISAPPROVED\n\n[{name0}](tg://user?id={bbb.id}) `was disapproved to PM you.`",
-                    buttons=[
-                        Button.inline("Approve PM", data=f"approve_{bbb.id}"),
-                        Button.inline("Block", data=f"block_{bbb.id}"),
-                    ],
-                )
+                try:
+                    await asst.edit_message(
+                        int(udB.get("LOG_CHANNEL")),
+                        _not_approved[bbb.id],
+                        f"#DISAPPROVED\n\n[{name0}](tg://user?id={bbb.id}) `was disapproved to PM you.`",
+                        buttons=[
+                            Button.inline("Approve PM", data=f"approve_{bbb.id}"),
+                            Button.inline("Block", data=f"block_{bbb.id}"),
+                        ],
+                    )
+                except KeyError:
+                    _not_approved[bbb.id] = await asst.send_message(
+                        int(udB.get("LOG_CHANNEL")),
+                        f"#DISAPPROVED\n\n[{name0}](tg://user?id={bbb.id}) `was disapproved to PM you.`",
+                        buttons=[
+                            Button.inline("Approve PM", data=f"approve_{bbb.id}"),
+                            Button.inline("Block", data=f"block_{bbb.id}"),
+                        ],
+                    )
             else:
                 await e.edit(f"[{name0}](tg://user?id={bbb.id}) was never approved!")
                 await asyncio.sleep(5)
@@ -509,12 +529,19 @@ async def blockpm(block):
         disapprove_user(user)
     except AttributeError:
         pass
-    await asst.edit_message(
-        int(udB.get("LOG_CHANNEL")),
-        _not_approved[user],
-        f"#BLOCKED\n\n[{aname.first_name}](tg://user?id={user}) has been **blocked**.",
-        buttons=Button.inline("UnBlock", data=f"unblock_{user}"),
-    )
+    try:
+        await asst.edit_message(
+            int(udB.get("LOG_CHANNEL")),
+            _not_approved[user],
+            f"#BLOCKED\n\n[{aname.first_name}](tg://user?id={user}) has been **blocked**.",
+            buttons=Button.inline("UnBlock", data=f"unblock_{user}"),
+        )
+    except KeyError:
+        _not_approved[user] = await asst.send_message(
+            int(udB.get("LOG_CHANNEL")),
+            f"#BLOCKED\n\n[{aname.first_name}](tg://user?id={user}) has been **blocked**.",
+            buttons=Button.inline("UnBlock", data=f"unblock_{user}"),
+        )
 
 
 @ultroid_cmd(
@@ -535,15 +562,25 @@ async def unblockpm(unblock):
         await eor(unblock, f"`{aname.first_name} has been UnBlocked!`")
     except Exception as et:
         await eod(unblock, f"ERROR - {str(et)}")
-    await asst.edit_message(
-        int(udB.get("LOG_CHANNEL")),
-        _not_approved[user],
-        f"#UNBLOCKED\n\n[{aname.first_name}](tg://user?id={user}) has been **unblocked**.",
-        buttons=[
-            Button.inline("Block", data=f"block_{user}"),
-            Button.inline("Close", data="deletedissht"),
-        ],
-    )
+    try:
+        await asst.edit_message(
+            int(udB.get("LOG_CHANNEL")),
+            _not_approved[user],
+            f"#UNBLOCKED\n\n[{aname.first_name}](tg://user?id={user}) has been **unblocked**.",
+            buttons=[
+                Button.inline("Block", data=f"block_{user}"),
+                Button.inline("Close", data="deletedissht"),
+            ],
+        )
+    except KeyError:
+        _not_approved[user] = await asst.send_message(
+            int(udB.get("LOG_CHANNEL")),
+            f"#UNBLOCKED\n\n[{aname.first_name}](tg://user?id={user}) has been **unblocked**.",
+            buttons=[
+                Button.inline("Block", data=f"block_{user}"),
+                Button.inline("Close", data="deletedissht"),
+            ],
+        )
 
 
 @callback(