#!/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()