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

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 = "\n" + 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("\033[32m---------用户 %s 删除成功---------\033[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="")