From 3e858c51a83a1e6a64ec5ace916ec3b4b52821f8 Mon Sep 17 00:00:00 2001
From: lifegpc <g1710431395@gmail.com>
Date: Sat, 11 Dec 2021 12:32:48 +0800
Subject: [PATCH] allow select chat if mulitply chat id returned

---
 tg_delete_all_my_msgs.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/tg_delete_all_my_msgs.py b/tg_delete_all_my_msgs.py
index 107d26e..650fe45 100644
--- a/tg_delete_all_my_msgs.py
+++ b/tg_delete_all_my_msgs.py
@@ -37,7 +37,25 @@ async def get_chat_id_from_name(lib: TdLib, name: str) -> int:
     if le == 1:
         return re['chat_ids'][0]
     else:
-        raise NotImplementedError('Multiply chat is returned.')
+        k = 1
+        for i in re['chat_ids']:
+            c = await lib.getChat(i)
+            if c is None:
+                raise ValueError('Chat not found.')
+            print(f"{k:02}. Chat ID: \t{i}")
+            print(f"    Chat Title: {c['title']}")
+            print(f"    Chat Type: \t{c['type']}")
+            k += 1
+        while True:
+            inp = input('Please select one chat: ')
+            try:
+                i = int(inp)
+                if i >= 1 and i <= le:
+                    return re['chat_ids'][i - 1]
+                if i in re['chat_ids']:
+                    return i
+            except ValueError:
+                pass
 
 
 async def main(lib: TdLib, arg):