#!/usr/bin/env python

# -*- coding: utf-8 -*-

import re

from db import db_handle

​

​

# 查询接口

def select_interface(user_inp):

# 调用查询语法解析接口

flag = select_prammar(user_inp) # 获得关键字，文件名，查询条件

flag_list = ['1', '2', '3', '4', '5']

if flag in flag_list:

return flag

keyword, file_name, condition = flag

# 开始查询

user_list = [] # 存储每一个用户字典的列表

show_list = []

user_data = db_handle.select(file_name)

if user_data == '6':

return '6'

location_1, location_2, location_3 = condition

eval_list = ['int(user_dic[location_1]) > int(location_3)',

'int(user_dic[location_1]) < int(location_3)']

if location_2 in ['']:

if not location_3.isdigit() or location_1 not in ['age', 'id', 'phone']:

return '2'

for user_dic in user_data:

if location_2 == '>':

if eval(eval_list[0]):

user_list.append(user_dic)

else:

if eval(eval_list[1]):

user_list.append(user_dic)

elif location_2 == '=':

for user_dic in user_data:

if location_3 == user_dic[location_1]:

user_list.append(user_dic)

elif location_2 == 'like':

# print('zheli')

for user_dic in user_data:

if location_3 in user_dic[location_1]:

user_list.append(user_dic)

# print(keyword)

if keyword[0] == '*':

# print(user_list, 'userlist')

show_list = [list(i.values()) for i in user_list]

title_table = [['id', 'name', 'age', 'phone', 'dep', 'start_work_time']]

show_list = title_table + show_list

# print(show_list)

return show_list

else:

# print(user_list, 'userlist')

for k in user_list:

show_list.append([k[i] for i in keyword])

show_list = [keyword] + show_list

return show_list

​

​

# 增加接口

def add_interface(user_inp):

# 调用增加语句解析接口

res = add_prammar(user_inp)

if res == '1':

return '1'

*_, file_name, name, age, phone, dep, start_work_time = res

# 调用查询接口

user_list = db_handle.select(file_name)

if user_list == '6':

return '6'

# 判断手机号码是否重复

for user_dic in user_list:

if phone in user_dic['phone']:

return '7'

current_id = int(user_list[-1]['id']) + 1

current_id = str(current_id)

add_user_info = [f'{current_id},{name},{age},{phone},{dep},{start_work_time}']

# 调用保存接口

user_data = [list(i.values()) for i in user_list]

user_data.append(add_user_info)

db_handle.save(file_name, user_data)

return '8'

​

​

# 删除接口

def del_interface(user_inp):

# 调用删除语句解析接口

user_data = []

res = del_prammar(user_inp)

if res == '1':

return '1'

file_name, user_id = res

# 获取数据

user_list = db_handle.select(file_name)

if not user_list:

return '6'

# 删除符合条件的信息

for user_dic in user_list:

if not user_dic['id'] == user_id:

user_data.append(list(user_dic.values()))

# 调用保存接口

db_handle.save(file_name, user_data)

return '9'

pass

​

​

# 更新接口

def update_interface(user_inp):

# 调用修改语法解析接口

res = update_prammar(user_inp)

if res == '1':

return '1'

file_name, set_name, set_data, where_name, where_data = res

# 获取数据信息

user_list = db_handle.select(file_name)

if not user_list:

return '6'

# 修改数据

for user_dic in user_list:

if user_dic[where_name] == where_data:

user_dic[set_name] = set_data

user_data = [list(i.values()) for i in user_list]

# 调用保存接口

db_handle.save(file_name, user_data)

return '10'

​

​

'''-----------------------------------------------------------------'''

​

​

# 语句类型判断

def prammar_type(user_inp):

'''解析用户输入语句的类型:param user_inp::return: 对应语句类型接口的内存地址'''

prammar_list = user_inp.split()

current_list = {'find': select_interface,

'add': add_interface,

'update': update_interface,

'del': del_interface}

# print('a')

if prammar_list[0] in current_list:

return current_list[prammar_list[0]]

else:

return False

​

​

# 查询语法解析

def select_prammar(user_inp):

title = ['id', 'name', 'age', 'phone', 'dep', 'start_work_time', '*']

conditon = ['id', 'name', 'age', 'phone', 'dep', 'start_work_time']

# 正则表达式

res = re.search('find (.*?) from (.*?) where (.*)', user_inp, re.I)

# print('1')

if not res: return '1'

# 判断查询条件是否正确

res2 = re.search('(.*?) ([>=

# print('2')

# print(res2)

if not res2: return '2'

# 判断展示条件是否正确

res3 = re.findall('[a-zA-Z*]+', res.group(1))

# print('3')

if not res3: return '3'

for i in res3:

if i in title:

pass

else:

return '4'

# print('4')

# print(res2.group(1))

if res2.group(1) in conditon:

pass

else:

return '5'

# print('5')

return [res3, res.group(2), res2.groups()] # 关键字，文件名，查询条件

​

​

# 新增语法解析

def add_prammar(user_inp):

# 正则表达式

res = re.search('add (.*?) ([a-zA-z]+?),(\d+?),(1[34567]\d{9}),(\w+?),(\d{1,4}-\d{1,2}-\d{1,2})', user_inp)

if not res: return '1'

return res.groups()

​

​

# 删除语法解析

def del_prammar(user_inp):

res = re.search('del from (.+?) where id = (.+)', user_inp, re.I)

if not res: return '1'

return res.groups()

​

​

# 修改语法解析

def update_prammar(user_inp):

res = re.search('update (.+?) set ([a-zA-Z]+?) = (.+?) where ([a-zA-Z]+?) = (.+)', user_inp)

if not res: return '1'

if res.group(2) not in ['name', 'age', 'phone', 'dep', 'start_work_time']:

return '1'

if res.group(4) not in ['name', 'age', 'phone', 'dep', 'start_work_time', 'id']:

return '1'

return res.groups()

​

​

# ----------------------------

# 加载数据

def data_load():

res = db_handle.select('staff_table')

return [list(i.values()) for i in res]

​