码迷,mamicode.com
首页 > 其他好文 > 详细

函数、文件操作实现数据增删改查---low版本

时间:2017-07-20 22:31:13      阅读:240      评论:0      收藏:0      [点我收藏+]

标签:脚本   python   


    首先说明这个脚本很low,目前水平有限,只能实现使用固定的语法对数据进行增删改查。但是现在脚本不low,怎么让明年的我来嘲笑今年的自己


需求
   a.可进行模糊查询,语法至少支持下面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"
     查到的信息,打印后,最后面还要显示查到的条数

   b.可创建新员工纪录,以phone做唯一键,staff_id需自增,语法如下:
     insert lyndon 24 13811688803 OP 20017-07-01

   c.可删除指定员工信息纪录,输入员工id,即可删除,语法如下:
     delete 删除序号(1-4)

   d.可修改员工信息,语法如下:
     UPDATE staff_table SET dept="Market" WHERE where dept = "IT"

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


3、测试信息
1,Alex Li,22,13651054608,IT,2013-04-01
2,Jack Wang,30,13304320533,HR,2015-05-03
3,Rain Liu,25,1383235322,Sales,2016-04-22
4,Mack Cao,40,1356145343,HR,2009-03-01


技术分享

def mem():
    print(‘‘‘
    ---------------------------使用说明------------------------------
    查询语法:
    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"
    创建语法:
    insert lyndon 24 13811688803 OP 20017-07-01
    删除语法:
    delete 序号(序号初始范围1-4)
    修改语法:
    UPDATE staff_table SET dept="Market" WHERE where dept = "IT"
    -----------------水平有限,只能满足以上语法----------------------
    【exit】      退出
    ‘‘‘)

def check_sql(sql):
    parse_sql={
        ‘insert‘:insert,
        ‘delete‘:delete,
        ‘update‘:update,
        ‘select‘:select
    }
    sql_l=sql.split(‘ ‘)
    func=sql_l[0]
    res=‘‘
    if func in parse_sql:
        res=parse_sql[func](sql_l)
    return res
    # print(sql_l)
def select(sql_l):      #[‘select‘, ‘name,age‘, ‘from‘, ‘staff_table‘, ‘where‘, ‘age‘, ‘>‘, ‘22‘]
    count=0
    if ‘age‘ in sql_l:
        with open(‘info.txt‘,‘r‘,encoding=‘utf-8‘) as f_read:
            for line in f_read:
                info_l=line.split(‘,‘)
                if int(info_l[2]) > 22:
                    count+=1
                    print(line)
    if ‘dept‘ in sql_l:
        with open(‘info.txt‘,‘r‘,encoding=‘utf-8‘) as f_read:
            for line in f_read:
                info_l=line.split(‘,‘)
                if info_l[4] == ‘IT‘:
                    count+=1
                    print(line)
    if ‘like‘ in sql_l:
        with open(‘info.txt‘,‘r‘,encoding=‘utf-8‘) as f_read:
            for line in f_read:
                info_l=line.split(‘,‘)
                if info_l[-1].split(‘-‘)[0] == ‘2013‘:
                    count += 1
                    print(line)
    print(‘共查询到 %s 条信息‘ %count)

def update(sql_l):      #[‘update‘, ‘staff_table‘, ‘set‘, ‘dept="market"‘, ‘where‘, ‘where‘, ‘dept‘, ‘=‘, ‘"it"‘]
    with open(‘info.txt‘, ‘r+‘, encoding=‘utf-8‘) as f_read:
        lines=f_read.readlines()
    with open(‘info.txt‘,‘w‘,encoding=‘utf-8‘) as f_write:
        for line in lines:
            if ‘IT‘ in line:
                line=line.replace(‘IT‘,‘Market‘,1)
                print(‘信息已更新 %s‘ %line)
            f_write.write(line)
def delete(sql_l):
    import os
    with open(‘info.txt‘, ‘r‘, encoding=‘utf-8‘) as f_read,            open(‘.info.swap.txt‘,‘w+‘,encoding=‘utf-8‘) as f_write:
        for line in f_read:
            info_l = line.split(‘,‘)
            if sql_l[1] == info_l[0] :
                print(‘%s 此信息已删除‘ %line)
            else:
                f_write.write(line)
    os.remove(‘info.txt‘)
    os.rename(‘.info.swap.txt‘,‘info.txt‘)
def insert(sql_l):      #[‘insert‘, ‘lyndon‘, ‘24‘, ‘13811688803‘, ‘op‘, ‘20017-07-01‘]
    phone=[]
    with open(‘info.txt‘, ‘r‘, encoding=‘utf-8‘) as f_read:
        for line in f_read:
            info_l = line.split(‘,‘)
            count=int(info_l[0])
            phone.append(info_l[3])
    if sql_l[3] not in phone:
        count+=1
        sql_l[0]=str(count)
        string=‘,‘.join(sql_l)
        with open(‘info.txt‘, ‘a‘, encoding=‘utf-8‘) as f:
            f.write(‘\n‘+string)
            print(‘%s 信息已插入‘ %string)
    else:
        print(‘此信息已存在‘)
while True:
    print(‘输入help查看帮助‘)
    sql=input(‘sql>>>‘).strip().lower()
    if len(sql) == 0:continue
    if sql == ‘exit‘:break
    if sql == ‘help‘:
        mem()
    im=check_sql(sql)


本文出自 “lyndon” 博客,请务必保留此出处http://lyndon.blog.51cto.com/11474010/1949364

函数、文件操作实现数据增删改查---low版本

标签:脚本   python   

原文地址:http://lyndon.blog.51cto.com/11474010/1949364

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!