#! /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 + "|" + ''

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