From 429243c7f2fe789afcd757fc203d1b0150c41d91 Mon Sep 17 00:00:00 2001
From: Aleksandr Vorobev <47173672+B1boid@users.noreply.github.com>
Date: Mon, 19 Jul 2021 22:08:00 +0300
Subject: [PATCH] + 'Edit' functionality

---
 bot.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/bot.py b/bot.py
index 1d3eaa9..87ccf2a 100644
--- a/bot.py
+++ b/bot.py
@@ -15,7 +15,7 @@
 
 keyboard1 = telebot.types.ReplyKeyboardMarkup(True, True)
 keyboard1.row('💰Balance', 'Show NFT image')
-keyboard1.row('Add adr', 'Show adr', 'Del adr')
+keyboard1.row('Add adr', 'Show adr', 'Edit adr', 'Del adr')
 users_addr = {}
 
 
@@ -85,6 +85,15 @@ def send_text(message):
     elif message.text == 'Add adr':
         bot.send_message(message.chat.id, text='Input your Eth address:')
         bot.register_next_step_handler(message, add_adr)
+        
+    elif message.text == 'Edit adr':
+        chat_id = message.chat.id
+        if chat_id in users_addr:
+            del users_addr[chat_id]
+            bot.send_message(message.chat.id, text='Input new Eth address:')
+            bot.register_next_step_handler(message, add_adr)   
+        else:
+            bot.send_message(message.chat.id, text='You do not have Eth address to edit.')    
 
     elif message.text == 'Del adr':
         chat_id = message.chat.id