Python函数案例——员工信息管理
员工信息管理
简单的员工信息增删改查程序
表信息
1,Alex Li,22,13651054608,IT,2013‐04‐01
2,Jack Wang,28,13451024608,HR,2015‐01‐07
3,Rain Wang,21,13451054608,IT,2017‐04‐01 增加
add staff_table Alex Li,25,134435344,IT,2015-10-29
以phone做唯一键(即不允许表里有手机号重复的情况)，staff_id需自增 查询支持三种语法
find name,age from staff_table where age > 22
find * from staff_table where dept = "IT"
find * from staff_table where enroll_date like "" 删除指定员工信息纪录
del from staff_table where dept = "IT" 更新记录
update staff_table set dept="Market" where dept = "IT"
update staff_table set age=25 where name = "Alex Li"
流程图
语法分析
def syntax_parser(input_sql):
"""
解析sql语句并执行
:param input_sql:
:return:
"""
# print_log("syntax_parser sql star")
syntax_list ={
'find': syntax_find,
'del': syntax_del,
'add': syntax_add,
'update': syntax_update
} if input_sql.split()[0] in syntax_list.keys() and 'staff_table' in input_str.split():
if 'where' in input_sql.split():
query_clause, where_clause = input_sql.split("where")
# print_log(query_clause + where_clause)
# 执行where条件
match_data = syntax_where(where_clause.strip())
input_action = query_clause.split()[0]
# 执行不同的action
syntax_list[input_action](match_data, query_clause)
else:
match_data = []
for ind, val in enumerate(STAFF_INFO["id"]):
row = []
for col in COLUMN_NAME:
row.append(STAFF_INFO[col][ind])
match_data.append(row)
syntax_list[input_sql.split()[0]](match_data, input_sql.strip()) else:
print_log("语法错误，find/del/add/updata name,age from [staff_table] [where] age [<,>,=,like][1]", 'error')
where语句分析
def syntax_where(where_clause):
"""
解析where条件
where age > 22
:param where_clause:
:return:
"""
# 操作字符
op_list = [">", "<", "=", "like"] for op_key in op_list:
if op_key in where_clause:
q_name, q_cond = where_clause.split(op_key)
if q_name.strip() in COLUMN_NAME and q_cond.strip() != "":
match_data = op_compare(q_name.strip(), q_cond.strip(),op_key)
return match_data
else:
if not q_name.strip() in COLUMN_NAME:
error_str = "语法错误,字段%s不存在" % q_name
else:
error_str = "条件值为空"
print_log(error_str, "error")
return False else:
print_log("语法错误,符号不在[<,>,=,like]中","error")
return False
where 语句的比较运算
def convert_str(f_val, f_cond, f_compare_str):
"""
字符串拼接成条件
:param f_cond:
:param f_val:
:return:
"""
if f_cond.find("'") != -1:
val = "'" + f_val + "'"
elif f_cond.find("\"") != -1:
val = "\"" + f_val + "\""
else:
val = "'" + f_val + "'"
f_cond = "'" + f_cond + "'"
if f_compare_str == 'in':
# 字符比较"'2015' in '2016-02-01'"
exp_str = "%s%s%s" % (val, f_compare_str, f_cond)
else:
# 字符比较
exp_str = "%s%s%s" % (f_cond, f_compare_str, val)
return exp_str def op_compare(q_name, q_cond, compare_str):
"""
解析where 语句的操作符
:param q_name:
:param q_cond:
:param compare_str:
:return:
"""
match_data = []
if compare_str == "=":
compare_str = "=="
for ind, val in enumerate(STAFF_INFO[q_name]):
if compare_str != "like" and q_cond.isdigit():
# 数字比较
exp_str = "%d%s%d" % (int(val), compare_str, int(q_cond))
elif compare_str != "like" and not q_cond.isdigit():
# 转换操作符两边字符串
# 把val两边加上'val'或"val" ,与输入字符串比较 'Sales' = 'Sales' or "Sales" = "Sales" # 字符比较
exp_str = convert_str(val, q_cond, compare_str)
# print_log(exp_str)
else:
# if compare_str = like then compare_str = ' in '
op_str = ' in '
# 字符比较
exp_str = convert_str(val, q_cond, op_str) # print_log(exp_str)
if eval(exp_str):
row_data = []
for col in COLUMN_NAME:
row_data.append(STAFF_INFO[col][ind])
match_data.append(row_data)
# print(tabulate(match_data, headers=COLUMN_NAME, tablefmt="grid"))
return match_data
解析删除语句
def syntax_del(dataset, query_clause):
"""
解析删除语句
del from staff_table where id=3
:param dataset:
:param query_clause:
:return:
"""
for row in dataset:
staff_id = row[0] # 得到id值
staff_index = STAFF_INFO['id'].index(staff_id) # 得到id值在STAFF_INFO[id]的索引
# print_log(staff_index)
for col in COLUMN_NAME:
STAFF_INFO[col].remove(STAFF_INFO[col][staff_index]) # 修改col_name值 save_db()
print_log("成功删除%s条纪录" % len(dataset))
解析增加语句
def syntax_add(dataset, query_clause):
"""
解析增加语句
add staff_table Alex Li,25,134435344,IT,2015-10-29
:param dataset: dataset = [[1,Alex Li,18,13651054608,开发,2013-04-01]]
:param query_clause:
:return:
"""
# 得到增加的值列表
add_data = [col.strip() for col in query_clause.split("staff_table")[-1].split(',')]
phone_ind = COLUMN_NAME.index("phone") # 得到手机所在列
if len(COLUMN_NAME) - 1 == len(add_data):
# 得到最后一行数据，自增长最后一行数据Id最大
max_id = dataset[-1][0]
# 自增长ID
max_id = int(max_id) + 1
# 把ID插入到第一列
add_data.insert(0,str(max_id))
# 得到手机号
phone_val = add_data[phone_ind]
# 判断手机号是否重复
if not (phone_val in STAFF_INFO["phone"]):
# 把数据插入到STAFF_INFO
for index, col in enumerate(COLUMN_NAME):
STAFF_INFO[col].append(add_data[index]) print(tabulate(STAFF_INFO, headers=COLUMN_NAME))
save_db()
print_log("成功添加1条纪录到staff_table表")
else:
print_log("手机号%s重复" %phone_val ,'error') else:
print_log("语法错误，列数不对,必须字段%s："% COLUMN_NAME[1:], "error")
解析修改语句
def syntax_update(dataset, query_clause):
"""
修改语句 update staff_table set age=25 where name='Alex Li'
:param dataset:
:param query_clause:
:return:
"""
if "set" in query_clause:
formula_str = query_clause.split("set")
col_name, new_val = formula_str[-1].strip().split('=')
# print(col_name, new_val)
# STAFF_INFO[col_name]
if new_val.find("'") == 0:
new_val = new_val.replace("'", "")
elif new_val.find("\"") == 0:
new_val = new_val.replace("\"", "")
for row in dataset:
staff_id = row[0] # 得到id值
staff_index = STAFF_INFO['id'].index(staff_id) # 得到id值在STAFF_INFO[id]的索引
STAFF_INFO[col_name][staff_index] = new_val # 修改col_name值
# print_log(STAFF_INFO)
save_db()
print_log("成功修改了%s条数据!" % len(dataset))
else:
print_log("语法错误，未检测到set", "error")
保存数据
def save_db():
"""
#把修改的数据保存到硬盘上
:return:
"""
f = open("%s.new"%DB_FILE, "w",encoding="utf-8")
# for k in COLUMN_NAME:
# row = []
# row = STAFF_INFO[k]
for ind, val in enumerate(STAFF_INFO["id"]):
row = []
for col in COLUMN_NAME:
row.append(STAFF_INFO[col][ind])
w_data = ",".join(row)
# print_log(w_data)
f.write(w_data+"\n")
f.close() os.remove(DB_FILE)
os.rename("%s.new"%DB_FILE, DB_FILE) # 回写原来的文件
解析查询语句
def syntax_find(dataset, query_clause):
"""
查询语句解析
:param dataset:
:param query_clause:
:return:
"""
query_col = query_clause.split("from")[0][4:].split(',') # 得到name,age
filter_col = [col.strip() for col in query_col] # 去除col中的空格
# print(query_col)
reformat_dataset = []
if '*' in filter_col: # 输出所有列
print(tabulate(dataset, headers=COLUMN_NAME, tablefmt="grid"))
print_log("匹配到%s条数据!" % len(dataset))
else:
for row in dataset: # 筛选列
filter_val = []
for col in filter_col:
index = COLUMN_NAME.index(col)
filter_val.append(row[index])
reformat_dataset.append(filter_val)
print(tabulate(reformat_dataset, headers=COLUMN_NAME, tablefmt="grid"))
print_log("匹配到%s条数据!" % len(reformat_dataset))
主函数
if __name__ == '__main__':
"""
用户信息查询
"""
# 把数据载入内存
STAFF_INFO = load_db()
while True:
input_str = input("[staff db:]").strip()
if not input_str : continue
syntax_parser(input_str)
详细代码
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2018/3/17 12:27
# @Site :
# @File : staff_info_manage.py
# @Software: PyCharm """
简单的员工信息增删改查程序
表信息
1,Alex Li,22,13651054608,IT,2013‐04‐01
2,Jack Wang,28,13451024608,HR,2015‐01‐07
3,Rain Wang,21,13451054608,IT,2017‐04‐01 增加
add staff_table Alex Li,25,134435344,IT,2015-10-29
以phone做唯一键(即不允许表里有手机号重复的情况)，staff_id需自增 查询支持三种语法
find name,age from staff_table where age > 22
find * from staff_table where dept = "IT"
find * from staff_table where enroll_date like "2013" 删除指定员工信息纪录
del from staff_table where dept = "IT" 更新记录
update staff_table set dept="Market" where dept = "IT"
update staff_table set age=25 where name = "Alex Li"
"""
from tabulate import tabulate
import os # 存储信息的文件
DB_FILE = "staff.db"
COLUMN_NAME = ['id', 'name', 'age', 'phone', 'dept', 'enroll_date'] # 打印信息
def print_log(msg, log_type="info"):
if log_type == 'info':
print("\033[32;1m%s\033[0m" %msg)
elif log_type == 'error':
print("\033[31;1m%s\033[0m" %msg) def load_db():
"""
载入员工信息
:return:
"""
staff_data = {}
# 构建字典空列表
#{'id': [],'name':[],'age': [],'phone':[],'dept':[],'enrolled_date'[]
for d in COLUMN_NAME:
staff_data[d] = [] with open(DB_FILE, 'r', encoding='utf-8') as f:
for line in f:
staff_datas = line.split(",")
# 构建员工信息字典
for ind, d in enumerate(staff_datas):
staff_data[COLUMN_NAME[ind]].append(d.strip()) # 去掉末尾回车 return staff_data def convert_str(f_val, f_cond, f_compare_str):
"""
字符串拼接成条件
:param f_cond:
:param f_val:
:return:
"""
if f_cond.find("'") != -1:
val = "'" + f_val + "'"
elif f_cond.find("\"") != -1:
val = "\"" + f_val + "\""
else:
val = "'" + f_val + "'"
f_cond = "'" + f_cond + "'"
if f_compare_str == 'in':
# 字符比较"'2015' in '2016-02-01'"
exp_str = "%s%s%s" % (val, f_compare_str, f_cond)
else:
# 字符比较
exp_str = "%s%s%s" % (f_cond, f_compare_str, val)
return exp_str def op_compare(q_name, q_cond, compare_str):
"""
解析where 语句的操作符
:param q_name:
:param q_cond:
:param compare_str:
:return:
"""
match_data = []
if compare_str == "=":
compare_str = "=="
for ind, val in enumerate(STAFF_INFO[q_name]):
if compare_str != "like" and q_cond.isdigit():
# 数字比较
exp_str = "%d%s%d" % (int(val), compare_str, int(q_cond))
elif compare_str != "like" and not q_cond.isdigit():
# 转换操作符两边字符串
# 把val两边加上'val'或"val" ,与输入字符串比较 'Sales' = 'Sales' or "Sales" = "Sales" # 字符比较
exp_str = convert_str(val, q_cond, compare_str)
# print_log(exp_str)
else:
# if compare_str = like then compare_str = ' in '
op_str = ' in '
# 字符比较
exp_str = convert_str(val, q_cond, op_str) # print_log(exp_str)
if eval(exp_str):
row_data = []
for col in COLUMN_NAME:
row_data.append(STAFF_INFO[col][ind])
match_data.append(row_data)
# print(tabulate(match_data, headers=COLUMN_NAME, tablefmt="grid"))
return match_data def syntax_where(where_clause):
"""
解析where条件
where age > 22
:param where_clause:
:return:
"""
# 操作字符
op_list = [">", "<", "=", "like"] for op_key in op_list:
if op_key in where_clause:
q_name, q_cond = where_clause.split(op_key)
if q_name.strip() in COLUMN_NAME and q_cond.strip() != "":
match_data = op_compare(q_name.strip(), q_cond.strip(),op_key)
return match_data
else:
if not q_name.strip() in COLUMN_NAME:
error_str = "语法错误,字段%s不存在" % q_name
else:
error_str = "条件值为空"
# print_log(error_str, "error")
return False else:
print_log("语法错误,符号不在[<,>,=,like]中","error")
return False def syntax_del(dataset, query_clause):
"""
解析删除语句
del from staff_table where id=3
:param dataset:
:param query_clause:
:return:
"""
for row in dataset:
staff_id = row[0] # 得到id值
staff_index = STAFF_INFO['id'].index(staff_id) # 得到id值在STAFF_INFO[id]的索引
# print_log(staff_index)
for col in COLUMN_NAME:
STAFF_INFO[col].remove(STAFF_INFO[col][staff_index]) # 修改col_name值 save_db()
print_log("成功删除%s条纪录" % len(dataset)) def syntax_add(dataset, query_clause):
"""
解析增加语句
add staff_table Alex Li,25,134435344,IT,2015-10-29
:param dataset: dataset = [[1,Alex Li,18,13651054608,开发,2013-04-01]]
:param query_clause:
:return:
"""
# 得到增加的值列表
add_data = [col.strip() for col in query_clause.split("staff_table")[-1].split(',')]
phone_ind = COLUMN_NAME.index("phone") # 得到手机所在列
if len(COLUMN_NAME) - 1 == len(add_data):
# 得到最后一行数据，自增长最后一行数据Id最大
max_id = dataset[-1][0]
# 自增长ID
max_id = int(max_id) + 1
# 把ID插入到第一列
add_data.insert(0,str(max_id))
# 得到手机号
phone_val = add_data[phone_ind]
# 判断手机号是否重复
if not (phone_val in STAFF_INFO["phone"]):
# 把数据插入到STAFF_INFO
for index, col in enumerate(COLUMN_NAME):
STAFF_INFO[col].append(add_data[index]) print(tabulate(STAFF_INFO, headers=COLUMN_NAME))
save_db()
print_log("成功添加1条纪录到staff_table表")
else:
print_log("手机号%s重复" %phone_val ,'error') else:
print_log("语法错误，列数不对,必须字段%s："% COLUMN_NAME[1:], "error") def syntax_update(dataset, query_clause):
"""
修改语句 update staff_table set age=25 where name='Alex Li'
:param dataset:
:param query_clause:
:return:
"""
if "set" in query_clause:
formula_str = query_clause.split("set")
col_name, new_val = formula_str[-1].strip().split('=')
# print(col_name, new_val)
# STAFF_INFO[col_name]
if new_val.find("'") == 0:
new_val = new_val.replace("'", "")
elif new_val.find("\"") == 0:
new_val = new_val.replace("\"", "")
for row in dataset:
staff_id = row[0] # 得到id值
staff_index = STAFF_INFO['id'].index(staff_id) # 得到id值在STAFF_INFO[id]的索引
print(STAFF_INFO[col_name][staff_index])
STAFF_INFO[col_name][staff_index] = new_val # 修改col_name值
# print_log(STAFF_INFO)
save_db()
print_log("成功修改了%s条数据!" % len(dataset))
else:
print_log("语法错误，未检测到set", "error") def save_db():
"""
#把修改的数据保存到硬盘上
:return:
"""
f = open("%s.new"%DB_FILE, "w",encoding="utf-8")
# for k in COLUMN_NAME:
# row = []
# row = STAFF_INFO[k]
for ind, val in enumerate(STAFF_INFO["id"]):
row = []
for col in COLUMN_NAME:
row.append(STAFF_INFO[col][ind])
w_data = ",".join(row)
# print_log(w_data)
f.write(w_data+"\n")
f.close() os.remove(DB_FILE)
os.rename("%s.new"%DB_FILE, DB_FILE) # 回写原来的文件 def syntax_parser(input_sql):
"""
解析sql语句并执行
:param input_sql:
:return:
"""
# print_log("syntax_parser sql star")
syntax_list ={
'find': syntax_find,
'del': syntax_del,
'add': syntax_add,
'update': syntax_update
} if input_sql.split()[0] in syntax_list.keys() and 'staff_table' in input_str.split():
if 'where' in input_sql.split():
query_clause, where_clause = input_sql.split("where")
# print_log(query_clause + where_clause)
# 执行where条件
match_data = syntax_where(where_clause.strip())
input_action = query_clause.split()[0]
# 执行不同的action
syntax_list[input_action](match_data, query_clause)
else:
match_data = []
for ind, val in enumerate(STAFF_INFO["id"]):
row = []
for col in COLUMN_NAME:
row.append(STAFF_INFO[col][ind])
match_data.append(row)
syntax_list[input_sql.split()[0]](match_data, input_sql.strip()) else:
print_log("语法错误，find/del/add/update name,age from [staff_table] [where] age [<,>,=,like][1]", 'error') def syntax_find(dataset, query_clause):
"""
查询语句解析
:param dataset:
:param query_clause:
:return:
"""
query_col = query_clause.split("from")[0][4:].split(',') # 得到name,age
filter_col = [col.strip() for col in query_col] # 去除col中的空格
# print(query_col)
reformat_dataset = []
if '*' in filter_col: # 输出所有列
print(tabulate(dataset, headers=COLUMN_NAME, tablefmt="grid"))
print_log("匹配到%s条数据!" % len(dataset))
else:
for row in dataset: # 筛选列
filter_val = []
for col in filter_col:
index = COLUMN_NAME.index(col)
filter_val.append(row[index])
reformat_dataset.append(filter_val)
print(tabulate(reformat_dataset, headers=COLUMN_NAME, tablefmt="grid"))
print_log("匹配到%s条数据!" % len(reformat_dataset)) if __name__ == '__main__':
"""
用户信息查询
"""
# 把数据载入内存
STAFF_INFO = load_db()
while True:
input_str = input("[staff db:]").strip()
if not input_str : continue
syntax_parser(input_str)
详细代码
Python函数案例——员工信息管理的更多相关文章
- python函数小案例
python函数 目录 python函数 1.写一个函数求三个数的和,并返回结果 2.写一个函数,求平均值,并返回结果 写一个函数,求每个数与平均值之间的差,并放回结果 1.写一个函数求三个数的和,并 ...
- Day03 - Python 函数
1. 函数简介 函数是组织好的,可重复使用的,用来实现单一或相关联功能的代码段. 函数能提高应用的模块性,和代码的重复利用率.Python提供了许多内建函数,比如print():也可以自己创建函数,这 ...
- Python入门(案例)
Python入门(案例) #一.上课案例: #输出hello wordprint('hello word') #python注释有两种#1.单行注释#这是单行注释#2.多行注释'''这是多行注释''' ...
- Python 函数定义和使用
# 函数的概念 # 概念 # 写了一段代码实现了某个小功能; 然后把这些代码集中到一块, 起一个名字; 下一次就可以根据这个名字再次使用这个代码块, 这就是函数 # 作用 # 方便代码的重用 # 分解 ...
- Python函数中参数类型
在学习Python函数的时候,函数本身的定义和调用并不是很复杂,但是函数的参数类型和用法的确有些复杂.在此做一个小结,加深理解. Python参数的定义 负责给函数提供一些必要的数据或信息,以保证函数 ...
- python函数闭包-装饰器-03
可调用对象 callable() # 可调用的(这个东西加括号可以执行特定的功能,类和函数) 可调用对象即 callable(对象) 返回为 True 的对象 x = 1 print(cal ...
- Python 函数返回值、作用域
函数返回值 多条return语句: def guess(x): if x > 3: return "> 3" else: return "<= 3&qu ...
- python 函数、参数及参数解构
函数 数学定义 y=f(x), y是x函数,x是自变量.y=f(x0,x1...xn) Python函数 由若干语句组成的语句块,函数名称,参数列表构成,它是组织代码的最小单位 完成一定的功能 函数作 ...
- Python函数07/有参装饰器/多个装饰器装饰一个函数
Python函数07/有参装饰器/多个装饰器装饰一个函数 目录 Python函数07/有参装饰器/多个装饰器装饰一个函数 内容大纲 1.有参装饰器 2.多个装饰器装饰一个函数 3.今日总结 3.今日练 ...
随机推荐
- Ubuntu上搭建SVN
参考文档:http://www.linuxidc.com/Linux/2016-08/133961.htm http://www.linuxidc.com/Linux/2015-01/111956.h ...
- C语言_初步了解一下指针
指针的基本概念 在计算机中,所有的数据都是存放在存储器中的. 一般把存储器中的一个字节称为一个内存单元, 不同的数据类型所占用的内存单元数不等,如整型量占2个单元,字符量占1个单元等.为了正确地访问这 ...
- MysqL 磁盘写入策略之innodb_flush_log_at_trx_commit
本文从参数含义,性能,安全角度阐述两个参数为不同的值时对db 性能,数据的影响,引擎是Innodb的前提下. 取值:0/1/2 innodb_flush_log_at_trx_commit=0,表示每 ...
- UVA - 10723 类似LCS
思路:dp(i, j)表示第一个串前i个字符和第二个串前j个字符需要的最短字符串长度,cnt(i, j)表示第一个串前i个字符和第二个串前j个字符需要的最短字符串的个数. 转移方程: if(s1[i] ...
- uva1025 动态规划
这道题的边界是dp(T,N)=0,状态dp(i,j)表示在时间i.第j个车站最少等待时间,有三个决策:1.等1分钟 2.如果有向左的车,向左 3.若果有向右的车,向右 转移方程就是dp(i,j)=m ...
- POJ - 2698 贪心
经典面替换算法,每次选择最远的那个碟片请求进行替换. AC代码 #include <cstdio> #include <cmath> #include <algorith ...
- DevExpress XtraGrid如何使单元格只读？
-----------------------------从别人那里copy来的-------------------------------------------------- 1. 设置Gr ...
- Redis笔记1-redis的搭建和使用
1. Redis的安装 1.1. Redis的安装 Redis是c语言开发的. 安装redis需要c语言的编译环境.如果没有gcc需要在线安装.yum install gcc-c++ 安装步骤: ...
- ORACLE中修改表的Schema的总结
前阵子遇到一个案例,需要将数据库中的几个表从USER A 移动到USER B下面,在ORACLE中,这个叫做更改表的所有者或者修改表的Schema.其实遇到这种案例,有好几种解决方法.下面我们通过实验 ...
- PHP 常量dirname(__file__)
PHP 常量dirname(__FILE__)取得当前文件的绝对路径. define('ROOT_PATH', str_replace('includes/2.php', '', str_replac ...