Python-S13作业-day4-之登陆，管理后台

需求：

本节作业，用户管理程序：
普通用户：

管理员用户：

　　　　　　　　查看所有普通用户，按照指定关键字搜索用户信息；
　　　　　　　　登录，注册，修改密码，查看本用户信息；
删除、添加普通用户
修改普通用户密码
　　　　　　　　提高普通用户权限

注意：

　　　　　　1、用户信息：要存文件里
　　　2、权限验证，要用装饰器

思路：

小感悟：写这个程序让我知道，其实写一个大的程序，

　　　　（1）就是把一个个功能(函数)拼凑起来；

　　　　（2）然后按照正常的逻辑执行调用就行了；

Readme：

1.这个程序共包含两个文件，主程序文件，db存放账号密码的文件；

2.代码结构:

　　　　最上面放的是login()函数；

　　　　中间放的是功能；

　　　　下面放的是入口管理功能；

主程序：

#!/usr/bin/python env
#_*_coding:utf-8 _*_
import sys
import shutil
import os

LOG_USER = {'is_login':False}
#使用字典形式，设置一个标志位，用于判断用户是否登陆成功；


def outer(func):
    def inner(*args,**kwargs):
        if LOG_USER['is_login']:  #判断用户是否登陆
            r = func()
            return r
        else:
            print("Please login.")  #否则提示没有登陆
    return inner
#这个装饰器给普通用户使用，
# 这里判断用户已经登陆，才能执行mamger_user()函数；

def outer1(func):
    def inner(*args,**kwargs):
        if LOG_USER['is_login'] and LOG_USER['user_type'] == '2':  #判断条件
            r = func()
            return r
        else:
            print("权限不够")  ##普通账号会提示“权限不足”；
    return inner
#这个装饰器给管理员账号使用；
#用户登陆，并且是管理员账号，才能执行manger_admin()函数；

def login(user,pwd):
    with open('db','r',encoding='utf-8') as f:
        for line in f:
            line = line.strip().split("|")
            if line[0] == user and line[1] == pwd:
                LOG_USER['is_login'] = True         #如果登陆成功，这里为True
                LOG_USER['user_type'] = line[-1]    #获取用户的UID
                LOG_USER['user_info'] = line[:]     #获取当前登陆用户所有信息
                return True
    return False
#登陆功能

def changepwd(user,new_pwd):
    with open('db','r',encoding='utf-8') as f1,open('db_new','w',encoding='utf-8') as f2:
        for line in f1:
            line_list = line.strip().split("|")
            if user == line_list[0]:
                new_str = line.replace(line_list[1],new_pwd)
                f2.write(new_str)
                continue
            f2.write(line)     #上面条件不成立，会先执行这里
    shutil.move('db','db_bak')  #备份
    os.rename('db_new','db')   #文件覆盖
#修改密码


def deleteUser(username):
    with open('db','r+') as f1,open('db_new','w') as f2:
        for line in f1:
            line_list = line.strip().split("|")
            if line_list[0] == username: #这里如果匹配，对这行文件不操作
                pass
            else:
                f2.write(line)   #把条件不匹配的行写到新文件中
    shutil.move('db','db_bak')
    os.rename('db_new','db')
#删除用户

def register(user,pwd,mail,phone):
    with open('db','a',encoding='utf-8') as f:
        temp = "
" + user + "|" + pwd + "|" + mail + "|" + phone + "|" +"1"
        f.write(temp)
        res = ''.join(temp)
        print("Register Info:",res)
#注册账号

def fetch(content):
    with open('db','r') as f:
        for line in f:
            if content in line:
                print(line.strip())
#模糊查询

def auth(username):
    with open('db','r',encoding='utf-8') as f1,open('db_new','w',encoding='utf-8') as f2:
        for line in f1:
            line_list = line.strip().split("|")
            if username == line_list[0]:
                new_str = line.replace(line_list[-1],"2")
                f2.write(new_str)
                continue
            f2.write(line)
    shutil.move('db','db_back')
    os.rename('db_new','db')
#用户权限升级

@outer1  #先执行outer1装饰器，条件成立再调用manger_admin()函数；
def manger_admin():
    num = input("1.查看所有信息  2.模糊查询 3.添加删除用户  4.密码管理  5.权限管理  6.退出")
    if num == '1':
        with open('db','r',encoding='utf-8') as f:
            res = f.read()
            print(res)
    elif num == '2':
        content = input("请输入要查找的关键字：")
        fetch(content)
    elif num == '3':
        r_num = input("1.register  2.DeleteUser  3.退出程序")
        if r_num == '1':
            user = input("input register user:")
            pwd = input("input register pwd:")
            mail = input("input register pwd:")
            phone = input("input register pwd:")
            register(user,pwd,mail,phone)
        elif r_num == '2':
            user = input("input Delete user:")
            deleteUser(user)
            print("33[32m---------用户 %s 删除成功---------33[0m" %user)
        elif r_num == '3':
            sys.exit()
    elif num == '4':
        old_user = input("input modify is username:")
        new_pwd = input("input new password::")
        changepwd(old_user,new_pwd)
        print("modify %s user is password sucess,new password is :%s"%(old_user,new_pwd))
    elif num == '5':
        user = input("请输入要升级为管理员权限的用户：")
        auth(user)
    elif num == '6':
        sys.exit()

@outer  #先执行outer函数，outer函数条件成立再执行manger_user()函数；
def manger_user():
    r_num = input("1.管理员配置  2.我的配置  3.退出")
    if r_num == '1':
        manger_admin()
    elif r_num == '2':
        rr_num = input("1.查看我的信息  2.修改密码  3.退出")
        if rr_num == '1':
            user = LOG_USER['user_info']
            print(user)
        elif rr_num == '2':
            old_user = input("input modify is username:")
            new_pwd = input("input new password::")
            changepwd(old_user,new_pwd)
        elif rr_num == '3':
            sys.exit()
    elif r_num == '3':
        sys.exit()

def main():
    print("welcome to user manger system".center(40,'*'))
    while True:
        num = input("1.登陆  2.注册  3.后台管理  4.退出")
        if num == '1':
            user = input("username:")
            pwd = input("password:")
            login_status = login(user,pwd)
            if login_status == True:
                print('login sucess')
            else:
                print("login fail.")
        elif num == '2':
            user = input("username:")
            pwd = input("password:")
            mail = input("mail:")
            phone = input("phone:")
            register(user,pwd,mail,phone)
        elif num == '3':
            manger_user()
        elif num == '4':
            sys.exit()
#入口文件

main()

账号密码文件：

admin|123123|faith_wmh@126.com|13141299640|2
xxx|223223|faith_wmh@226.com|242|2
wmh|123|faith@126.com|1221|2
nbbb|123123|faith@|123123|1

流程图：

写作业过程中，学到的内容：

【re.match：把不是1开头的行，查找出来】

import re
for line in open("dbfile"):
    if not re.match("1",line):
        print(line[:-1])

【删除内容为，PermissionError的字段】

lines = [l for l in open("file.txt", "r") if l.find("PermissionError") != 0]
fd = open("file.txt", "w")
fd.writelines(lines)
fd.close()

【输入行号，删除指定行】

import fileinput
num = input("input delete num:")
f = fileinput.input("dbfile",inplace=True)
for line in f:
    if f.filelineno() == num:
        print('')
    else:
        print(line,end="")