Python作业之用户管理

作业

流程图没有画，懒，不想画

readme没有写，懒，不想写。看注释吧233333

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ == 'Djc'

LOGIN_IN = {'is_login': False}      # 记录当前的普通登录用户


# 普通用户装饰器，此装饰器的功能是验证是否有用户登录
def outer(func):
    def inner():
        if LOGIN_IN['is_login']:    # 检验是否有用户登录了
            func()
        else:
            print("Please login!")  # 若没有，跳转menu函数
    return inner


# 管理员装饰器
def mag_outer(func):
    def inner(*args, **kwargs):
        if bool(int(LOGIN_IN['user_power'])):
            func(*args, **kwargs)
        else:
            print("Sorry,you can't running this function!")
    return inner


def login():
    usr = input("Please enter your name: ")
    pwd = input("Please enter your password: ")

    f = open("user_information", 'r')
    for i in f:
        if usr == i.split("|")[0]:  # 检验登录用户是否存在，不存在跳入注册函数
            break
    else:
        print("This usr is not exist!")

    f.seek(0)
    # 遍历用户文件，登录成功将用户名放在全局字典中
    for line in f:
        # user_power = line.strip().split("|")[3]
        if usr == line.split("|")[0] and pwd == line.strip().split("|")[1]:
            user_power = line.strip().split("|")[3]         # 获取用户权限
            print("Log in successful,welcome to you")
            LOGIN_IN['is_login'] = True
            LOGIN_IN['current'] = usr
            LOGIN_IN['password'] = pwd
            LOGIN_IN['user_power'] = user_power
            break
    else:  # 否则提示密码错误，退出程序
        print("Sorry,your password is wrong!")
    f.close()


# 注册
def register():
    usr = input("Please enter your name: ")
    pwd = input("Please enter your password: ")
    signature = input("Please enter your signature: ")
    f = open("user_information", 'r+')
    flag = True                # 设立一个标识符
    for i in f:
        if i.split("|")[0] == usr:              # 检测此用户名是否存在，存在则不允许注册
            print("Sorry,this name is exist,try again!")
            flag = False
            break

    f.seek(0)                   # 将文件指针位置放回起始位置，这一步非常重要！！！！
    for i in f:
        if flag and i.strip():
            new_usr = '\n' + usr + "|" + pwd + '|' + signature + "|" + '0'
            f.write(new_usr)
            print("Register successful")
            break
    f.close()        # 关闭文件


# 用户查看自己信息，用装饰器装饰。具体可参考装饰器的应用
@outer
def usr_view():
    print("Welcome to you!")
    with open("user_information", 'r+') as f:
        for line in f:
            if line.strip().split("|")[0] == LOGIN_IN['current']:      # 遍历文件，输出此时登录者的信息
                print(line)


# 退出当前用户
@outer
def usr_exit():
    LOGIN_IN['is_login'] = False


# 用户改变密码函数
@outer
def change_pwd():
    print("Welcome to you!")
    new_pwd = input("Please enter the password that you want to modify: ")     # 新密码
    in_put = open("user_information", 'r')
    out_put = open("user_information", 'r+')
    for line in in_put:
        if LOGIN_IN['password'] == line.strip().split("|")[1]:                                         # 验证是否是用户本人想修改密码
            break
    else:
        print("Sorry,you may not own!")
        menu()
    in_put.seek(0)                                                                   # 将文件指针返回文件起始位置
    for line in in_put:
        if "|" in line and LOGIN_IN['current'] in line and LOGIN_IN['is_login']:
            temp = line.split("|")
            temp2 = temp[0] + '|' + new_pwd + '|' + temp[2] + '|' + temp[3]                        # 将新密码写入原文件
            out_put.write(temp2)
        else:
            out_put.write(line)                                                      # 将不需要做改动的用户写入原文件
    print("Yeah,modify successful")
    out_put.close()
    in_put.close()


def user_log():
    pass

'''
-----------------------------------------------------
* 管理员用户登录
* 可以登录，注册，查看本用户信息
* 删除，添加普通用户
* 查看所有普通用户，按照指定关键字搜索用户信息（模糊搜索）
* 提高普通用户权限？
-----------------------------------------------------
'''


@mag_outer
def manager():
    print("Welcome to you,our manager!")
    ma_pr = '''
    (V)iew every user
    (D)elete user
    (A)dd user
    (S)ercher user
    (I)mprove user power
    (B)ack
    (Q)exit
    Enter choice:'''
    while True:
        try:
            choice = input(ma_pr).strip()[0].lower()
        except (KeyError, KeyboardInterrupt):
            choice = 'q'
            
        if choice == 'v':
            view_usr()
        if choice == 'd':
            name = input("Please enter the name that you want to delete? ")
            count = 0
            with open('user_information', 'r') as f:
                for line in f:
                    if line.strip().split("|")[0] == name:
                        delete_usr(count)
                        break
                    count += 1

        if choice == 'a':
            add_user()
        if choice == 's':
            find_user()
        if choice == 'i':
            improve_user()
        if choice == 'b':
            menu()
        if choice == 'q':
            exit()


# 查看所有用户的信息
@mag_outer
def view_usr():
    with open("user_information", 'r+') as f:
        for line in f:
            print(line, end='')


# 管理员删除用户
@mag_outer
def delete_usr(lineno):
    fro = open('user_information', "r")        # 文件用于读取

    current_line = 0
    while current_line < lineno:
        fro.readline()
        current_line += 1            # 将文件指针定位到想删除行的开头

    seekpoint = fro.tell()           # 将此时文件指针的位置记录下来
    frw = open('user_information', "r+")      # 文件用于写入，与用于读取的文件是同一文件
    frw.seek(seekpoint, 0)           # 把记录下来的指针位置赋到用于写入的文件

    # read the line we want to discard
    fro.readline()  # 读入一行进内内存 同时！ 文件指针下移实现删除

    # now move the rest of the lines in the file
    # one line back
    chars = fro.readline()           # 将要删除的下一行内容取出
    while chars:
        frw.writelines(chars)        # 写入frw
        chars = fro.readline()       # 继续读取，注意此处的读取是按照fro文件的指针来读

    print("Delete successful!")
    fro.close()
    frw.truncate()                   # 截断，把frw文件指针以后的内容清除
    frw.close()


@mag_outer
def find_user():
    name = input("Please enter the name that you want to find: ")
    with open("user_information", "r") as f:
        for line in f:
            if line.strip().split("|")[0] == name:
                print(line)
                break


@mag_outer
def improve_user():
    name = input("Please enter the name that you want to find: ")
    power = input("What's power do you want to give ?")
    in_put = open("user_information", 'r')
    out_put = open("user_information", 'r+')
    for line in in_put:
        if line.strip().split("|")[0] == name:
            temp = line.split("|")
            temp1 = temp[0] + '|' + temp[1] + '|' + temp[2] + '|' + power + '\n'
            out_put.write(temp1)
        else:
            out_put.write(line)
    in_put.close()
    out_put.close()


@mag_outer
def add_user():
    register()


@mag_outer
def manager_log():
    import logging

    logger = logging.getLogger(LOGIN_IN['current'])           # 设立当前登录的管理员为logger
    logger.setLevel(logging.DEBUG)                             # 设立日志输出等级最低为DEBUG

    file_handler = logging.FileHandler('user_log')            # 创建向文件输出日志的handler
    file_handler.setLevel(logging.DEBUG)                       # 文件中日志输出等级最低为DEBUG

    formatter = logging.Formatter("%(asctime)s  %(name)s  %(levelname)s  %(message)s")  # 设立日志输出格式
    file_handler.setFormatter(formatter)                       # 将格式添加到handler中

    logger.addHandler(file_handler)                            # 将handler注册到logger

    logger.debug()


# 菜单函数
def menu():
    pr = '''
    (L)ogin
    (R)egister
    (U)ser view
    (C)hange pwd
    (M)anager
    (E)xit usr
    (Q)exit
    Enter choice:'''
    while True:
        try:
            choice = input(pr).strip()[0].lower()
        except (KeyboardInterrupt, KeyError):
            choice = 'q'
    
        if choice == 'l':
            login()
        if choice == 'r':
            register()
        if choice == 'u':
            usr_view()
        if choice == 'c':
            change_pwd()
        if choice == 'm':
            manager()
        if choice == 'e':
            usr_exit()
        if choice == 'q':
            exit()


if __name__ == '__main__':
    menu()