实现SQL基本语法

#!/usr/bin/env python 2 # writen by Lisp at 2017/5/15 3 4 """ 5 对员工信息文件，实现增删改查操作 6 7 1. 可进行模糊查询，语法至少支持下面3种: 8 9 a. select user_name,user_age from db.emp where user_age > 22 10 b. select * from db.emp where user_dept = "IT" 11 c. select * from db.emp where user_date like "2013" 12 d. select * from db.emp limit 3 13 e. select * from db.emp where user_id > 12 and user_id < 15 or user_name like 李 14 f. select * from db.emp where not user_id > 3 15 16 查到的信息，打印后，最后面还要显示查到的条数 17 18 2. 可创建新员工纪录，以phone做唯一键，staff_id需自增 19 20 a. insert into db.emp values 张光棍,31,13111111111,CEO,2011-11-11 21 22 3. 可删除指定员工信息纪录，输入员工id，即可删除 23 24 a. delete from db.emp where id >25 25 26 4. 可修改员工信息，语法如下: 27 28 a. UPDATE db.emp SET dept="Market" WHERE user_dept = "IT" 29 30 注意：以上需求，要充分使用函数，请尽你的最大限度来减少重复代码！ 31 32 33 思路: 34 1. 先从文件读出所有数据到列表,每行数据按字典的格式存储 35 2. 更新,插入,删除操作均操作列表,操作完以后,执行文件写入操作 36 37 """ 38 # 记录 input 输入的历史命令, 可以使用上下左右箭头; 不同环境可能不一样. 39 import readline 40 import os 41 42 43 def get_all_info(file_name): 44 """ 45 通过db file 得到所有用户信息; 此处无需判断异常, 因为在调用此函数之前, 均已做好判断 46 :param file_name: str db_filename 47 :return: list 返回所有用户信息的列表, 用户信息以dict形式保存 48 """ 49 user_info_list = [] 50 with open(file_name, ‘r‘, encoding=‘utf-8‘) as r_f: 51 for each_line in r_f.readlines(): 52 each_list = each_line.strip().split(‘,‘) 53 user_info_list.append(dict(zip(tk_temp, each_list))) 54 return user_info_list 55 56 57 def write_to_file(db_name, write_list): 58 """ 59 将内容写入文件; 此处同样无需判断异常 60 :param db_name get_filename()函数可得到 61 :param write_list 要往文件中写入的数据 [{},{},{}...] 形式 62 :return: 无返回,只写文件 63 """ 64 # 将字段名字组成字符串 先列表生成,再join 65 temp_str = ‘,‘.join([‘{‘ + x + ‘}‘ for x in tk_temp]) 66 with open(db_name, ‘w+‘, encoding=‘utf-8‘) as w_f: 67 for each_dic in write_list: # 遍历要写的数据列表 68 if write_list[0] == each_dic: # 判断首行 69 temp_data = ‘{0}‘.format(temp_str).format_map(each_dic) 70 else: 71 temp_data = ‘\n{0}‘.format(temp_str).format_map(each_dic) 72 w_f.writelines(temp_data) 73 74 75 def get_where_clause(para_str, para_item): 76 """ 77 处理where后面的字符串 78 id -> 每行记录的ID, user_name -> 每行记录的user_name ... 79 and 和 or 和 not 可以不做处理 80 like -> like前后值翻转, like变为in 81 limit 最后处理 82 :param para_item: dict 用户详细信息 -- 文件中的每行记录 83 :param para_str: content -- where后的字符串 84 :return: union 85 """ 86 # 遍历字段名,把where后面的字段名全部替换为字段值 87 # temp_str = para_str 88 for ITEM in tk_temp: 89 # 如: where user_id > 10 --> where 1 > 10 90 para_str = para_str.replace(ITEM, para_item[ITEM].lower()) 91 92 if ‘"‘ in para_str: 93 para_str = para_str.replace(‘"‘, ‘‘) 94 if "‘" in para_str: 95 para_str = para_str.replace("‘", ‘‘) 96 # 处理like关键字 97 temp_list = para_str.split() # 定义临时列表, 用空格分开替换后的where子句 98 if ‘like‘ in para_str: # 判断是否有like语句 99 if ‘ like ‘ in para_str: # 如果有like, 则判断是否被空格分开;因为上面的列表是按照空格分割的list 100 for temp in temp_list: # 遍历temp_list 101 if temp == ‘like‘: # 如果有like的列表元素 102 like_flag = temp_list.index(temp) # 找出like的下标 103 temp_list[like_flag] = ‘in‘ # 把like替换为in 104 if temp_list[like_flag - 1] != ‘not‘: # 找出like的下标后,判断like的前面是否有not关键字 105 # 没有,则前后两个元素互换; 操作空间是temp_list 106 temp_list[like_flag - 1], temp_list[like_flag + 1] = "‘" + str( 107 temp_list[like_flag + 1]) + "‘", "‘" + str(temp_list[like_flag - 1]) + "‘" 108 else: 109 # 有,则not前一个元素与like后一个元素互换 110 temp_list[like_flag - 2], temp_list[like_flag + 1] = "‘" + str( 111 temp_list[like_flag + 1]) + "‘", "‘" + str(temp_list[like_flag - 2]) + "‘" 112 else: 113 return 901, ‘like关键字附近‘ # 返回语法错误和提示 114 para_str = ‘ ‘.join(temp_list) # 再用空格将处理过like的列表组成字符串 115 # 处理 类似 user_dept=it 有一个等号的情况; 需要把一个等号变为两个等号 user_dept==it 才可以被eval 116 for temp in temp_list: # 遍历临时列表 117 if ‘=‘ in temp and ‘==‘ not in temp: # 如果列表元素中有=, 而且还没被替换; 匹配[‘user_id=2‘, ‘=‘, ‘user_dept‘, ‘=it‘] 118 if ‘<=‘ not in temp and ‘>=‘ not in temp: # 如果是 >= , <= 则不处理 119 if temp != ‘=‘: # 匹配user_id=2, user_id=, =2的情况 120 temp_para = temp.split(‘=‘) # 则,将item用=分割, 获取=两边的内容 121 if temp.startswith(‘=‘): # =2 122 bf_temp_index = temp_list.index(temp) - 1 123 bf_temp = temp_list[bf_temp_index] 124 para_str = para_str.replace(temp, ‘=="‘ + temp_para[-1] + ‘"‘) 125 para_str = para_str.replace(bf_temp + ‘ ‘, ‘"{}"‘.format(bf_temp)) 126 elif temp.endswith(‘=‘): # user_id= 127 bh_temp_index = temp_list.index(temp) + 1 128 bh_temp = temp_list[bh_temp_index] 129 para_str = para_str.replace(temp, ‘\"‘ + temp_para[0] + ‘\"==‘) 130 para_str = para_str.replace(bh_temp + ‘ ‘, ‘"{}"‘.format(bh_temp)) 131 else: # user_id=2 132 # 替换原字符串中的temp, 替换为 ‘user_dept‘==‘IT‘ 133 para_str = para_str.replace(temp, "‘" + temp_para[0] + "‘==‘" + temp_para[1] + "‘") 134 else: # 匹配‘=‘的情况,则说明=两边是空格, 那直接替换原字符串 = 为 == 135 temp_index = temp_list.index(temp) 136 old_str = temp_list[temp_index - 1] + ‘ = ‘ + temp_list[temp_index + 1] 137 new_str = ‘"{}" == "{}"‘.format(temp_list[temp_index - 1], temp_list[temp_index + 1]) 138 para_str = para_str.replace(old_str, new_str) 139 return para_str 140 141 142 def get_ok_dic_id(para_str, user_info_list): 143 """ 144 遍历整个用户信息表, 返回符合where后面条件的dic在user_info_list中下标的列表, 使用与delete，select，update 145 :param user_info_list: 用户信息列表 146 :param para_str 传给get_where_clause函数, 返回能被eval的字符串 147 :return: union 148 """ 149 res_list = [] 150 if not para_str: 151 return list(range(len(user_info_list))) 152 for dic_item in user_info_list: # 遍历user_info_list, 153 # 调用函数,处理where子句,返回可以eval的字符串 154 eval_str = get_where_clause(para_str, dic_item) 155 if eval_str[0] == 901: 156 return eval_str 157 try: 158 if eval(eval_str): # 判断eval的字符串是否成立; 成立则追加成立的字典下标到返回列表 159 res_list.append(user_info_list.index(dic_item)) 160 except (NameError, SyntaxError): 161 return 901, ‘Where子句‘ 162 return res_list 163 164 165 def get_filename(para_str): 166 """ 167 这个函数用来解决 sql 语句当中 db.emp 对应文件的问题 168 :param para_str: 小写字符串 db.emp 169 :return: union 170 """ 171 # 获取当前文件的绝对路径(不包含文件名) 172 real_path = os.path.dirname(os.path.realpath(__file__)) 173 # 替换参数中字符串的.为/ 174 db_path = para_str.replace(‘.‘, ‘/‘) 175 # 字符串拼接，组成数据库文件的绝对路径 176 file_path = real_path + ‘/‘ + db_path 177 # 判断文件是否存在 178 if os.path.isfile(file_path): 179 return 950, db_path # 返回数据文件的相对路径 180 else: 181 return 951, ‘{}不存在‘.format(db_path) # 数据文件不存在 182 183 184 def get_res_list(content, user_info_list, cl_name=‘‘): 185 """ 186 select函数专用; 简化 select * from ... 和 select column1, column2 from ... 的情况 187 :param cl_name: 查询字段,默认为空代表* 188 :param user_info_list: 用户信息列表 189 :param content: where子句 190 :return: res_list 191 """ 192 t_flag = False 193 res_list = [] 194 cont_list = [] 195 # 获取字符串形式的字段名{user_name},{user_age}... 196 # 因查询的字段，显示不同 197 if not cl_name: 198 temp_str = ‘,‘.join([‘{‘ + x + ‘}‘ for x in tk_temp]) 199 else: 200 new_cl_list = cl_name.split(‘,‘) 201 for each_cl in new_cl_list: 202 if each_cl not in tk_temp: 203 return 921, ‘字段名 {} 不存在‘.format(each_cl) 204 temp_str = ‘,‘.join([‘{‘ + x + ‘}‘ for x in new_cl_list]) 205 206 if content == ‘‘: 207 for item in user_info_list: 208 temp = ‘[{0}]‘.format(temp_str).format_map(item) 209 res_list.append(temp) 210 # select * from db.emp [ where... | limit... ] 211 else: 212 # 如果有limit,把limit后面那个值取出来,content变为limit前的str;更改limit标志位 213 if ‘limit‘ in content: 214 cont_list = content.split(‘limit‘) 215 if cont_list[0]: 216 content = cont_list[0] 217 else: 218 content = ‘‘ 219 t_flag = True 220 # 根据content 取出匹配到的dic的下标 221 se_id = get_ok_dic_id(content, user_info_list) 222 if se_id: 223 if se_id[0] == 901: 224 return se_id 225 else: 226 # 遍历下标,将内容先追加到一个列表 227 for each_id in se_id: 228 temp = ‘[{0}]‘.format(temp_str).format_map(dict(user_info_list[each_id])) 229 res_list.append(temp) 230 else: 231 res_list = se_id 232 233 if t_flag: 234 return res_list[:int(cont_list[-1])] 235 else: 236 return res_list 237 238 239 def my_sql_select(para_str): 240 """ 241 查询 242 :param: para_str: 接收用户输入的字符串 小写 243 :return: 244 """ 245 para_list = para_str.split() 246 247 try: 248 # 找到from关键字的下标 249 from_id = para_list.index(ky_list[5]) 250 except ValueError: 251 # 无关键字 252 return 901, ‘关键字{}缺失‘.format(ky_list[5]) 253 254 # select 到 from 之间 判断长度 255 if len(para_list[1:from_id]) >= 1: 256 cl_name = ‘‘.join(para_list[1:from_id]) 257 else: 258 return 901, ‘{}与{}之间‘.format(ky_list[0], ky_list[5]) 259 260 if len(para_list[from_id + 1:]) >= 2: 261 try: 262 # 查找where的下标 263 where_id = para_list.index(ky_list[6]) 264 if len(para_list[from_id + 1:where_id]) != 1: 265 return 901, ‘数据文件部分有语法错误‘ 266 if len(para_list[where_id + 1:]) > 0: 267 content = para_str.split(ky_list[6])[-1].strip() 268 else: 269 return 901, ‘关键字{}后缺少条件‘.format(ky_list[6]) 270 271 except ValueError: 272 # 没有where 273 try: 274 limit_id = para_list.index(ky_list[-2]) 275 if len(para_list[from_id + 1:limit_id]) != 1: 276 return 901, ‘关键字{}位置不对‘.format(ky_list[-2]) 277 if len(para_list[limit_id:]) == 2: 278 content = ‘ ‘.join(para_list[limit_id:]) 279 else: 280 return 901, ‘{} 后参数异常‘.format(ky_list[-2]) 281 except ValueError: 282 return 901, ‘缺少 {} 或者 {} 关键字‘.format(ky_list[-3], ky_list[-2]) 283 elif len(para_list[from_id + 1:]) == 1: 284 content = ‘‘ 285 else: 286 return 901, ‘缺少数据文件‘ 287 288 db_name = para_list[from_id + 1] 289 # 分割关键字之间的str结束, 得到: 290 # select-->from = cl_name 291 # from-->where = db_name 292 # where--> = content(maybe include limit) 293 294 # 通过get_filename函数将db.emp类型转文件路径db/emp 295 file_tup = get_filename(db_name) 296 file_name = file_tup[1] 297 # 将文件内容读取出来，保存到user_info_list列表 298 if file_tup[0] == 951: 299 return file_tup 300 user_info_list = get_all_info(file_name) 301 302 # select * from ... 303 if cl_name == ‘*‘: 304 # select * from db.emp + content 305 # 这里这种情况直接调用函数, 因为就cl_name不一样,其他都基本一致. 306 # 把cl_name传进去,进行匹配判断. 307 res_list = get_res_list(content, user_info_list) 308 if res_list: 309 if res_list[0] == 901 or res_list[0] == 921: 310 return res_list 311 else: 312 # select user_id ,name , xx from db.emp + content 313 res_list = get_res_list(content, user_info_list, cl_name) 314 if res_list: 315 if res_list[0] == 901 or res_list[0] == 921: 316 return res_list 317 318 for each_res in res_list: 319 print(each_res) 320 return 920, ‘总共{}条记录‘.format(len(res_list)) 321 322 323 def my_sql_insert(para_str): 324 """ 325 往数据文件插入数据. 326 1. 可以直接将数据插入到文件； 327 2. 先append到列表，然后再将列表覆盖写入文件的方式； 328 这里为了练习函数的使用，采用方法2。 329 :param para_str: 小写字符串 insert into [db_file] values xxx 330 :return: 331 """ 332 # 将接收的参数通过空格来分割，转为列表 333 para_list = para_str.split() 334 335 try: 336 values_id = para_list.index(ky_list[4]) # 找到values关键字的位置；ky_list[4]=values 337 except ValueError: 338 return 901, ‘关键字{}缺失‘.format(ky_list[4]) # 无values关键字 339 340 if len(para_list[2:values_id]) != 1: # 判断into之后到values关键字之间是否只有一个元素 341 return 901, ‘数据文件部分有语法错误‘ # 否，则返回选择的数据库和表有误 342 else: 343 db_name = para_list[2] # 是，数据文件暂定 344 345 if len(para_list[values_id + 1:]) < 1: # 判断values关键字之后是否再有无空格 346 return 901, ‘未发现要插入的数据‘ # 如果有，插入的数据格式有误 347 else: 348 # 如果有values，则以values分割，取后面的字符串，以,分割得到一个列表 349 temp_list = para_str.split(ky_list[4])[1].split(‘,‘) 350 # 匹配列表长度，与用户信息dict长度是否一致 351 if len(temp_list) == 5: 352 for temp_item in temp_list: 353 # 去掉空格 354 temp_index = temp_list.index(temp_item) 355 temp_list[temp_index] = temp_item.strip() 356 else: 357 return 921, ‘要插入的数据 格式有误‘ 358 content = ‘,‘.join(temp_list) # 没有，则确定插入的数据 359 360 # 通过上述判断，可得出要插入的内容content和要插入的文件名db_name 361 # 通过get_filename函数将db.emp类型转文件路径db/emp 362 file_tup = get_filename(db_name) 363 file_name = file_tup[1] 364 # 将文件内容读取出来，保存到user_info_list列表 365 if file_tup[0] == 951: 366 return file_tup 367 user_info_list = get_all_info(file_name) 368 369 # 遍历user_info_list，判断手机号是否唯一 370 for item in user_info_list: 371 if content.split(‘,‘)[2] in item.values(): 372 return 911, ‘手机号{}已经存在‘.format(content.split(‘,‘)[2]) # 返回手机号已经存在 373 374 # 上述判断通过，则这里表明输入的参数中，数据文件以及要插入的数据都是合法的 375 # 得到主键ID 376 insert_id = int(user_info_list[-1][tk_temp[0]]) + 1 377 378 # 得到要插入的完整内容 id,name,age,phone,dept,date 379 insert_str = str(insert_id) + ‘,‘ + content 380 381 # 先将得到的内容通过,分割得到列表；再与字段名的列表tk_temp组成dict，追加到user_info_list 382 user_info_list.append(dict(zip(tk_temp, insert_str.split(‘,‘)))) 383 384 # 将user_info_list写入文件 385 write_to_file(file_name, user_info_list) 386 # 打印插入成功提示 387 print(‘{2} func: insert(), db_file: {0}, values: {1} {3}‘.format(file_name, content, ‘{‘, ‘}‘)) 388 return 910, ‘当前数据库中最大主键 ID = {0}‘.format(str(insert_id)) # 返回插入成功 389 390 391 def my_sql_delete(para_str): 392 """ 393 删除 394 delete from db.emp where id >25 395 delete * from db.emp 396 删除成功提示 397 :param: para_str: str 398 :return: 399 """ 400 # 清空文件标识 401 tc_flag = False 402 403 # 将接收的参数通过空格来分割，转为列表 404 para_list = para_str.split() 405