import os
count=
0
path=r
'E:\PYTHON学习\excises\day10\infomation.txt'
def make_dic():
'''将文件转化为列表套字典'''
with open(path,'r',encoding=
'utf-8') as f:
l1=[{
'staff_id': i.split(
',')[0],
'name': i.split(
',')[1],
'age': i.split(
',')[2
],
'phone':i.split(
',')[3],
'dept':i.split(
',')[4
],
'enroll_date':i.split(
',')[5]}
for i
in f]
return l1
command=input(
'please input your command>>:')
#输入命令
def deco_delete_add(func):
'''实现删除和添加新用户功能'''
def wrapper(*args,**
kwargs):
cmd=
args[0]
if cmd.isdigit():
#判断输入的命令是查询命令还是员工ID，如果是ID删除用户
for dic
in staff_table:
if cmd.strip() == dic[
'staff_id']:
staff_table.remove(dic)
with open(path,'r',encoding=
'utf-8') as f:
for line
in f:
ls_line=line.split(
',')
if cmd.strip() == ls_line[0]:
#查找文件中与输入的ID相符的那一行
pass
else:
# 不是ID的那些行写入一个新文件里。
with open(r
'E:\PYTHON学习\excises\day10\infomation_copy.txt',
'a',encoding=
'utf-8') as f2:
f2.write(line)
os.remove(path) #将旧文件删除，并将新建的文件覆盖旧文件，实现文件修改
os.rename(
'infomation_copy.txt',path)
elif 'select' in cmd:
#不是ID，执行主函数的查询功能
s=func(*args,**
kwargs)
return s
elif 'UPDATE' in cmd:
up_user(cmd)
else:
#添加用户信息Yang Gao,25,13811223348,Student,2017-06-08
add_user(cmd)
return wrapper
def sl_where(cmd_x):
global count
l_cmd = cmd_x.strip().split(
'where')
cmd_where=l_cmd[1].strip().split(maxsplit=2)
#吧where后面的语句分割为3段
cmd_where[-1]=remove_symbol(cmd_where[-1
])
for dic
in staff_table:
#将列表循环，遍历每一个字典
if cmd_where[1].strip() ==
'>':
if int(dic[cmd_where[0]]) > int(cmd_where[-1
]):
count+=1
yield(dic)
elif cmd_where[1].strip() ==
'=':
if dic[cmd_where[0]] == cmd_where[-1
]:
count+=1
yield(dic)
elif cmd_where[1].strip() ==
'like':
if cmd_where[-1]
in dic[cmd_where[0]]:
count += 1
yield (dic)
def add_user(cmd_add):
flag = 1
cmd_phone = cmd_add.strip().split(
',')[2].strip()
# 将命令中的phone提取出来
for dic
in staff_table:
if cmd_phone == dic[
'phone']:
flag =
0
print(
'当前phone已存在')
if flag:
n = len(staff_table) + 1
# 计算要插入信息的序号
with open(path,
'a', encoding=
'utf-8') as f3:
cmd_add_write =
'\n' + str(n) +
',' +
cmd_add
f3.write(cmd_add_write)
print(
'add user successful')
def up_user(cmd_up):
g2=
sl_where(cmd_up)
s = cmd_up.split(
'SET')[-1].split(
'WHERE')
s = s[0].split(maxsplit=2)
#列表['dept','=','"market"']
s[-1]=s[-1
].strip()
s[-1]=remove_symbol(s[-1
])
for dic
in g2:
dic[s[0]]=s[-1
]
with open('copy_2.txt',
'w',encoding=
'utf-8') as f:
for dic
in staff_table:
line=
'%s,%s,%s,%s,%s,%s'%(dic[
'staff_id'],dic[
'name'],dic[
'age'],dic[
'phone'],dic[
'dept'],dic[
'enroll_date'])
f.write(line)
os.remove(path) # 将旧文件删除，并将新建的文件覆盖旧文件，实现文件修改
os.rename(
'copy_2.txt', path)
def remove_symbol(smb):
if '\'' in smb:
smb1 = smb.strip(
'\'')
return smb1
elif '\"' in smb:
smb1 = smb.strip(
'\"')
return smb1
else:
return smb
@deco_delete_add#main=deco_delete_add(main)
def main(cmd):
'''实现模糊查询功能'''
sl_where(cmd)
def sl_from(g1):
l_cmd = cmd.strip().split(
'where')
cmd_from = l_cmd[0].strip().split(
'from')[0]
cmd_from=cmd_from.strip().split(
'select')[1].strip()
# '*' or 'name,age'
for i
in g1:
for key
in i:
if key
in cmd_from.split(
',')
or cmd_from ==
'*':
print(i[key], end=
' ')
print()
g1=
sl_where(cmd)
sl_from(g1)
print(
'查询到的记录为%d条' %
count)
staff_table=
make_dic()
main(command)
1,Alex lin,30,13651054608,market,2013-04-01
2,Jack Wang,22,13304320533,HR,2015-05-03
3,Rain Liu,25,1383235322,Sales,2016-04-22
4,Mack Cao,40,1356145343,HR,2009-03-01
5,Yu Yang,25,13811223344,Student,2017-06-16
6,Liu Yang,25,13811223345,Student,2017-06-20
7,Yang Yang,25,13811223346,Student,2017-06-20
8,Yang Yang,25,13811223347,Student,2017-06-08
9,Yang Gao,25,13811223348,Student,2017-06-08
10,Yang Gao,25,13811223350,Student,2017-06-08
11,Yang Gao,25,13811223351,Student,2017-06-08
12,Yang Gao,25,13811223352,Student,2017-06-08
转载于:https://www.cnblogs.com/yuyang26/p/7041867.html
相关资源：实现对员工信息的增删改查实例（全注解）
