python员工信息管理_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'