X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=m_lib%2Fmcrypt.py;h=13357371147af122e0f08d1880ff6295a95318b0;hb=ee5ae8bff7bd4b242788e3ba7807a49a59f79ebb;hp=76922a0efe71b521642e84d18055cce890cd7055;hpb=fb3af94919a9ee18ba1c35f927f19837f057fa1f;p=m_lib.git

diff --git a/m_lib/mcrypt.py b/m_lib/mcrypt.py
index 76922a0..1335737 100755
--- a/m_lib/mcrypt.py
+++ b/m_lib/mcrypt.py
@@ -26,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)
@@ -35,9 +40,8 @@ def test():
    pwd_file.close()
    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):