python-作业：员工信息表

标签：itemsdisplayispexcehttplast帮助tafexit

程序可实现以下功能：1、查询，输入select name,age from staff_table where age > 22，查询到符合要求的信息； 输入select * from staff_table where dept = \”IT\”，查询到符合要求的信息； 输入select * from staff_table where enroll_date like \”2013\”，查询到符合要求的信息。2、创建，输入insert Mickle,22,13651054608,IT,2013-04-01，加信息加入的员工信息表中。3、删除，输入delect id，将指定的id号员工信息删除。4、修改，输入update staff_table set dept = \”Market\” where dept = \”IT\”，将要修改的内容修改。

程序还提供help功能，为用户输入正确的sql语句格式提供帮助。

import os #导入os模块，方便文件的删除和改名

def info_display(): #打印程序启动信息
print(\”-\”.center(60, \”-\”))
print(\”欢迎来到员工管理系统\”.center(50,\” \”))
print(\”【输入help显示帮助信息】\”.center(50, \” \”))
print(\”-\”.center(60, \”-\”))
sql = input(\”请输入\\033[1;31msql\\033[0m信息>>>>\”).strip(\”\”)
if sql == \”q\” or sql == \”quit\”: #用户如果输入q，程序退出
exit(\” Bye Bye \”.center(60, \”-\”))
elif sql == \”help\”: #用户输入help，打印help信息，help信息为打印的格式
print(\”\\033[1;35m 查询 输入格式：\\n\\tselect name,age from staff_table where age > 22\\033[0m\”)
print(\”\\033[1;35m 创建 输入格式：\\n\\tinsert Mickle,22,13651054608,IT,2013-04-01\\033[0m\”)
print(\”\\033[1;35m 修改 输入格式：\\n\\tupdate staff_table set dept = \\\”Market\\\” where dept = \\\”IT\\\”\\033[0m\”)
print(\”\\033[1;35m 删除 输入格式：\\n\\tdelete 5\\033[0m\”)
else:
sql_parse(sql) #用户输入正确的sql语法，调用sql解析函数

def sql_parse(sql): #对用户输入的sql语句进行解析，根据sql语句调用不同的函数
func_choice = {\”insert\”: insert, \”delete\”: delete, \”update\”: update, \”select\”: select}
sql_list = sql.replace(\”,\”, \” \”).split() # 将sql解析成列表形式
if sql_list[0] == \”select\”: #如果用户输入的sql语句是查询
sql_dict = {\”from\”: [], \”where\”: [], \”limit\”: []}
tag = False
for item in sql_list: #根据sql解析出sql_dict
if tag and item in sql_dict:
tag = False
if not tag and item in sql_dict:
tag = True
key = item
continue
if tag:
sql_dict[key].append(item.strip(‘\\\”‘))
func_choice.get(sql_list[0])(sql_dict) #解析出sql_dict后，传入查询函数，并调用查询函数
elif sql_list[0] == \”insert\”: #如果用户输入的sql语句是创建
sql_dict = {}
sql_list.remove(\”insert\”)
sql_dict.setdefault(\”values\”,sql_list) #解析出sql_dict，传入创建函数
insert(sql_dict) #调用创建函数
elif sql_list[0] == \”delete\”: #解析出sql_dict后，传入删除函数，并调用查询函数
delete_id = int(sql_list[1]) #获取用户要删除的id
delete(delete_id) #调用删除函数
elif sql_list[0] == \”update\”: #如果用户输入的sql语句是修改
sql_dict = {\”set\”:[],\”where\”:[],\”update\”:[]}
tag = False
for items in sql_list: #根据用户的sql，解析出sql_dict字典
if tag and items in sql_dict:
tag = False
if not tag and items in sql_dict:
tag = True
key = items
continue
if tag:
sql_dict[key].append(items.strip(‘\\\”‘))
update(sql_dict) #传入修改函数，并调用修改函数
return sql_list

def insert(sql_dict): #创建信息函数
with open(\”staff_table\”,\”ab+\”) as f:
offs = -100
while True:
f.seek(offs,2)
lines = f.readlines()
if len(lines) > 1:
last = lines[-1]
break
offs *= 2
last = last.decode(encoding=\”utf-8\”)
last_id = int(last.split(\”,\”)[0])
new_id = last_id + 1
record = sql_dict.get(\”values\”)
record.insert(0,str(new_id))
record_str = \”,\”.join(record)+\”\\n\”
f.write(bytes(record_str,encoding=\”utf-8\”))
f.flush()
print(\”\\033[1;31m创建成功 \\033[0m\”)

def delete(delete_id): #删除函数
with open(\”staff_table\”,\”r\”,encoding=\”utf-8\”) as f1,open(\”staff_table_bak\”,\”w\”,encoding=‘utf-8‘) as f2:
del_count = 1
for line in f1:
if del_count != delete_id:
f2.write(line)
elif del_count == delete_id:
pass
del_count += 1
print(\”\\033[1;31m删除成功 \\033[0m\”)
os.remove(\”staff_table\”) #删除原员工信息文件
os.rename(\”staff_table_bak\”,\”staff_table\”) #修改新员工信息文件名

def update(sql_dict): #修改函数
set_list = sql_dict.get(\”set\”) #获取要修改的内容
set_key = set_list[0]
where_list = sql_dict.get(\”where\”) #获取要修改的文件内容
with open(\”staff_table\”, \”r\”, encoding=\”utf-8\”) as f1, open(\”staff_table_bak\”, \”w\”, encoding=‘utf-8‘) as f2:
for line in f1:
title = \”id,name,age,phone,dept,enroll_date\”
dic = dict(zip(title.split(\”,\”),line.split(\”,\”))) #将文件每一行打包成字典的形式
if logic_action(dic,where_list): # 逻辑判断，调用逻辑判断函数
line = line.replace(dic[set_key],set_list[2]) #修改需要修改的那一行内容
f2.write(line) #将原文件写进新文件
print(\”\\033[1;31m修改成功 \\033[0m\”)
os.remove(\”staff_table\”)
os.rename(\”staff_table_bak\”, \”staff_table\”)

def select(sql_dict): #查询函数
title = \”id,name,age,phone,dept,enroll_date\”
f = open(\”staff_table\”, \”r\”, encoding=\”utf-8\”) # 1、找到数据库
res = []
for line in f:
dict1 = dict(zip(title.split(\”,\”), line.strip().split(\”,\”)))
where_list = sql_dict.get(\”where\”)
if len(where_list) != 0:
if logic_action(dict1, where_list): # 逻辑判断，调用逻辑判断函数
res.append(line.strip())
else:
res = f.readlines()
print(\”查询到的信息有 \\033[1;35m %s \\033[0m 条\”%len(res))
for i in res:
print(\”分别是\\033[1;35m [%s]\\033[0m \”%i)
return res

def logic_action(dict1, where_list): #逻辑判断函数，如果文件的信息符合要求，就返回True
tag = False
if where_list[1] == \”<\”:
if dict1[where_list[0]] < where_list[2]:
tag = True
elif where_list[1] == \”=\”:
if dict1[where_list[0]] == where_list[2]:
tag = True
else:pass
elif where_list[1] == \”>\”:
if dict1[where_list[0]] > where_list[2]:
tag = True
else:pass
elif where_list[1] == \”like\”:
if where_list[2] in dict1[where_list[0]]:
tag = True
return tag

while True:
try:
info_display()
except IndexError as e:
print(\”【%s】，请输入正确的格式\”%e) #如果用户输入的格式不正确，就打印提示