
comprookie2000
By: a guest on
Apr 3rd, 2009 | syntax:
Python | size: 1.88 KB | hits: 336 | expires: Never
#!/usr/bin/python
import cPickle
import os
from sys import exit
def get_pw():
print '\nYour current passwords are: \n'
for k, v in pw.iteritems():
print k, v[0],v[1]
load_pw(pw)
def add_pw():
try:
key = raw_input('Enter Password Title: ')
username = raw_input('Enter Username: ')
password = raw_input('Enter Password: ')
pw[key] = username, password
load_pw(pw)
print '\nYour current Passwords are: \n'
for k, v in pw.iteritems():
print k, v[0], v[1]
response = raw_input('\nDo you want to enter another password? (y/n) ')
response = response.lower()
if response == 'y':
add_pw()
else:
print 'Goodbye'
except KeyError, e:
print '\nError! Enter Password Data!\n'
def load_pw(pw):
fname = open('pw_data.dat', 'w')
object = cPickle.Pickler(fname)
object.dump(pw)
fname.close()
def del_pw():
try:
print '\nYour current Passwords are: \n'
for k, v in pw.iteritems():
print k
answer = raw_input('\nWhich Password do you want to remove?\n')
del pw[answer]
print '\nDeleted:', answer
print '\nYour current Passwords are: \n'
for k, v in pw.iteritems():
print k, v[0],v[1]
load_pw(pw)
except KeyError, e:
print '\nError! Enter Password to be removed.\n'
def menu():
print '''
Passwords
Press I: Password Info
Press A: Add Password
Press D: Remove Password
Press X: Exit
'''
answer = raw_input('\nEnter >> ')
answer = answer.lower()
if answer == 'a':
add_pw()
menu()
elif answer == 'i':
get_pw()
menu()
elif answer == 'd':
del_pw()
menu()
elif answer == 'x':
print 'Exit'
exit()
else:
print 'Goodbye'
if __name__ == '__main__':
if os.path.exists('pw_data.dat'):
try:
fname = open('pw_data.dat', 'rb')
data = cPickle.Unpickler(fname)
pw = data.load()
load_pw(pw)
finally:
fname.close()
else:
pw = {}
menu()