简单的用户登录脚本

#!/usr/bin/env python
import time

db = {}

def new_user():
    prompt = 'login desired: '
    value = []
    while True:
        name = raw_input(prompt)
        if db.has_key(name):
            prompt = 'name taken, try another'
            continue
        else:
            break

    pwd = raw_input('passwrod: ')
    value.append(pwd)
    value.append(None)

    db[name] = value

def old_user():
    prompt = '''
        enter user name please
        login: 
    '''
    name = raw_input(prompt)
    prompt = '''
           enter password name please
           password: 
       '''
    pwd = raw_input(prompt)
    password = db.get(name)[0]
    if password == pwd:
        print 'welcome back ', name
        timestamp = db.get(name)[1]
        if timestamp is None:
            timestamp = time.localtime()
        current_time = time.localtime()
        duration = time.mktime(current_time) - time.mktime(timestamp);
        if duration > 10.0:
            print 'You already logged in at: %.2f' % duration
        db.get(name)[1] = current_time
    else:
        print 'login incorrect'

def managemenu():
    prompt = """
    (D)elete user
    (S)how all user
    Enter chice: 
    """
    try:
        choice = raw_input(prompt).strip()[0].lower()
    except EOFError, KeyboardInterrupt:
        choice = 'q'

    if choice == 'd':
        name = raw_input('user: ')
        del db[name]
    elif choice ==  's':
        for key in db.keys():
            print 'user: ', key, ' password: ', db[key]

def showmenu():
    prompt = """
    (N)ew User Login
    (E)xisting User Login
    (Q)uit
    (M)anage
     Enter chice:
    """
    done = False
    while not done:
        chosen = False
        while not chosen:
            try:
                choice = raw_input(prompt).strip()[0].lower()
            except EOFError, KeyboardInterrupt:
                chioce = 'q'

            print '\n you picked: [%s]' % choice
            if choice not in 'neqm':
                print 'invalid option, try again'
            else:
                chosen = True

        if choice == 'q':
            done = True
        elif choice == 'n':
            new_user()
        elif choice == 'e':
            old_user()
        elif choice == 'm':
            managemenu()


if __name__ == '__main__':
    showmenu()