network index

whallets_cli.py

@@ -71,44 +71,49 @@ def remove_wallet():
 
 def _get_inputs():
     """Get infor to add a new wallet"""
-    ntw = int(input(tabulate(tc._choose_network())))
+    ntw_i = _check_network()
     name = input(tc._prompt_name())
     addr = input(tc._prompt_address())
     inf = input(tc._prompt_info())
     twtr = input(tc._prompt_twitter())
     ens = input(tc._prompt_ens())
-    return ntw, name, addr, inf, twtr, ens
+    return ntw_i, name, addr, inf, twtr, ens
 
+def _check_network():
+    """Check if the existing network"""
+    ntw = int(input(tabulate(tc._choose_network())))
+    i = ntw - 1
+    if i != 0:
+        print(tc._missing_operation())
+        _new_choice()
+    else:
+        return i
 
-def check_wallet(ntw, name, addr, twtr, ens):
+def check_wallet(ntw_i,name, addr, twtr, ens):
     """
     Scans through wallets to check if a new wallet is not already in the
     database
     """
+    i = ntw_i
 
-    i = ntw - 1
-    if i != 0:
-        print(tc._missing_operation())
-        _new_choice()
-    else:
-        dict = get_wallets()[i]
-        for key, value in dict.items():
-            user = key
-            wallets = value['wallets']
-            info = value['info']
-            twitter = value['twitter']
-            ens_saved = value['ens']
-            if user == name:
-                x = 0
-            elif addr in wallets.values():
-                x = 1
-            elif twtr == twitter:
-                x = 2
-            elif ens == ens_saved:
-                x = 3
-            else:
-                x = 4
-            return x
+    dict = get_wallets()[i]
+    for key, value in dict.items():
+        user = key
+        wallets = value['wallets']
+        info = value['info']
+        twitter = value['twitter']
+        ens_saved = value['ens']
+        if user == name:
+            x = 0
+        elif addr in wallets.values():
+            x = 1
+        elif twtr == twitter:
+            x = 2
+        elif ens == ens_saved:
+            x = 3
+        else:
+            x = 4
+        return x