《Python核心编程》第二版第405页第十三章练习 续二 -Python核心编程答案-自己做的-

这是自己做的练习，可能有错误，欢迎讨论和各种优化重构方案。
根据读者反馈，或者code review，我对本篇文章答案或者相关内容的更新补充，一般会被添加在本篇博客的评论中。
我尽量保证每题的答案代码是完整的，不仅仅是函数或者类，打开Python 2.7的IDLE，将代码完整拷贝进去，就能调试运行。
欢迎访问Balian在博客园的家http://www.cnblogs.com/balian

13-4.
用户注册。建立一个用户数据库（包括登录名、密码和上次登录时间戳）类（参考练习7-5和练习9-12），来管理一个系统。该系统要求用户在登录后才能访问某些资源。这个数据库类对用户进行管理，并在实例化操作时加载之前保存的用户信息，提供访问函数来添加或更新数据库的信息。在数据修改后，数据库会在垃圾回收时将新信息保存到磁盘（参见__del__()）。
【注】
我使用D盘EX13_4文件夹里的accounts.txt文件记录用户名和密码。只有一个用户Balian，密码是apple。用同位置的文件userlog.txt记录上次登录时间戳。如果用户Balian登陆正确，就返回上次登陆时间，和登陆成功信息。如果用户名或者密码错误，都将提示登陆失败，失败3次退出程序。对用户进行管理指如果登陆成功，可更新密码。这样做的目的是想使代码尽量简单，毕竟这只是一个练习题。
accounts.txt文件内容（仅一行）
balian 123456
userlog.txt为一个空文本文件
运行程序前，请提前准备以上的文件夹和文件。

【答案】
代码如下：

#-*- encoding: utf-8 -*-

class DBMnagement(object):
    '数据库管理类'
    
    def __init__(self, nm, pwd):
        self.name = nm
        self.password = pwd
        
    def updatePWD(self, newpwd):
        self.password = newpwd
        
    def __del__(self):
        pass
        
#import os
import time

def CreateLog(username):
    '登陆成功写时间戳到log文件'
    timeStamp = time.ctime(time.time())
    log = 'User' + ' - ' + username + ', last time login Succeed at ' + timeStamp + '\n'
    logFile = open('d:\\EX13_4\\userlog.txt', 'a')
    logFile.write(log)
    logFile.close()        

def ReadLog():
    '登陆成功，从log文件读取上次时间戳'
    logFile = open('d:\\EX13_4\\userlog.txt')
    logInfo = logFile.readlines()
    if len(logInfo) >= 1:
        print logInfo[-1]
    logFile.close() 
    
def ChangePWDinLog(username, newpwd):
    '修改用户密码'
    credentialFile = open('d:\\EX13_4\\accounts.txt', 'w') 
    credentialtxt = username + ' ' + newpwd
    credentialFile.write(credentialtxt)
    credentialFile.close()
    print '密码已修改 ... Password updated successfully. \n'

# 从这里开始是主程序
print '欢迎使用最简单的用户管理系统'
print '尝试用系统中已经存在的用户名和密码登录，有三次机会'
print '登陆成功后输入 updatepwd 命令可以修改密码，成功后退出系统'
print '登陆成功后输入 quit 命令退出系统\n'
    
credentialFile = open('d:\\EX13_4\\accounts.txt') # 从文本文件中获取用户名和密码
credentialInfo = credentialFile.readlines()
credentialFile.close()

usernameinDB = credentialInfo[-1].split()[0] # 这里credentialInfo[-1]是一个字符串，包括用户名和口令，空格隔开的
passwordindm = credentialInfo[-1].split()[1]

errorTimes = 3
while errorTimes:
    print '请输入用户名和密码：'
    input_username = raw_input('Username ... :  ')
    input_password = raw_input('Password ... :  ')
    if input_username == usernameinDB and input_password == passwordindm:
        print '登陆成功  ... Login Successful'
        CurrentUser = DBMnagement(input_username, input_password)
        ReadLog()
        CreateLog(input_username)
        input_command = raw_input('请输入你的命令, quit or updatepwd ... \n')
        if input_command == 'quit':
            del CurrentUser
            print '\n已退出系统  ... Logout successfully.'
            break
        elif input_command == 'updatepwd':
            newpwd = raw_input('\n请输入新密码, new password ... :  ')
            CurrentUser.updatePWD(newpwd)
            ChangePWDinLog(CurrentUser.name, CurrentUser.password)
            del CurrentUser
            print '\n已退出系统  ... Logout successfully.'
            break
        else:
            print '错误命令  ... Wrong Command'
            del CurrentUser
            print '\n已退出系统  ... Logout successfully.'
            break
    else:
        print '用户名或口令错误  ... Wrong Username or Password.\n'
        errorTimes -= 1
        if errorTimes == 0: print '已退出系统  ... Logout successfully.'
         

【执行结果】

欢迎使用最简单的用户管理系统
尝试用系统中已经存在的用户名和密码登录，有三次机会
登陆成功后输入 updatepwd 命令可以修改密码，成功后退出系统
登陆成功后输入 quit 命令退出系统

请输入用户名和密码：
Username ... :  balian
Password ... :  123
用户名或口令错误  ... Wrong Username or Password.

请输入用户名和密码：
Username ... :  balian
Password ... :  123456
用户名或口令错误  ... Wrong Username or Password.

请输入用户名和密码：
Username ... :  balian
Password ... :  apple
登陆成功  ... Login Successful
User - balian, last time login Succeed at Wed Sep 12 12:32:35 2012

请输入你的命令, quit or updatepwd ... 
updatepwd

请输入新密码, new password ... :  egg
密码已修改 ... Password updated successfully. 


已退出系统  ... Logout successfully.            
            
        

【参考】

http://blog.csdn.net/lgfei/article/details/92044
Python中写文本文件的方法
http://magustest.com/blog/softwaretesting/how-to-write-a-file-by-using-python/