file_name = 'passwd.txt'
def read():
with open(file_name, 'r', encoding='utf-8') as f:
data = f.read().split('\n')
data = [i for i in data if i]
return data
def add():
website = input('Input website:')
password = input('Input password:')
with open(file_name, 'a', encoding='utf-8') as f:
f.write('{}\t{}'.format(website, password) + '\n')
def query():
lines = read()
print('查询结果为{}条'.format(len(lines)))
for index, line in enumerate(lines):
print("{}\t{}".format(index + 1, line))
def delete():
input_website = input('Input delete website:')
lines = read()
for line in lines:
read_website = line.split('\t')[0]
if read_website == input_website:
lines.remove(line)
print('{} delete success!'.format(input_website))
break
else:
print('{} delete error!'.format(input_website))
with open(file_name, 'w', encoding='utf-8') as f:
f.write('\n'.join(lines) + '\n')
def modify():
input_website = input('Input modify website:')
lines = read()
for line in lines:
read_website = line.split('\t')[0]
if read_website == input_website:
print('current 【{}】 exist,value is 【{}】'.format(input_website, line))
new_website = input('Input new website:')
new_password = input('Input new password:')
lines[lines.index(line)] = "{}\t{}".format(new_website, new_password)
print('modify success!')
break
else:
print('current 【{}】 not find!,modify failed!'.format(input_website))
with open(file_name, 'w', encoding='utf-8') as f:
f.write('\n'.join(lines) + '\n')
query()
while True:
print('=========================')
print('增加:1')
print('删除:2')
print('修改:3')
print('查询:4')
print('退出:0')
print('=========================')
s = input('Input id:')
if s == '0':
print('已退出')
break
elif s == '1':
add()
elif s == '2':
delete()
elif s == '3':
modify()
elif s == '4':
query()
else:
print('Id input error!')