X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=m_lib%2Ftty_menu.py;h=d6fd92069f1f20ebd8a2400721337a57fb5d27bd;hb=d1c355c4dbae68d29ed7e5b1e3dd5a77bdcbd2cb;hp=897ecc9a8b2764ef02c1db51f4f03bbaef39152d;hpb=fb3af94919a9ee18ba1c35f927f19837f057fa1f;p=m_lib.git

diff --git a/m_lib/tty_menu.py b/m_lib/tty_menu.py
index 897ecc9..d6fd920 100755
--- a/m_lib/tty_menu.py
+++ b/m_lib/tty_menu.py
@@ -3,7 +3,12 @@
 
 
 from __future__ import print_function
-import string
+
+
+try:
+   raw_input
+except NameError:  # Python 3
+   raw_input = input
 
 
 def hmenu(prompt, astring):
@@ -32,8 +37,8 @@ def vmenu(item_list, prompt, format = "%d. %s"):
    result = raw_input(prompt)
 
    try:
-      result = string.atoi(result)
-   except string.atoi_error:
+      result = int(result)
+   except ValueError:
       result = -1
 
    return result