Updating profile info works

modules/menu/callbacks/get_code/callbacks.py

@@ -0,0 +1,35 @@
+from modules.decorators.callback import async_callback
+from modules.client import GeneratedClient
+from rich import print
+import os
+import re
+
+
+@async_callback
+async def get_code_callback(session_name: str):
+    print('Getting messages...')
+    client = GeneratedClient(session_name)
+
+    await client.start()
+
+    messages = [item async for item in client.get_chat_history(777000, limit=3)]
+
+    for message in messages:
+        try:
+            group = re.search(r': (\d{5})\D', message.text).groups()[0]
+            print(f"Confirmation code: \n{group}")
+            break
+        except Exception as e:
+            assert e
+            continue
+
+    print('Last messages from Telegram: ')
+    for message in messages:
+        print('-' * os.get_terminal_size().columns)
+        print(message.date)
+        print(message.text)
+    print('-' * os.get_terminal_size().columns)
+
+    await client.stop()
+
+    input()