今天分享又来了呀。ღ( ´･ᴗ･` ) 一起学习进步ღゝ◡╹)ノ♡
摘要：最近在开发etl工具，其中使用到了Python命令行编程，记录一下。
IMDG：
加油，好好学习，天天向上~
Q:
IMDG 重要性？
pycharm中用Terminal打开，这个目录就是项目所在的目录
D:\PythonCode\blibli\WorkTool>
Python运行程序的指令
D:\PythonCode\blibli\WorkTool>python start.py
传入第一个 参数指令
D:\PythonCode\blibli\WorkTool>python start.py hello
1 命令行编程
import sys
# 命令行编程
# 标准库 sys
# 1构建程序的框架 增加一个info指令
print('wtt')
print(sys.argv)
在终端中运行：
D:\PythonCode\blibli\WorkTool>python start.py
wtt
['start.py']
D:\PythonCode\blibli\WorkTool>python start.py hello
wtt
['start.py', 'hello'] 可以看见，这是一个列表，列表里面每个元素都是字符串。第一个元素是我们程序的名字。
D:\PythonCode\blibli\WorkTool>python start.py hello word python
wtt
['start.py', 'hello', 'word', 'python']import sys
# 命令行编程
# 标准库 sys
# 1构建程序的框架 增加一个info指令
print('wtt')
print(sys.argv)
programName=sys.argv[0] #程序的名字，保存为一个变量
print(programName)
终端执行：
D:\PythonCode\blibli\WorkTool>python start.py
wtt
['start.py']
start.py# 工作情况分析
import sys
# 命令行编程
# 标准库 sys
# 1构建程序的框架 增加一个info指令
programName=sys.argv[0] #程序的名字，保存为一个变量
if len(sys.argv) >1:
pass
else:
print("命令使用：")
print("\tpython %s 命令[params]" % programName)
---------------------------------
D:\PythonCode\blibli\WorkTool>python start.py
命令使用：
python start.py 命令[params]# 工作情况分析
import sys
# 命令行编程
# 标准库 sys
# 1构建程序的框架 增加一个info指令
programName = sys.argv[0] # 程序的名字，保存为一个变量
def do_info():
'''info指令'''
pass
if len(sys.argv) > 1:
command = sys.argv[1]
print("command = %s" % command)
if command == "info":
do_info()# 如果命令是info，我想调用函数，执行info要做的事情
else:
print("命令使用：")
print("\tpython %s 命令[params]" % programName)
print("\tpython %s info" % programName)start1.py
# 工作情况分析
import sys
import xlrd #读取excel的第三方库
# 命令行编程
# 标准库 sys
# 1构建程序的框架 增加一个info指令
# 2给info指令增加第一个参数的实现，传入filepath；输出数据包含的月份信息
def __loadWorkList(filePath): # 私有方法，读取我们的工作簿
wb =xlrd.open_workbook(filename=filePath)
sheet =wb.sheet_by_index(0) #打开第一个sheet
res=[] # 我们要获得结果，结果应该是一个list
for i in range(sheet.nrows)[2:-1]: #对列表进行切片。从第3行开始到倒数第二行。前面去掉2行，后面去掉一行
#sheet.row_values(i) #获取每一行的值
# print(sheet.row_values(i))
res.append(sheet.row_values(i))
return res
def do_info():
"""info指令"""
_help ='正确的指令输入格式: python %s info filepath' %programName # 字符串，来提示用户输入正确的命令行格式
if len(sys.argv) >2: # 传入的第二个指令参数，我希望是程序的路径
filePath = sys.argv[2]
# print(filePath) # 这行是用来测试是否获取到传入的路径的
res =__loadWorkList(filePath)
# months= [r[3][:7] for r in res] # 数据日期:2020-06,2020-06,2020-06,2020-06,2020-06,2020-06
months= {r[3][:7] for r in res} #数据日期: 2020 - 06 推导式写法，每一行拿出来，每一行都是一个列表。取出列表的第三项，截取第三项的前7位。外面符合是{}，那就是集合推导式
# print(months) #['2020-06', '2020-06', '2020-06'] {'2020-06'}
print('数据日期:'+','.join(months)) #字符串的join方法，将序列中的每一个元素用,拼接起来
else:
print(_help)
programName=sys.argv[0]
if len(sys.argv) > 1:
command =sys.argv[1]
# print('command =%s'%command)
if command=='info':
do_info()
else:
print('命令使用方法：')
print("\tpython %s COMMAND [PARAMS] " % programName)
print("\tpython %s info " % programName)学会封装函数，让每一个函数专注于某一项功能。
知道每个函数是用来解决什么问题的。
start2.py
时间戳 代表1970年1月1日0时，到现在这个时间的间隔秒数
# 工作情况分析
import sys
import xlrd
import re
# 命令行编程
# 1 构建程序的框架 增加一个info指令
# 2 给info指令增加第一个参数的实现， filepath命令行参数。输出数据包含的月份信息
# 3 给info指令增加第二个参数 month命令行参数。 规定格式：yyyy-mm格式的字符串 表示年和月 输出每个人工作统计结果
def __loadWorkList(filePath):
wb =xlrd.open_workbook(filename=filePath)
sheet =wb.sheet_by_index(0)
res=[]
for i in range(sheet.nrows)[2:-1]:
#print(sheet.row_values(i))
res.append(sheet.row_values(i))
return res
def calworktimes(date,begin,end): # 3个形参和一个返回值
"""函数逻辑：通过日期、签到、签退时间，计算当天的加班的分钟"""
return 1
def creatPersonData(filePath,month):
res = __loadWorkList(filePath) # 读取内容之后，循环遍历文件的内容
#首先要筛选，这个month的行
res =[line for line in res if line[3][:7] ==month] # 拿到Excel中 每一行的第三位元素（列） 截取前七位
# print(res)
username=None # 先把变量给定义出来，定义为临时变量
user ={} # 用字典数据结构来保存这个人的信息
userDataList=[] # 列表，存放所有人的记录
for r in res: # 遍历所有记录
if username ==r[0]: # 还是这个人 相同的姓名
# 累计当前人的数据
if r[8] =='正常' and r[9] =='正常': # Excel中第8和9个元素（从0开始） 签到状态为‘正常’
user ['zc'] =user ['zc']+1 # 正常 + 1
else:
user['qt'] = user['qt']+1 # 其他 +1
# 没换人时，把数据给累积起来
user ['addwork'] =user ['addwork'] +calworktimes(r[3],r[4],r[5]) # Excel处理加班时长 后面是个函数，加班时间和3个字段有关 日期、签到、签退 所以我们要传入这3个参数
else: # 换人了
if username: # 不是第一行的情况，才保存 等价于 if username !=None:
userDataList.append(user) # 保存 上一个人的数据 到列表中 。假如username是null，还是临时变量的时候，是不能够保存的
# 初始化个人数据
user ={'name':r[0],'zc':0,'qt':0,'addwork':0} # 人名 正常的天数 不正常天数 加班小时数 进行一个数据结构的封装
username =r[0]
for u in userDataList:
print(u)
def do_info():
"""info指令"""
_help ='请输入正确格式的指令: python %s info filePath [YYYY-MM]' %programName
if len(sys.argv) >2:
filePath = sys.argv[2]
if len(sys.argv)>3: #输入的第3个命令行参数 info filePath month
month =sys.argv[3]
print(month)
# 统计每个人的工作数据 使用正则表达式来看看输入的参数是否满足格式的要求！
if not re.match(r'^\d{4}-\d{2}$',month): # 结果是false的情况下 检查月份的格式 \d表示一个数字 {4}重复4次
print(_help) # 打印提示信息
return # 程序一定要结束
else:
creatPersonData(filePath,month) # 函数 从filePath这个文件里读取month这个月份的数据
else:
res =__loadWorkList(filePath)
months= {r[3][:7] for r in res} # 推导式写法
print('数据日期:'+','.join(months))
else:
print(_help)
return #因为输出不符合我们的预期，我们一定要return来使程序结束
programName=sys.argv[0]
if len(sys.argv) > 1:
command =sys.argv[1]
# print('command =%s'%command)
if command=='info':
do_info()
else:
print('命令使用方法：')
print("\tpython %s COMMAND [PARAMS] " % programName)
print("\tpython %s info FILEPATH [YYYY-MM] " % programName)start3.py
import sys
import xlrd
import re
import datetime
# 命令行编程
# 标准库 sys
# 1 构建程序的框架 增加一个info指令
# 2 给info指令增加第一个参数的实现， filepath输出数据包含的月份信息
# 3 给info指令增加第二个参数 month 是一个yyyy-mm格式的字符串表示年和月 输出每个人工作统计结果
# 4 完善加班时间统计的方法
def __loadWorkList(filePath):
wb = xlrd.open_workbook(filename=filePath)
sheet = wb.sheet_by_index(0)
res = []
for i in range(sheet.nrows)[2:-1]:
# print(sheet.row_values(i))
res.append(sheet.row_values(i))
return res
def calworktimes(date, begin, end):
"""通过日期、签到、签退时间，计算当天的加班的分钟"""
# 我们要知道这个日期是工作日还是休息日
# 加上int()，把返回的字符串转换成整数
week = int(datetime.datetime.strptime(date, '%Y-%m-%d').strftime(
'%w')) # strptime()把字符串转换成时间戳,并且 转换成格式化时间 最后是转换成星期几，星期天是0 星期一是1
minute = 0
if 0 < week < 6: # 周一到周五 也就是工作日
if re.match(r'^\d{2}:\d{2}:\d{2}$', end): # 判断时间格式是否正确
b_time = datetime.datetime.strptime('17:30:00', '%H:%M:%S') # 开始时间
e_time = datetime.datetime.strptime(end, '%H:%M:%S') # 结束时间
if e_time > b_time:
minute = (e_time - b_time).seconds // 60 # 拿到 加班的 间隔的秒数，除以60 拿到分钟
else: # 非工作日
if re.match(r'^\d{2}:\d{2}:\d{2}$', begin) and re.match(r'^\d{2}:\d{2}:\d{2}$', end): # 判断时间格式是否正确
b_time = datetime.datetime.strptime(begin, '%H:%M:%S')
e_time = datetime.datetime.strptime(end, '%H:%M:%S')
if e_time > b_time:
minute = (e_time - b_time).seconds // 60
return minute # 返回直接的加班时间
# calworktimes('2020-07-31',None,None)
# sys.exit(0)
def creatPersonData(filePath, month):
res = __loadWorkList(filePath)
res = [line for line in res if line[3][:7] == month]
print(len(res))
username = None
user = {}
userDataList = []
for r in res: # 遍历所有记录
if username == r[0]: # 还是这个人
# 累计当前人的数据
if r[8] == '正常' and r[9] == '正常':
user['zc'] = user['zc'] + 1
else:
user['qt'] = user['qt'] + 1
user['addwork'] = user['addwork'] + calworktimes(r[3], r[4], r[5])
else: # 换人了
if username: # 不是第一行的情况，才保存
userDataList.append(user) # 保存上一个人的数据
# 初始化个人数据
user = {'name': r[0], 'zc': 0, 'qt': 0, 'addwork': 0}
username = r[0]
with open('data/' + month + '.txt', 'w') as df: #保存下来，在当前目录下，子目录data，以month为名字的txt文件。覆盖内容w。
for u in userDataList: # u 是一个对象，还不能直接往文件里写入。需要格式化下
linestr = '{}|{}|{}|{}'.format(u['name'], u['zc'], u['qt'], u['addwork']) # 用 |进行分割
df.write(linestr + '\n') # 每行有换行符
def do_info():
"""info指令"""
_help = '指令格式: python %s info FILEPATH [YYYY-MM]' % programName
if len(sys.argv) > 2:
filePath = sys.argv[2]
if len(sys.argv) > 3:
month = sys.argv[3]
# 统计每个人的工作数据
if not re.match(r'^\d{4}-\d{2}$', month): # 检查月份的格式
print(_help)
return
else:
creatPersonData(filePath, month)
else:
res = __loadWorkList(filePath)
months = {r[3][:7] for r in res} # 推导式写法
print('数据日期:' + ','.join(months))
else:
print(_help)
return
programName = sys.argv[0]
if len(sys.argv) > 1:
command = sys.argv[1]
# print('command =%s'%command)
if command == 'info':
do_info()
else:
print('命令使用方法：')
print("\tpython %s COMMAND [PARAMS] " % programName)
print("\tpython %s info FILEPATH [YYYY-MM] " % programName)注意，出现报错
FileNotFoundError: [Errno 2] No such file or directory: 'data/2020-06.txt'
解决方法：
自己在项目中新建个文件夹 data。python start3.py info e:\6.xls 2020-06
start4.py
# 工作情况分析
import sys
import xlrd
import re
import datetime
# 命令行编程
# 标准库 sys
# 1 构建程序的框架 增加一个info指令
# 2 给info指令增加第一个参数的实现， filepath输出数据包含的月份信息
# 3 给info指令增加第二个参数 month 是一个yyyy-mm格式的字符串表示年和月 输出每个人工作统计结果
# 4 完善加班时间统计的方法
# 5 增加对假期特殊的处理
__以下是对数据获取和加工的功能的代码___
def __loadWorkList(filePath):
wb =xlrd.open_workbook(filename=filePath)
sheet =wb.sheet_by_index(0)
res=[]
for i in range(sheet.nrows)[2:-1]:
# print(sheet.row_values(i))
res.append(sheet.row_values(i))
return res
def loadHoliday():
res =[]
with open('data/holiday.txt','r') as hf : # 读取出来
for line in hf.readlines():
_holiday =line.split('|')[0]
_week = line.split('|')[1]
_type = line.split('|')[2].strip() # 去掉回车符号
#周一到周五，并且是休息日
if (0<int(_week)<6 and _type.lower() =="h") or \
((int(_week)==0 or int(_week)==6) and _type.lower())=='w': # 周六 周日 工作日
res.append(_holiday)
return res
def calworktimes(date,begin,end):
"""通过日期、签到、签退时间，计算当天的加班的分钟"""
# 我们要知道这个日期是工作日还是休息日
week =int(datetime.datetime.strptime(date,'%Y-%m-%d').strftime('%w'))
minute =0
holidays =loadHoliday()
# 周一到周五 非休息日 和 周六周日 的工作日
if (0< week <6 and date not in holidays) or ((week==0 or week==6) and date in holidays): # 不在假期的列表里 满足这些条件的，就是 工作日
if re.match(r'^\d{2}:\d{2}:\d{2}$',end):
b_time =datetime.datetime.strptime('17:30:00','%H:%M:%S')
e_time =datetime.datetime.strptime(end,'%H:%M:%S')
if e_time>b_time:
minute =(e_time-b_time).seconds //60
minute = minute if minute > 60 else 0 # 三元表达式 只有加班时间大于60分钟才算加班，否则时长就是0
else:
if re.match(r'^\d{2}:\d{2}:\d{2}$',begin) and re.match(r'^\d{2}:\d{2}:\d{2}$', end):
b_time = datetime.datetime.strptime(begin, '%H:%M:%S')
e_time = datetime.datetime.strptime(end, '%H:%M:%S')
if e_time>b_time:
minute =(e_time-b_time).seconds //60
minute = minute if minute <480 else 480 # 非工作日，做多加班480分钟
return minute
# calworktimes('2020-07-31',None,None)
# sys.exit(0)
def creatPersonData(filePath,month):
res = __loadWorkList(filePath)
res =[line for line in res if line[3][:7] ==month]
# print(res)
username=None
user ={}
userDataList=[]
for r in res: # 遍历所有记录
if username ==r[0]: # 还是这个人
# 累计当前人的数据
if r[8] =='正常' and r[9] =='正常':
user ['zc'] =user ['zc']+1
else:
user['qt'] = user['qt']+1
user ['addwork'] =user ['addwork'] +calworktimes(r[3],r[4],r[5])
else: # 换人了
if username: # 不是第一行的情况，才保存
userDataList.append(user) # 保存上一个人的数据
# 初始化个人数据
user ={'name':r[0],'zc':0,'qt':0,'addwork':0}
username =r[0]
with open('data/'+month+'.txt','w') as df:
for u in userDataList:
linestr ='{}|{}|{}|{}'.format(u['name'],u['zc'],u['qt'],u['addwork'])
df.write(linestr+'\n')
def do_info():
"""info指令"""
_help ='指令格式: python %s info FILEPATH [YYYY-MM]' %programName
if len(sys.argv) >2:
filePath = sys.argv[2]
if len(sys.argv)>3:
month =sys.argv[3]
# 统计每个人的工作数据
if not re.match(r'^\d{4}-\d{2}$',month): # 检查月份的格式
print(_help)
return
else:
creatPersonData(filePath,month)
else:
res =__loadWorkList(filePath)
months= {r[3][:7] for r in res} # 推导式写法
print('数据日期:'+','.join(months))
else:
print(_help)
return
def do_addholiday(): # 增加节假日
_help ='正确的指令格式: python %s addholiday YYYY-MM-DD TYPE(w:上班 h:休息)'%programName
if len(sys.argv) >3:
_holiday =sys.argv[2] #下划线开头，明确表示这个是局部变量
_type =sys.argv[3]
if not re.match(r'^\d{4}-\d{2}-\d{2}$',_holiday): # 检查月份的格式 如果不是我们希望的日期参数格式
print(_help)
return
if _type.lower() !='w' and _type.lower() !='h': # 让type不区分大小写
print(_help)
return
else:
print(_help)
return
# 将新增的日期放置文件的末尾 希望有个文件来存储我们节假日的信息
holiday_file ='data/holiday.txt'
with open(holiday_file,'a+') as hf : # 往文件的末尾去添加
hf.write('{}|{}|{}\n'.format(_holiday,datetime.datetime.strptime(_holiday,'%Y-%m-%d').strftime('%w'),_type)) # 往文件的最后一行追加 并且增加每行数据时添加换行符
print('增加新的假日：{}{}'.format(_holiday,_type))
def do_listholiday():
with open('data/holiday.txt','r') as hf: # 只读的方式打开文件
for line in hf.readlines():
print('日期:{} 星期 {}{}'.format(line.split('|')[0],line.split('|')[1],
'工作日' if line.split('|')[2].strip()=='w' else "休息日")) # 根据w或者其他的字符，写文字 工作日 or 休息日
programName=sys.argv[0]
if len(sys.argv) > 1:
command =sys.argv[1]
# print('command =%s'%command)
if command=='info':
do_info()
elif command=='addholiday':
do_addholiday()
elif command == 'listholiday':
do_listholiday()
else:
print('命令使用方法：')
print("\tpython %s COMMAND [PARAMS] " % programName)
print("\tpython %s info FILEPATH [YYYY-MM] " % programName)
print("\tpython %s addholiday YYYY-MM-DD TYPE(w:上班 h:休息)] " % programName)
print("\tpython %s listholiday " % programName)
- END -
本文为原创文章
❤：在这里跟我一起学习技术、职场、人生、原理、健身、摄影、生活等知识吧！
❤： 欢迎点个关注一起学习，进步充实人生。
982
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
