# -*- coding:utf-8 -*-
import os
staff_db = 'staff_table.txt'
staff = 'staff_table'
staff_order = ['id', 'name', 'age', 'phone', 'dept', 'enroll_date']
#返回list_data=common_data
def print_search(list_data):
if len(list_data) != 0:
print('影响了 33[1;31m%s 33[0m条记录'% len(list_data)) #高亮度打印list_data数据长度
for i in list_data: #逐条遍历list_data 并打印
print(i)
else:
print("没有对应得数据！")
def syntax_update(con_sen,common_data):
# update 修改语句
if len(common_data) == 0:
print('没有记录')
return common_data
else:
index_num = 0
id_data = []
new_data = []
for index, i in enumerate(staff_order):
if i == con_sen[0].strip():
index_num = index
break
for index, i in enumerate(common_data):
id_data.append(i.split(',')[0])
f = open(staff_db, 'r+', encoding='utf-8')
f_new = open('staff_table_new.txt', 'w', encoding='utf-8')
for line in f:
temp_li =line.split(',')
if temp_li[0] in id_data:
if con_sen[1].strip().isdigit():
temp_li[index_num] = con_sen[1].strip()
else:
temp_li[index_num] = eval(con_sen[1].strip())
line = ','.join(temp_li)
new_data.append(line.strip())
f_new.write(line)
f.close()
f_new.close()
os.replace(f_new.name, f.name)
return new_data
def syntax_add(add_data):
# add 语句增加
f = open(staff_db, 'r+', encoding='utf-8')
id_data, phone, new_list = [], [], []
for index, line in enumerate(f):
id_data.append(line.split(',')[0])
phone.append(line.split(',')[3])
if add_data[2] in phone:
print('phone已存在')
else:
data = str(int(id_data[-1])+1) + ','+ ','.join(add_data)
new_list.append(data)
f.write(data+'
')
f.close()
return new_list
def syntax_delete(common_data):
# del 语句删除
if len(common_data) == 0:
print('没有该条记录'.center(20,'-'))
return common_data
else:
str_data = ''
f = open(staff_db, 'r+', encoding='utf-8')
for line in f:
if line.split(',')[0] == common_data[0].split(',')[0]:
continue
else:
str_data += line
f.seek(0)
f.truncate()
f.write(str_data)
f.close()
print('已从文件中删除'.center(20, '-'))
return common_data
def syntax_find(con_sen,common_data): #common_data(存储匹配数据) con_sen 例如find * 前两位数据
# find 语句查找
if '*' in con_sen.split('find')[1]: #*号在con_sen
return common_data #返回common_data
sen = con_sen.split('find')[1].split(',')
li = []
for index, i in enumerate(staff_order):
for j in sen:
if j.strip() == i:
li.append(index)
new_data = []
for i in common_data:
str_data = ''
for index, j in enumerate(i.split(',')):
for k in li:
if k == index:
str_data += j + ','
break
new_data.append(str_data[:-1])
return new_data
#common_data(存储匹配数据) #con_sentence =find name,age from staff_table data =age>22
def syntax_where(common_data, con_sentence):
# 查找语句分配
if con_sentence == '': #con_sentence =空
add_data = common_data.split(',') #赋值common_data.split
return syntax_add(add_data) #执行syntax_add(add_data)函数
else:
if 'from' in con_sentence: #如果里面有from 语句
con_sen = con_sentence.split('from') #con_senterce.split用from 分
if con_sen[1].strip() == staff: #如果con_sen[1]字符串里有‘staff’
if 'find' in con_sen[0]: #如果‘find’在con_sen[0]
return syntax_find(con_sen[0],common_data) #执行syntax_find()
elif 'del' in con_sen[0]: #如果del 在common_data
return syntax_delete(common_data)
else:
print('输入的语句有问题')
return []
elif 'set' in con_sentence and 'update' in con_sentence:
con_sen = con_sentence.split('set')
if 'staff_table' in con_sen[0]:
return syntax_update(con_sen[1].split('='), common_data)
else:
print('输入的语句有问题')
return []
else:
print('输入的语句有问题')
return []
#common_op == con_where = [] # 存储区配的数据结果
def common_op(where_data): #common_op(data)
# > < = like 公共方法
list_where_data = [] #赋值从命令行匹配的符号
con_where = [] #存储区配的数据结果
index_num = 0
op = ['>', '<', '=', 'like']
for index, i in enumerate(op): #遍历OP 符号。
if i in where_data: #如果符号 和 where_data匹配上
list_where_data = where_data.split(i) #用符号分，赋值给list_where_data
index_num = index #匹配上后赋值索引值
break #跳出循环
f = open(staff_db, 'r', encoding='utf-8')
for index, i in enumerate(staff_order):
if i == list_where_data[0].strip(): #如果匹配上 age 赋值 I
for line in f: #遍历数据库
data = line.strip().split(',') # 转成列表，赋值data
if index_num == 0: #如果索引为0 对应的大于，
if eval(data[index].strip()) > eval(list_where_data[1].strip()):#
# print("data,listwhere_data",data,list_where_data)
con_where.append(line.strip()) #列表data大于list_where_data,就添加list到con_where
elif index_num == 1: #index_num=1 对应的小于
if eval(data[index].strip()) < eval(list_where_data[1].strip()):
# print("data:,listwhere_data:", eval(data[index].strip()) , eval(list_where_data[1].strip()))
# print('data1,data2,data3,listwhere_data1,listwhere_data2',data[1],data[2],data[3],list_where_data[0],list_where_data[1])
con_where.append(line.strip()) # 添加年龄小于的值到con_where
elif index_num == 2:
if data[index].strip().isdigit(): # 判断原有的内容是否是整数类型
if eval(data[index].strip()) == eval(list_where_data[1].strip()):
con_where.append(line.strip())
else:
if data[index].strip().lower() == eval(list_where_data[1].strip().lower()):
con_where.append(line.strip())
elif index_num == 3:
if data[index].strip().split('-')[0].strip() == eval(list_where_data[1].strip()):
con_where.append(line.strip())
break
f.close()
return con_where
def syntax_parser(cmd):
if 'add' in cmd: #如果add存在
print_search(syntax_where(cmd.split(staff)[1], '')) #执行print_search(syntax_where(cmd.split(staff)[1],))
elif 'where' in cmd: #如果有where
con_sentence, data = cmd.split('where') #con_sentence =find name,age from staff_table data =age>22
print_search(syntax_where(common_op(data), con_sentence))#执行print_search(sumtax_where(common_op(data),find name,age from staff_table)
def main():
while True:
cmd = input('输入语句>>>:').strip().lower()
if not cmd:
continue
syntax_parser(cmd)
if __name__ == '__main__':
main()