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()

看完了這篇文章，相信你對“python實現用戶管理系統的方法”有了一定的了解，如果想了解更多相關知識，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！