快一个月木有更新了，在这段时间内，博主除了忙工作外，就一直在研究python编程，据圈内朋友闲聊，python应该是未来一个运维攻城狮除了shell以外必备的LINUX脚本语言，所以就硬着头皮，按照老男孩的python视频讲座，把自己的研究成果贴出来，目前还没学完，期待老男孩的讲座更加给力些。
PS: 1.老男孩的python视频讲座很不错，希望能继续免费下去。。。
2.博主编程很烂，有什么地方有问题，还请多多指教。。。
3.最近买了本PYTHON核心编程，啃书ing。。。
系统环境: Ubuntu 12.10 server 64-bit
PYTHON: Python 2.7.3
# mkdir python
# vi contact_list.txt
-------------------------
1 zhangsan IT 0001
2 lisi IT 0002
3 wangwu HR 0003
4 zhaoliu OP 0004
-------------------------
# vi staff_list.py
----------------------------------------------------
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
|
#!/usr/bin/env python#encoding:utf-8 # 支持中文输入import sys
import getpass
import shutil
import commands
import time
import fileinput
staff_list = 'contact_list.txt'
# 参数配置user = 'admin'
passwd = '123456'
s = file(staff_list)
ss = s.readlines()
a = file(staff_list,'a')
counter = 0
_counter = 0
# 认证登陆while True:
# 计数器,超过3次强制退出
if counter <= 3:
# 空用户名判断
name = raw_input("please input your name: ").strip()
if len(name) == 0:
print "empty name,try again!"
continue
# 用户名密码判断,密码隐藏
# pwd = raw_input("please input your password: ")
pwd = getpass.getpass('please input your password:')
if pwd == passwd and name == user:
print "Welcome to login,%s" %name
else:
print "name or password is not valid,please try again!"
counter +=1
continue
break
else:
print "exceeded 3 times user login..exit the script"
sys.exit()
# 选择增删改查while True:
item = raw_input('''\033[36;1mWelcome to login %s, what do you want to do?
-----------------------press 'p' for printpress 'a' for addpress 'd' for deletepress 'u' for updatepress 's' for selectpress 'q' for quit-----------------------please make your choise: \033[0m''' % user)
# 打印所有
if item == 'p':
while True:
user_select = open(staff_list,'r')
s_ = user_select.read()
print ' '
print '\033[32;1mThe content of the file\033[0m '
print '\033[32;1m--------------------------\033[0m '
print s_
print '\033[32;1m--------------------------\033[0m '
print ' '
break
# 增加
elif item == 'a':
while True:
user_add_num = raw_input(("\033[32;1mplease input your number: \033[0m ").strip())
user_add_name = raw_input(("\033[32;1mplease input your name: \033[0m ").strip())
user_add_dept = raw_input(("\033[32;1mplease input your department: \033[0m ").strip())
user_add_id = raw_input(("\033[32;1mplease input your id: \033[0m ").strip())
user_item = '%s\t%s\t%s\t%s' %(user_add_num,user_add_name,user_add_dept,user_add_id)
a.write("\n%s" %user_item)
a.flush()
print "\033[32;1mAdd item:\033[0m"
print "\033[32;1m------------------\033[0m"
print user_item
print "\033[32;1m------------------\033[0m"
print "\033[32;1mAdded successful!\033[0m"
# 删除空行
del_blank_in = open('contact_list.txt','r')
del_blank_out = open('contact_list_new.txt','w')
lines = del_blank_in.readlines()
for blank in lines:
if blank.split():
del_blank_out.writelines(blank)
del_blank_in.close()
del_blank_out.close()
# 覆盖原文件
shutil.move('contact_list_new.txt','contact_list.txt')
user_add_choise = raw_input('press Q for quit or press any key to continue: ')
if user_add_choise == 'Q':
print 'bye!'
break
# 删除
elif item == 'd':
while True:
user_del_input = raw_input("please input sth to delete: ").strip()
if len(user_del_input) == 0:
print "empty input,try again!"
else:
# 输入值与源文件比对，有则丢弃，没有则添加到新文件，最后新文件覆盖源文件，实现删除功能
with open('contact_list.txt','r') as ff:
with open('contact_list.txt.new','w') as gg:
for line in ff.readlines():
if user_del_input not in line:
gg.write(line)
if user_del_input in line:
print "\033[32;1mDelete item:\033[0m"
print "\033[32;1m------------------\033[0m"
print " %s " %line
_counter += 1 # 计数器,判断输入值命中次数
print "\033[32;1m------------------\033[0m"
print "\033[32;1mDeleted successful!\033[0m"
if _counter == 0:
print 'nothing delete!'
shutil.move('contact_list.txt.new','contact_list.txt')
# 退出删除
user_del_input_quit = raw_input("\033[32;1mpress Q for quit or press any key to continue? \033[0m").strip()
if user_del_input_quit == 'Q':
break # 查询
elif item == 's':
while True:
match_yes = 0
#输入判断，忽略空格输入,加入颜色
user_select_input = raw_input("\033[32;1mplease input sth to search:\033[0m ").strip()
contact_file = file (staff_list)
if len(user_select_input) == 0:
print "empty input,try again!"
else:
while True:
line = contact_file.readline()
if len(line) == 0:
break
if user_select_input in line:
match_yes = 1
print line
else:
pass
if match_yes == 0 :
print "No match item found"
# 退出查询
user_select_input_quit = raw_input("\033[32;1mpress Q for quit or press any key to continue? \033[0m").strip()
if user_select_input_quit == 'Q':
break # 修改
elif item == 'u':
while True:
# 输入为空以及匹配查询内容判断
user_update_input_from = raw_input("\033[32;1mplease search sth to update: \033[0m").strip()
update_match = 0
update_file = file(staff_list).readlines()
for n_ in range(len(update_file)):
if user_update_input_from in update_file[n_]:
update_match = 1
else:
pass
if update_match == 0:
print "No match item found"
elif len(user_update_input_from) == 0:
print "empty input,try again!"
else:
# 将匹配到的字符修改为新字符
while True:
user_update_input_to = raw_input("\033[32;1mupdate %s to what?\033[0m " %(user_update_input_from)).strip()
if len(user_update_input_to) == 0:
print "empty input,try again!"
else:
for line_ in fileinput.input(staff_list,inplace = 1,backup='.bak'):
line_ = line_.replace(user_update_input_from,user_update_input_to)
print line_
# 打印修改字符的行
print "\033[32;1mupdate item:\033[0m"
output_ = commands.getoutput("diff contact_list.txt contact_list.txt.bak|grep '^>.*' | sed 's/^>//g'")
print "\033[32;1m---------------------------\033[0m"
print output_
print "\033[32;1m---------------------------\033[0m"
print "\033[32;1mupdate successfully!\033[0m"
# 删除空行
del_blank_in = open('contact_list.txt','r')
del_blank_out = open('contact_list_new.txt','w')
lines = del_blank_in.readlines()
for blank in lines:
if blank.split():
del_blank_out.writelines(blank)
del_blank_in.close()
del_blank_out.close()
# 覆盖原文件
shutil.move('contact_list_new.txt','contact_list.txt')
break
# 退出更新
user_update_input_quit = raw_input("\033[32;1mpress Q for quit or press any key to continue? \033[0m").strip()
if user_update_input_quit == 'Q':
break # 退出
elif item == 'q':
print 'bye!'
sys.exit()
else:
print "\033[31;1mnot a valid key word\033[0m"
time.sleep(1)
|
-----------------------------------------------------
本文转自 showerlee 51CTO博客，原文链接：http://blog.51cto.com/showerlee/1308764，如需转载请自行联系原作者