python第十六天,昨天来晚了,作业终于完成了

作业 1： 员工信息表程序，实现增删改查操作

可进行模糊查询，语法至少支持下面3种:
　　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"
查到的信息，打印后，最后面还要显示查到的条数
可创建新员工纪录，以phone做唯一键，staff_id需自增
可删除指定员工信息纪录，输入员工id，即可删除
可修改员工信息，语法如下:
　　UPDATE staff_table SET dept="Market" WHERE where dept = "IT"

注意：以上需求，要充分使用函数，请尽你的最大限度来减少重复代码

 

实现了统一入口,自动判断增删改查,输出结果:

详细代码:

#！usr/bin/env python
#-*-coding:utf-8-*-
# Author calmyan
import os ,sys
BASE_DIR=os.path.dirname(os.path.abspath(__file__))#获取相对路径转为绝对路径赋于变量
sys.path.append(BASE_DIR)#增加环境变量

select_='select'#定义关键字
from_='from'
where_='where'
update_='update'
set_='set'
WHERE_='WHERE'
UPDATE_='UPDATE'
SET_='SET'
delete_='delete'
insert_='insert'
values_='values'
into_='into'

def sql_parsing(sql):#sql解析函数 主入口
    if sql.startswith('select'):#判断语句类型
        _dict_=select_par(sql)#调用相关函数
    elif sql.startswith('update') or sql.startswith('UPDATE')  :#判断语句类型
        _dict_=update_par(sql)
    elif sql.startswith('insert'):#判断语句类型
        _dict_=insert_par(sql)
    elif sql.startswith('delete'):
        _dict_=delete_par(sql)
    else:
        return print('输入有误，请重新输入!')
    return _dict_#返回解析完成的列表

def select_par(sql):#select语句的解析函数
    list_key=parsing(sql,index_select=-1,index_from=-1,index_where=-1,select_=select_,where_=where_,from_=from_)#查询调用
    return list_key
def update_par(sql):#update语句的解析函数
    list_key=parsing(sql,index_update=-1,index_set=-1,update_=update_,set_=set_,where_=where_)#更新修改调用
    return list_key
def insert_par(sql):#insert语句的解析函数
    list_key=parsing(sql,index_insert=-1,index_into=-1,index_values=-1,insert_=insert_,values_=values_)#添加调用
    return list_key

def delete_par(sql):#delete语句的解析函数
    list_key=parsing(sql,index_delete=-1,index_from=-1,index_where=-1,delete_=delete_,where_=where_,from_=from_)#删除调用
    return list_key

#where子句再格式化
def where_chard(str_list):#where子句再格式化
    key=[]#定义一个临时列表变量
    list_key=['<','>','=']#运算符
    tag=False#定义点为假
    str_=''#字符串拼接变量
    chrd=''#运算符拼接
    for i in str_list:
        if i in list_key :#如果点为假,并且在关键字字典中
            tag=True#点为真
            if len(str_)!=0:#不为空时
                key.append(str_)#列表添加字符串
                str_=''
            chrd+=i#运算符拼接
        if not tag :#点为假
            str_+=i#字符拼接
        if tag and i not in list_key:#如果点为真,并且在不在关键字字典中
            tag=False#转为假
            key.append(chrd)#关键字字典中所对应的添加这个值
            chrd=''
            str_+=i
    key.append(str_)
    if len(key)==1:#判断like运算符
        key=key[0].split('like')#用like分割
        key.insert(1,'like')
        cq='\"'
        if cq in key[2]:#判断是否有包含 ”
            key[2]=cq_c(key[2],cq)
    return key#返回一个列表
#格式化where 子句函数
def where_format(where_list):#格式化where 子句函数
    list_format=[]
    key_=['and','or','not']
    str_=''#字符连接变量
    for i in where_list:
        if len(i)==0:continue#如果为空就不连接
        if i not in key_:##如果不是关键字就进行连接
            str_+=i
        elif i in key_ and len(str_)!=0:
            str_=where_chard(str_)#进行再处理
            list_format.append(str_)#之前的字符串添加到列表
            list_format.append(i)#关键 字添加到列表
            str_=''#清空变量 备用
    str_=where_chard(str_)#进行再处理
    list_format.append(str_)#最后的字符串
    return list_format

#格式化select函数
def select_format(select_list):#格式化select函数
    if select_list[0]=='*':#如果为* 就不改动
        return select_list#直接返回
    else:
        list_=str_conn(select_list).split(',')#字符串连接函数
    return list_
#字符串连接函数
def str_conn(list):
    str_=''
    for i in list:
        str_+=i
    return str_

#语句的解析总函数
def parsing(sql,**kwargs):#语句的解析总函数
    list_l=sql.split(' ')#分割存为列表
    index_from=-1#下标定义from
    index_select=-1#select
    index_where=-1#where
    index_update=-1#update
    index_set=-1#set
    index_delete=-1#delete
    index_insert=-1#insert
    index_values=-1#valuse
    index_into=-1
    list_key={'select':[],'update':[],'delete':[],'insert':[],'set':[],'from':[],'into':[],'where':[],'values':[]}#定义要处理的关键字字典
    for index,i in enumerate(list_l):#获取下标
        if i==select_:#如果是关键字select字典下标
            index_select=index
        if i==update_ or i==UPDATE_:#如果是关键字update字典下标,全部转为小写
            index_update=index
        if i==set_ or i==SET_:
            index_set=index
        if i==insert_:#如果是关键字insert字典下标
            index_insert=index
        if i==into_:
            index_into=index
        if i==values_:
            index_values=index
        if i==delete_:
            index_delete=index
        if i==from_:#如果是关键字from字典下标
            index_from=index
        if i==where_ or i==WHERE_:#如果是关键字where字典下标
            index_where=index

    for index,i in enumerate(list_l):#用下标界定，对进行关键字字典的添加
        if index_select !=-1 and index>index_select and index<index_from:#在select和from中添加到select中,不为空时
            list_key[select_].append(i)#添加当前值
        if index_update !=-1 and index>index_update and index<index_set:#如果是update语句添加一个1值
            list_key[update_].append(i)
        if index_insert !=-1 and index==index_insert:#如果是insert语句添加一个1值
            list_key[insert_].append(1)
        if index_into !=-1 and index>index_into and index<index_values:#如果是into后的语句添加一个1值
            list_key[into_].append(i)
        if index_delete !=-1 and index==index_delete:#如果是delete语句添加一个1值
            list_key[delete_].append(1)
        if index_from!=-1 and  index>index_from and index<index_where:#添加from中的值
            list_key[from_].append(i)
        if index_set!=-1 and index>index_set and index<index_where:#添加set中的值
            list_key[set_].append(i)
        if index_where!=-1 and index>index_where:#添加到where列表中
            list_key[where_].append(i)
        if index_values!=-1 and index>index_values:#添加到values列表中
            list_key[values_].append(i)
    if list_key.get(where_):#如果字典在的列表不为空
        list_key[where_]=where_format(list_key.get(where_))#进行格式化,重新赋于字典,调用格式化函数
    if list_key.get(select_):#如果字典select在的列表不为空
        list_key[select_]=select_format(list_key.get(select_))#进行格式化,重新赋于字典
    if list_key.get(values_):#如果字典values在的列表不为空
        list_key[values_]=select_format(list_key.get(values_))#进行格式化,重新赋于字典
    if list_key.get(set_):#如果字典set在的列表不为空
        list_key[set_]=where_format(list_key.get(set_))#进行格式化,重新赋于字典
    return list_key

#执行部分 主入口
def sql_perform(sql_dict):#执行函数
    if sql_dict[from_]:
        file_name='db\\'+sql_dict[from_][0]#获取文件路径
    elif sql_dict[into_]:#判断是不是在into里面
        file_name='db\\'+sql_dict[into_][0]#获取文件路径
    else:
        file_name='db\\'+sql_dict[update_][0]#获取文件路径
    user_info=open(file_name,'r+',encoding='utf-8')
    key=['id','name','age','phone','dept','enroll_date']
    if sql_dict[select_]:#不为空,就调用select的函数
        perform_select(sql_dict,user_info,key)#传入 字典调用select 函数
        user_info.close()#关闭文件
    if sql_dict[update_]:#不为空,就调用update的函数
        perform_update(sql_dict,user_info,key,file_name)#传入 字典调用update函数
        user_info.close()#关闭文件
    if sql_dict[insert_]:#不为空,就调用insert的函数
        perform_insert(sql_dict,user_info,file_name,key)#传入 字典调用insert函数
    if sql_dict[delete_]:#不为空,就调用insert的函数
        del_tag=True#标示是删除
        perform_update(sql_dict,user_info,key,file_name,del_tag)


#执行select操作的函数
def perform_select(sql_dict,user_info,key):#执行select操作的函数
    list_l=[]#定义要输出的列表
    conut=0#定义计数
    for line in user_info:
        file=line.strip().split(',')#每一行存为列表
        file_dict=dict(zip(key,file))#用zip拉链成字典 file_dict
        if sql_dict[where_]:#where条件不为空
            where_l=sql_dict[where_]#定义where子句变量
            if where_conn(where_l,file_dict):#调用where子句判断函数  如果返回为真
                if sql_dict[select_][0]!='*':#如果查询条件不为 *
                    file.pop(0)#删除ID号
                    file_dict2=dict(zip(sql_dict[select_],file))#进行对比拼接 如只输出 name age
                    list_l.append(file_dict2)
                else:
                    list_l.append(file_dict)#添加到输出的列表
                conut+=1
    for i in list_l:#输出
        print(i)
    print('总共查询结果数为：%s 条'%conut)
    return #返回处理后的信息

#where子句判断函数where函数传入员工字典,where子句
def where_conn(where_l,file_dict):#where子句判断函数where函数传入员工字典,where子句
    reg=[]#临时列表
    for i in where_l:#循环判断where_l中的条件
        if type(i) is list:#判断是否是列表
            i_key,conn,i_val=i#三元赋值 key,运算，内容
            if i[1]=='=':#如果运算符是 = 进行拼接
                conn=i[1]+'='
            if file_dict[i_key].isdigit():#如果员工表内键对应的值是整数
                dict_vla=int(file_dict[i_key])#员工表内的值赋于一个变量
                i_vla=int(i_val)#where条件语句的值赋于一个变量 用于对比
            else:#否则
                dict_vla="'%s'"%file_dict[i_key]#员工表内键对应的值赋于字符串
            if 'like' in i:
                if i_val in dict_vla:#如果where条件
                    i='True'#返回真
                else:
                    i='False'#返回假
            else:
                i=str(eval("%s%s%s"%(dict_vla,conn,i_val)))#进行员工表与条件的比对 再转为字符串
        reg.append(i)#判断后添加到列表
    reg=eval(' '.join(reg))#用 join拼接，最后运算
    return reg

#执行update操作的函数
def perform_update(sql_dict,user_info,key,file_name,tag=False):#执行update操作的函数
    list_l=[]#定义要更新的列表
    conut=0#定义计数
    for line in user_info:
        file=line.strip().split(',')#每一行存为列表
        file_dict=dict(zip(key,file))#用zip拉链成字典 file_dict
        if sql_dict[where_]:#where条件不为空
            where_l=sql_dict[where_]#定义where子句变量
            set_l=sql_dict[set_]#定义要更新的字符
            if where_conn(where_l,file_dict):#调用where子句判断函数  如果返回为真
                if tag:#如果为删除标示
                    line=''
                    list_l.append(line)#添加到文件列表
                    conut+=1
                else:
                    line_list=line.split(',')#分割为列表
                    cq='\"'
                    if cq in set_l[0][2]:#判断是否有包含 ”
                        set_l[0][2]=cq_c(set_l[0][2],cq)
                    for index, i in enumerate(line_list):#判断替换
                        if i == file_dict[where_l[0][0]]:
                            line_list[index]=set_l[0][2]#对字典进行替换
                    line=','.join(line_list)
                    list_l.append(line)#添加到文件列表
                    conut+=1
            else:
                list_l.append(line)
    write_file(list_l,file_name)#写入文件函数
    if tag:
        print('总共删除了 %s 条记录！'%conut)
    else:
        print('总共更新了 %s 条记录！'%conut)
    return #返回处理后的信息
#去 " 函数
def cq_c(set_l,cq):
    set__l=''
    for i in set_l:
        if i==cq:
            pass
        else:
            set__l+=i
    return set__l
#写文件函数
def write_file(list_l,file_name):#写文件函数
     with open(file_name,'w',encoding='utf-8') as new_file:
        for i in list_l:
            new_file.write(i)#写入文件
        new_file.flush()

#执行insert操作的函数
def perform_insert(sql_dict,user_info,file_name,key):#执行insert操作的函数
    list_user=sql_dict[values_]#提取要更新的列表
    phone_=list_user[2]#提取电话号码
    conut=1
    tag=True
    for line in user_info:
        conut+=1#统计行数
        file=line.strip().split(',')#每一行存为列表
        file_dict=dict(zip(key,file))#用zip拉链成字典 file_dict
        if phone_==file_dict['phone']:#如果有存在
            tag=False
            break
        else:
            pass
    if tag:#如果检测结果为真才能添加\
        list_user.insert(0,str(conut))#将字ID插入列表
        str_user='\n%s'%','.join(list_user)#需要加回车
        user_info.close()#关闭文件
        with open(file_name,'a',encoding='utf-8') as file_a:
            file_a.write(str_user)#
            file_a.flush()
            return print('信息已经更新!')
    else:
        return print('电话号码已经存在，请重新输入！')

#执行delete操作的函数
def perform_delete(sql_dict):#执行delete操作的函数
    select_dict=sql_dict
    return select_dict#返回处理后的信息


#程序开始
print("\033[35;1m欢迎进入员工信息表查询系统\033[0m".center(60,'='))
if __name__ =='__main__':
    while True:
        print('查询修改示例'.center(50,'-').center(60,' '))
        print('''
        \033[32;1m1、【查】:  select name,age from staff_table where age>22 and age<35 or name like li\033[0m
        \033[33;1m2、【改】:  UPDATE staff_table SET dept="Market" WHERE dept="IT"\033[0m
        \033[34;1m3、【增】:  insert into staff_table values Alex Li,22,13651054608,IT,2013-04-01\033[0m
        \033[31;1m3、【删】:  delete from staff_table where id=5 \033[0m
        \033[37;1m4、【退】:  exit\033[0m
        ''')
        sql=input('sql->').strip()#定义显示提示符
        if sql=='exit':break#如果输入exit退出
        if sql==0:continue#如何没输入重新提示
        sql_dict=sql_parsing(sql)#解析输入的sql语句 赋于变量
        try:
            sql_action=sql_perform(sql_dict)#执行解析后的语句 传入解析后的变量
        except Exception as e:
            print('你的输入格式有误,请重新输入!')

 

两天半时间才搞定,休息一下下