用python實現簡單用戶管理系統

《python核心編程》第七章練習題第五題

一:題目描述：

 userpw2.py。下面的問題和例題7.1中的管理名字-密碼的鍵值對數據程序有關。

(a) 修改那個腳本，使他能記錄用戶上次的登陸日期和時間（用time模塊），並與用戶密碼一起保存起來。程序的界面要求用戶輸入用戶名和密碼的提示。無論用戶名是否登陸成功，都應有提示，在用戶登陸成功后，應更新相應用戶的上次登陸時間戳。如果本次登陸與上次登陸在時間上相差不超過四個小時，則通知該用戶：“You already in at :<last_login_timestamp>”。

(b) 添加一個管理菜單，其中有以下兩項：（1）刪除一個用戶 （2）顯示系統中所有用戶的名字和他們密碼的清單。

(c) 口令目前沒有加密。請添加一段對口令加密的代碼

(d) 為程序添加圖形界面，例如，用Tkinter。(圖形化界面開發比較復雜，這里沒有用到。)

(e) 要求用戶名布區分大小寫。

(f ) 加強對用戶名的限制，不允許符號和空白符。

(g) 合並“新用戶”和“老用戶”兩個選項。如果一個新用戶試圖用一個不存在的用戶名登陸，詢問該用戶是否是新用戶，如果是肯定的，就創建該用戶。否則按老用戶的方式登陸。

 

 

二:

程序中用到的模塊解釋如下：

(1) re：正則表達式引擎，python中調用正則表達式的方法

(2) pickle：對象持久化，將數據寫入到磁盤中

(3) datetime：時間處理，用於記錄用戶登陸時間戳

(4) base64：base64加密模塊

(5) hashlib：hash加密模塊

 

完整代碼:

 

#-*- coding:utf-8 -*-  
#2017.7.17 
  
import re  
import pickle  
import base64,hashlib  
from datetime import datetime     
  
def Initialization(file_name):  
    '''''程序初始化，創建user.ini和time.ini文件'''  
    dict_test={'admin':'db69fc039dcbd2962cb4d28f5891aae1'}  #創建超級管理員，默認密碼為admin  
    f = file(file_name,'a+')   #以追加的方式打開文件，避免文件被修改  
    if len(f.readlines()) ==0:  #判斷程序是否為空，只在第一次運行的時候初始化  
        if file_name=='user.ini':  
            pickle.dump(dict_test, f, True)  
        else:  
            pickle.dump({},f, True)  
    f.close()  
  
def encodepass(passwd):  
    '''''采用base64和md5雙層加密，破解可能幾乎為0'''  
    m = hashlib.md5()  
    pwd = base64.b64encode(passwd)  
    m.update(pwd)  
    return m.hexdigest()  
  
def time_order(user):  
    '''''記錄用戶登陸時間，結果保存在time.ini文件中'''  
    ft = file('time.ini','r')  
    dbt = pickle.load(ft)  
    if user not in dbt:  
        dbt.setdefault(user,datetime.today())  
    else:  
        time_value = dbt[user]  
        t = datetime.today()-time_value  
        try:  
            if t.hour<=4:  
                print 'You already logged in at:<last_login_timestamp>'  
        except:  
            print 'You already logged in at:<last_login_timestamp>'  
        dbt[user] = datetime.today()  
    ft = file('time.ini','w')  
    pickle.dump(dbt, ft, True)  
    ft.close()  
          
  
def newuser(db):  
    '''''用戶創建程序，由olduser調用'''  
    while True:  
        name = raw_input('Please input the username:')  
        if re.match(r'\w', name):   #采用正則表達式檢測用戶名是否合法  
            pass  
        else:  
            print 'Username should be made of A~Z、a~z、0~9、_'  
            continue  
        for valuename in db.keys():  
            if name.lower() == valuename.lower():  
                break  
        else:  
            break  
    passwd = raw_input('Please input the password:')  
    db[name] = encodepass(passwd)  
      
def olduser(db):  
    '''''用戶登陸程序'''  
    name = raw_input('Login:')  
    if name in db:  
        pwd = raw_input('passwd:')  
        passwd = db.get(name)  
        if passwd == encodepass(pwd):  
            print 'Welcome back!',name  
            time_order(name)  
        else:  
            print 'Login incorrent!'  
    else:  
        YN = raw_input('Do you want to instead a new user? Yes or No:')  
        if YN.lower()=='yes':  
            newuser(db)  
    print '\n',  
          
def deluser(db):  
    '''''刪除一個用戶，但必須以管理員的身份'''  
    print 'Please login as admin'  #管理員的身份才能刪除用戶  
    name = raw_input('Login:')  
    pwd = raw_input('passwd:')  
    passwd = db.get(name)  
    if passwd == encodepass(pwd) and name=='admin':  
        user = raw_input('Please input a user name:')  
        if user != 'admin':  
            if db.pop(user):  
                print 'Delete Current!'  
        else:  
            print 'Con not delete admin!'  
    else:  
        print 'Wrong passwprd'  
      
def checkuser(db):  
    '''''查看所有用戶，但必須以管理員的身份'''  
    print 'Please login as admin'  #管理員的身份才能查看所有用戶  
    name = raw_input('Login:')  
    pwd = raw_input('passwd:')  
    passwd = db.get(name)  
    if passwd == encodepass(pwd) and name == 'admin':  
        for key in db:  
            print 'username: %10s ====> password: %10s' % (key,db[key])  
    else:  
        print 'You can not check all users!'  
  
def resetuser(db):  
    '''''修改密碼，但必須正確的輸入老密碼'''  
    name = raw_input('Please input the username：')  
    passwd  = raw_input('Please input old password：')  
    if db[name] == encodepass(passwd):  
        passwd = raw_input('Please input new password：')  
        db[name] = encodepass(passwd)  
    else:  
        print 'Wrong password!'  
  
def showmenu():  
    '''''程序運行的主函數'''  
    fu = file('user.ini','r')  
    db = pickle.load(fu)  
    prompt = '''''(L)ogin Now 
(Q)uit 
(D)elet User 
(C)heck All User 
(R)eset Password 
Enter choice:'''  
      
    done = False  
    while not done:  
        chosen = False  
        while not chosen:  
            try:  
                choice = raw_input(prompt).split()[0].lower()  
            except (EOFError,KeyboardInterrupt):  
                choice = 'q'  
            print '\nYou picked:[%s]' % choice  
            if choice not in 'lqdcr':  
                print 'invalid option,try again'  
            else:  
                chosen = True  
                      
        if choice == 'q':done = True  
        if choice == 'l':olduser(db)  
        if choice == 'd':deluser(db)  
        if choice == 'c':checkuser(db)  
        if choice == 'r':resetuser(db)  
          
    fu = file('user.ini','w')  
    pickle.dump(db,fu,True)  
    fu.close()  
          
          
if __name__ == '__main__':  
    '''''系統有一個用戶名為admin 密碼為admin的超級用戶,請立即修改密碼！'''  
    print 'Welcome to User Information Management System!'  
    Initialization('user.ini')  
    Initialization('time.ini')  
    showmenu()