记住密码

转自：http://www.cnblogs.com/way_testlife/archive/2012/01/07/2298755.html

刚开始学python，看到这篇文章，果断学习了下，收获良多，感谢这位兄弟。

自己模仿着写了份-_-:

#!/usr/bin/env python
# -*- coding: utf-8 -*-
' key save '
__author__ = ''
try:
    import json
except ImportError:
    import simplejson as json
__all__ = ['fShow', 'fAddAccount', 'fDelAccount', 
            'fSearch', 'fModify']
def fShow():
    if dSitePassport:
        for k, v in dSitePassport.items():
            for kV, vV in v.items():
                print k, kV, vV
    else:
        print "empty..."
def fAddAccount():
    site = raw_input('Input the site:')
    uid = raw_input('Input your account id:')
    passwd = raw_input('Input you password:')
    if site in dSitePassport:
        dValue = dSitePassport[site]
        if uid in dValue:
            print 'This account id is already exist.'
            return 
        else:
            dValue[uid] = passwd
    else:
        dTemp = {}
        dTemp[uid] = passwd
        dSitePassport[site] = dTemp
        dTemp = {}
def fDelAccount():
    while True:
        site = raw_input('Input the site you want to delete:')
        if site not in dSitePassport:
            print 'Cannot find: %s' % site
            con = raw_input('Do you want to continue? y(es) or n(o)?')
            if con.upper() not in ('Y', 'YES'):
                return 
        else:
            break
    dValue = dSitePassport[site]
    print 'the account of %s list as follows:' % site
    for k, v in dValue.items():
        print k, v
    if len(dValue) == 1:
        del dSitePassport[site]
    else:
        uid = raw_input('Input the account id you want to delete:')
        if uid not in dValue:
            print 'Input error!'
            return 
        else:
            dValue.pop(uid)
    print 'delete successful'
def fSearch():
    site = raw_input('Input the site you want to find:')
    if site not in dSitePassport:
        print 'Sorry, cannot find [%s].' % site
        return 
    dValue = dSitePassport[site]
    for k, v in dValue.items():
        print k, v
def fModify():
    while True:
        site = raw_input('Input the site you want to modify:')
        if site not in dSitePassport:
            print 'Can not find %s.' % site
            con = raw_input('Do you want to continue? y(es) or n(o)?')
            if con.upper() not in ('Y', 'YES'):
                return 
        else:
            break
    dValue = dSitePassport[site]
    for k, v in dValue.items():
        print k, v
    if len(dValue) != 1:
        OldId = raw_input('Input the account id you want to modify:')
        if OldId not in dValue:
            print 'Input error'
            return
        dValue.pop(OldId)
    NewId = raw_input('Input the new account id:')
    NewPasswd = raw_input('Input the new password:')
    dValue[NewId] = NewPasswd
            
dMenu = {'s':fShow, 'a':fAddAccount,
         'd':fDelAccount, 'f':fSearch,
         'm':fModify}
def fMenu():
    Menu = '''
(s)how (a)dd (d)elete 
(f)ind (m)odify (q)uit
Enter choice:'''
    while True:
        try:
            MenuChoice = raw_input(Menu).strip()[0].lower()
        except (EOFError, KeyboardInterrupt, IndexError):
            MenuChoice = 'q'
        print 'you picked %s' % MenuChoice
        if MenuChoice not in 'sadfmq':
            print 'Invalid option. Try again.'
            continue
        if MenuChoice == 'q':
            jsonFile = json.dumps(dSitePassport)
            open('passport.db', 'w').write(jsonFile)
            break
        dMenu[MenuChoice]()
if __name__ == '__main__':
    try:
        dSitePassport = json.loads(open('passport.db').read())
        dPassport = {}
        for k, v in dSitePassport.items():
            for kV, vV in v.items():
                dPassport[kV] = vV
    except:
        print 'There is no passport.db, ready to creatde.'
        dSitePassport = {}
        dPassport = {}
    fMenu()