利用列表对文件进行增删改查

wwwlwscom

1. import os
   
2. import os.path
   
3. 
   
4. while True:
   
5.     #进入之前要验证用户名与密码
   
6.     user = input('Please input username:').strip()
   
7.     while user == '' or user != 'roy':
   
8.         user = input('Please input username:').strip()
   
9.         continue
   
10.     else:
    
11.         password = input('Please input password:').strip()
    
12.         while password == '' or password != 'abc':
    
13.             password = input('Please input password:').strip()
    
14.             continue
    
15.     print('Welcome enter system!!!')
    
16.    
    
17.     #不断循环要操作的项目，输入q或quit就退出
    
18.     while True:
    
19.         print('1. Add User\n2. Modify User\n3. Del User\n4. Find User\n')
    
20.       
    
21.         #把文件读出来放到列表
    
22.         f = open('content.txt','r')
    
23.         li = f.readlines()
    
24.         f.close()
    
25.         num = 0
    
26. 
    
27.         choose = input('Please choose:').strip().lower()
    
28. 
    
29.         if choose == 'q' or choose == 'quit':
    
30.             print('ByeBye!!')
    
31.             break
    
32.         #增加用户
    
33.         elif choose == '1':
    
34.             fadd = open('content.txt','w')
    
35.             u = input('Please input id:')
    
36.             p = input('Please input name:')
    
37.             s = input('Please input sex:')
    
38.             d = input('Please input Department:')
    
39.             li.append('\n'+u+'\t'+p+'\t'+s+'\t'+d+'\t')
    
40.             fadd.writelines(li)
    
41.             fadd.flush()
    
42.             continue
    
43. 
    
44. 
    
45.         #修改文件内容
    
46.         elif choose == '2':
    
47.             co = 0
    
48. 
    
49.             ok = input('Please input old key:')
    
50.             nk = input('Please input new key:')
    
51.             
    
52.             frm = open('content.txt','w')
    
53. 
    
54.             for lines in li:
    
55.                 if ok in li[co]:
    
56.                     li[co] = li[co].replace(ok,nk)
    
57.                 co += 1
    
58.             frm.writelines(li)
    
59.             frm.flush()
    
60.             continue
    
61. 
    
62.         #删除用户
    
63.         elif choose == '3':
    
64.             co = 0
    
65.             rm = input('Please input remove key index:')
    
66.             frm = open('content.txt','w')
    
67. 
    
68.             for lines in li:
    
69.                 if rm in li[co]:
    
70.                     del li[co]
    
71.                 co += 1
    
72.             frm.writelines(li)
    
73.             frm.flush()
    
74.             continue
    
75. 
    
76.         #查找内容
    
77.         elif choose == '4':
    
78.             find_name = input('Please input find key words:').strip()
    
79.             for line in li:
    
80.                 if find_name in line:
    
81.                     print (line)
    
82.                     num += 1
    
83.             else:
    
84.                 if num == 0:
    
85.                     print('Not a valid key words!')
    
86.                     continue
    
87.                 else:
    
88.                     print('Find %d data' % num)
    
89.                     num = 0
    
90.             continue
    
91.         else:
    
92.             print('Input error!')
    
93.             continue
    
94. 
    
95.     f.close()
    
96.     break

复制代码

conn

:titter:

骑猪撞宝马

  有点看不懂。。。