python 简单的用户登陆

sofh7777

#!/usr/bin/env python　　
#coding:utf-8
　　
import hashlib
　　
from datetime import datetime
　　
ul = {}
　　
def newuser():
　　
    a = []
　　
    prompt = "Please enter username again:"
　　
    while True:
　　
        name = raw_input(prompt).lower()
　　
        if not name.isalnum() and '' in name :
　　
            print('name format error')
　　
            continue
　　
        else :
　　
            if ul.has_key(name):
　　
                prompt = "username already exists"
　　
                continue
　　
            else:
　　
                break
　　
    passwd = raw_input('enter password:')
　　
    m=hashlib.md5()
　　
    m.update(passwd)
　　
    a.append(m.hexdigest())
　　
    a.append(datetime.now())
　　
    ul[name]=a
　　
    print 'new user is %s ' \
　　
          'regiter time is %s' %(name,ul[name][1])
　　
def olduser():
　　
    name= raw_input('Please enter username again:').lower()
　　
    passwd = raw_input('Please enter your password:')
　　
    m=hashlib.md5()
　　
    m.update(passwd)
　　
    pwd = ul.get(name)
　　
    if pwd[0]==m.hexdigest():
　　
        newtime = datetime.now()
　　
        if (newtime-ul[name][1]).days == 0 and (newtime-ul[name][1]).seconds < 14400:
　　
            print 'you already logged in at %s: ' %(ul[name][1])
　　
        else:
　　
            pwd[1]=newtime
　　
            print 'welcome back %s, login time is %s' %(name,passwd[1])
　　
    else:
　　
        print 'login incorrect'
　　
def removeuser():
　　
    print ul
　　
    name=raw_input('input a user name to remove: ').lower()
　　
    if name in ul:
　　
        ul.pop(name)
　　
        print 'remove successful'
　　
    else:
　　
        print 'this uesr not exist'
　　

　　
def login():
　　
    while True:
　　
            name=raw_input('Please enter username:').lower()
　　
            if not name.isalnum() and '' in name:
　　
                print 'name format error'
　　
                continue
　　
            else:
　　
                if not ul.has_key(name):
　　
                    print 'user name is not in userlist'
　　
                    answer=raw_input('register a new user? (y/n):').lower()
　　
                    if 'y'==answer:
　　
                        newuser()
　　
                        break
　　
                    elif 'n'==answer:
　　
                        break
　　
                else:
　　
                    print 'user name is already in db'
　　
                    olduser()
　　
                    break
　　

　　
def showmenu():
　　
    prompt = """
　　
    (U)ser login
　　
    (R)emove user
　　
    (Q)uit
　　
    Enter choice: """
　　
    done = False
　　
    while not done:
　　
        chosen = False
　　
        while not chosen:
　　
            try:
　　
                choice = raw_input(prompt).strip()[0].lower()
　　
            except (EOFError,KeyboardInterrupt):
　　
                choice= 'q'
　　
            print"\nYou picked [%s]" %choice
　　
            if choice not in 'urq':
　　
                print("invalid option,try again")
　　
            else :
　　
                chosen = True
　　
        if choice == 'q':done = True
　　
        if choice == 'u':login()
　　
        if choice == 'r':removeuser()
　　

　　
if __name__ == '__main__':
　　
    showmenu()