python员工信息管理

#!/usr/bin/env python
from sys import exit
from os import system
from getpass import getpass
data_file = 'staff_list.txt'
data = open(data_file).readlines()
# define user and password
_user = 'root'
_passwd = 'zl'
# define _search, _add method
def _search(key):
    result = []
    for line in data:
        newline = line.strip().split()
        if key in newline:
            result.append(line)
    return result
def _add(values):
    n = 0
    for line in values:
        data.append(line)
        if line in data:
            n += 1
while True:
    username = raw_input('\033[32;1mPlease Enter your username:\033[0m')
    if not username:
        print '\033[31;1mNot allow empty for username!\033[0m'
        continue
    elif username == _user:
        while True:
            passwd = getpass('\033[32;1mEnter Password:\033[0m')
            if not passwd:
                print '\033[31;1mNot allow empty for password, try again!'
                continue
            elif passwd and passwd == _passwd:
                print 'Hi, \033[35;1m%s\033[0m, Welcome to back!' % username
                  
                ## start search,add,modify,delete
                while True:
                    select = "Operation options:\n\t1. Search\n\t2. Add\n\t3. Modify\n\t4. Delete\n You can press 'q' to exit.\n>"
                    user_select = raw_input(select)
                    if user_select.lower() == 'q':
                        print '\033[32;1msystem exiting...\033[0m'
                        exit()
                      
                    ## 1. search
                    elif user_select == '1':
                        while True:
                            search_key = raw_input("\033[32;1mEnter something to search, press 'q' to quit: \
                                    \033[0m\nYou can enter ID, UserName, Job or PhoneNumber.\n>").strip()
                            if search_key == 'q':
                                print '\033[32;1mexiting search mode...\033[0m'
                                break
                              
                            # call _search function to get search result
                            search_result = _search(search_key)
                            if search_result:
                                print '\033[36;1mSearched item:\033[0m'
                                for ret in search_result:
                                    print ret,
                            else:
                                print '\033[31;1mNot matched.\033[0m'
                    ## 2. add items
                    elif user_select == '2':
                        add_prompt = "\033[32;1mInput format reference, press 'q' to exit: \
                                      \033[0m \nID  UserName  Job  PhoneNumber\n>"
                        while True:
                            add_input = raw_input(add_prompt).strip()
                            if add_input == 'q':
                                print '\033[32;1mexiting add mode...\033[0m'
                                break
                            add_input_list = add_input.split()
                            if len(add_input_list) == 4:
                                input_id, input_user, input_job, input_tel = add_input_list
                            else:
                                print '\033[31;1mID, UserName, Job or PhoneNumber parameter is incorrect.\033[0m'
                                continue
                            # Determine whether there ID
                            for line in data:
                                newline = line.strip().split()
                                if input_id == newline[0]:
                                    print "\033[31;1mThe ID '%s' already exists, please re-enter.\033[0m" % input_id
                                    break
                            else:
                                if input_id and input_user and input_job and input_tel:
                                    # add input_line to data
                                    user_line = "\t".join(add_input_list)+"\n"
                                    data.append(user_line)
                                    if user_line in data:
                                        print '\033[35;1mSuccessfully added an records.\033[0m'
                                      
                                    # write data to file
                                    open(data_file, 'w').writelines(data)
                                    # sort data_file
                                    cmd = 'sort -n staff_list.txt >tmp_staff.txt; cat tmp_staff.txt > staff_list.txt; rm -f tmp_staff.txt'
                                    system(cmd)
                    ## 3. modify items
                    elif user_select == '3':
                        modify_prompt = "\033[32;1mWhat you want to modify, press 'q' to exit:\033[0m \
                                        \nUserName, Job=IT or UserName, Phone=123456789\n>"
                        while True:
                            modify_input = raw_input(modify_prompt).strip()
                            if modify_input == 'q':
                                print '\033[32;1mexiting modify mode...\033[0m'
                                break
                            name, modify_item = [ s.strip() for s in modify_input.split(',')]
                            #print 'name=%s, modify_item=%s' % (name, modify_item)
                            item_key, item_value = modify_item.split('=')
                            #print 'item_key=%s, item_value=%s' %(item_key, item_value)
                            fina_data = []
                            for dataline in data:
                                newline = dataline.split()
                                username = newline[1]
                                #print username, job, phone
                                  
                                if username == name:
                                    if item_key == 'Job':
                                        newline[2] = item_value
                                    elif item_key == 'Phone':
                                        newline[3] = item_value
                                    #print newline
                                    this_line = ''
                                    for s_item in newline:
                                        this_line += s_item+'\t'
                                    this_line = this_line.strip()+"\n"
                                    fina_data.append(this_line)
                                else:
                                    fina_data.append(dataline)
                            # write in file
                            try:
                                fp = open(data_file, 'w')
                                for fina_line in fina_data:
                                    print fina_line,
                                    fp.write(fina_line)
                            except IOError, err:
                                print 'Error:', err
                            else:
                                fp.close()
                    ## 4. delete items
                    elif user_select == '4':
                        while True:
                            del_prompt = "\033[32;1mEnter the delete option, press 'q' to exit: \
                                          \033[0m\neg: ID=1 or UserName=Jack or PhoneNumber=11223344\n>"
                            del_input = raw_input(del_prompt).strip()
                            if del_input == 'q':
                                '\033[32;1mexiting delete mode...\033[0m'
                                break
                              
                            del_k, del_v = del_input.split('=')
                            print del_k, del_v
                            for del_line in data:
                                new_deline = del_line.split()
                                if del_v in new_deline:
                                    confirm = raw_input("You select fowing line to delete:\n%s\n\033[33;1mAre you sure[yes/no]?\033[0m" % del_line)
                                    if confirm == 'yes':
                                        data.remove(del_line)
                                        print '\033[35;1mSuccess delelted.\033[0m'
                        # write in file
                        try:
                            fp = open(data_file, 'w')
                            for lines in data:
                                fp.write(lines)
                        except IOError, err:
                            print 'Error:', err
                        else:
                            fp.close()
            else:
                print '\033[31;1mWrong password, try again!\033[0m'
                continue
    else:
        print '\033[31;1mA valid username, Please re-enter!\033[0m'