X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=m_lib%2Fmcrypt.py;h=13357371147af122e0f08d1880ff6295a95318b0;hb=413c257c07051e4be5588021514dcf42ec0c56ed;hp=acbead1f3f3426dcf8437597d39730977028302f;hpb=8d79317cdc8220ff3c2192e68fc1dfc749c3ea1c;p=m_lib.git

diff --git a/m_lib/mcrypt.py b/m_lib/mcrypt.py
index acbead1..1335737 100755
--- a/m_lib/mcrypt.py
+++ b/m_lib/mcrypt.py
@@ -5,6 +5,7 @@
 #
 
 
+from __future__ import print_function
 import random, crypt
 
 
@@ -25,6 +26,11 @@ def gen_salt():
 
 
 def test():
+   try:
+      raw_input
+   except NameError:  # Python 3
+      raw_input = input
+
    passwd = raw_input("Enter password: ")
    salt = gen_salt()
    encrypted = crypt.crypt(passwd, salt)
@@ -32,17 +38,16 @@ def test():
    pwd_file = open("test.pwd", 'w')
    pwd_file.write("%s:%s" % ("user", encrypted))
    pwd_file.close()
-   print "Password file written"
+   print("Password file written")
 
-   import string
    pwd_file = open("test.pwd", 'r')
-   username, encrypted = string.split(pwd_file.readline()[:-1], ':')
+   username, encrypted = pwd_file.readline()[:-1].split(':')
    pwd_file.close()
 
    if crypt.crypt(encrypted, encrypted):
-      print "Password verified Ok"
+      print("Password verified Ok")
    else:
-      print "BAD password"
+      print("BAD password")
 
 if __name__ == "__main__":
    test()