一、环境
windows10/mac + python3.6
python第三方库 xlsxwriter、PIL、argparse
二、需求
1、运行每条测试case成功与否都需要把截图放在img文件夹里；
2、把 （平台）_img 文件夹中的图片压缩写到small_img文件夹里；
3、根据图片命名规则，顺序写入所属case number对应行，顺序写入每条case所有截图；
4、根据平台来贴到excel最合适的位置；
5、最重要一点，是给媳妇写的，提升工作效率；
三、文件树示例
三、Paste_pictures.py
#!/usr/bin/env python3 # coding=utf-8 import xlsxwriter import datetime import os import logging LOGGER = logging.getLogger(__name__) def write_img_for_xls(file_name="test.xlsx", img_dir_name="img", sheet_name="案例截图", set_img_row=210.75, set_img_column=23.38, x_scale=0.14, y_scale=0.14, ): """ 读取同目录img文件夹下的所有图片，格式支持png\jpg\bmp。 图片必须遵从 1-1.png、1-2.png、2-1.png、2-2.png格式。 注意：是将图片写入新的excel文件，如果老的excel中有数据，将会替换所有数据。 file_name: 要写入的xlsx文件名 img_dir_name: 图片存放的目录，必须与脚本在同一目录下 set_img_row：设置行高 set_img_column：设置列宽 x_scale：图片宽度缩放比例 y_scale：图片高度缩放比例 :return: Nothing """ start_time = datetime.datetime.now() xls_path = os.path.join(os.getcwd(), file_name) if not os.path.isfile(xls_path): raise MyException("what?你居然不把{}文件跟脚本放在同一个目录下!".format(file_name)) img_path = os.path.join(os.getcwd(), img_dir_name) if not os.path.isdir(img_path): raise MyException("what?你都没有{}文件夹，我咋给你贴图啊~".format(img_dir_name)) all_img = os.listdir(img_path) if not all_img: raise MyException("忽悠我呢？{}文件夹下啥都没有~".format(img_dir_name)) w_book = xlsxwriter.Workbook(xls_path) img_sheet = w_book.add_worksheet(name=sheet_name) count_num = 0 try: for unit_img in all_img: try: img_num = unit_img.split("-") row = int(img_num[0]) column = int(img_num[1].split(".")[0]) suffix = (unit_img.split(".")[1]) if column == 0: LOGGER.warning("图片名称格式有误直接略过！错误文件名：{}，“-”前后数字必须从1开始！".format(unit_img)) continue except ValueError: LOGGER.warning("[-]图片命名规则有误直接略过！错误文件名是：{}".format(unit_img)) continue LOGGER.info(">> 正在贴图到第{}条用例，第{}列...".format(row, column)) img_sheet.set_column(firstcol=0, lastcol=0, width=8.38) img_sheet.write(row - 1, 0, "案例{}".format(row)) small_img = os.path.join(os.getcwd(), "{}/{}-{}.{}".format(img_dir_name, row, column, suffix)) img_sheet.set_column(firstcol=column, lastcol=column, width=set_img_column) img_sheet.set_row(row=row - 1, height=set_img_row) img_config = { x_scale: x_scale, y_scale: y_scale } result = img_sheet.insert_image(row - 1, column, small_img, img_config) img_sheet.write_url(row - 1, column + 1, url="https://my.oschina.net/medivhxu/blog/1590012") if not result: LOGGER.info("[+] 写入成功！") count_num += 1 else: LOGGER.error("[-] 写入失败!") except Exception as e: raise MyException("受到不明外力袭击，程序...挂了....\n{}".format(e)) finally: try: w_book.close() except PermissionError: raise MyException("你开着{}文件我让我咋写。。。赶紧关了!".format(file_name)) LOGGER.info("--------------贴图完成--------------") LOGGER.info("程序贴图数量：{}，贴图成功数量：{}，贴图失败数量：{}".format(len(all_img), count_num, len(all_img) - count_num)) class MyException(Exception): pass write_img_for_xls()
四、run
五、result
价值：
手动贴图需要半小时？1小时？贴错了？不，这些我都不要，仅需不到5秒，全部搞定，然后就可以干点想干的事~
性能分析：
其实贴图并不需要4秒+，因为xlsxwriter这个模块是自动创建cell的，但是这不是主要原因，主要原因是因为图片太大了，所以保存时间会随着图片加载到内存而线性增长（图片较大或过多，容易导致脚本直接崩溃），优化方式是选用openpyxl模块和最中要的图片预处理。
六、PIL图片压缩
1、code
2、再次运行Paste_pictures.py
可以明显看出，保存文件的时间有非常显著的提升。
七、模块化
把以上两个功能合并，增加平台类型，根据需求增加了3个平台的图片缩放比例和宽高，增加运行log，增加作为主程序命令行运行。以后如果扩展的话直接调用或者写个类，增加几个返回值就可以了。
#!/usr/bin/env python3 # coding=utf-8 import xlsxwriter import datetime import os import argparse import logging from PIL import Image LOGGER = logging.getLogger(__name__) PLATFORM = {"phone": { "x_scale": 0.29, "y_scale": 0.29, "width": 23.38, "height": 210.75 }, "pad": { 'x_scale': 0.2, "y_scale": 0.2, "width": 58, "height": 230 }, "oppo": { 'x_scale': 0.29, 'y_scale': 0.3, "width": 22.17, "height": 230 } } def _check_os_dir(read_dir_name, small_dir_name="small", xlsx_file_name="test.xlsx"): ''' 检测img文件夹及文件夹中的内容、xlsx文件是否存在 :param read_dir_name: 要读取图片的文件夹 :param small_dir_name: 压缩的图片文件夹 :param xlsx_file_name: excel文件名 :return: all_img：所有图片对象 xls_path：excel文件路径 img_path：图片文件路径 ''' full_name = read_dir_name + "_img" img_path = os.path.join(os.getcwd(), full_name) LOGGER.info(img_path) assert os.path.isdir(img_path), "what?你都没有{}文件夹，我咋给你贴图啊！！！".format(full_name) all_img = os.listdir(img_path) assert all_img, "{}文件夹里啥也没有，咋贴！！！".format(full_name) xls_path = os.path.join(os.getcwd(), xlsx_file_name) assert os.path.isfile(xls_path), "{}文件不存在！！！".format(xlsx_file_name) # small_full_name = small_dir_name + datetime.datetime.now().strftime("%Y%m%d%H%M%S") if full_name == small_dir_name: return all_img, xls_path, img_path # os.mkdir("{}".format(small_full_name)) # LOGGER.warning("新建文件夹{},压缩后的图片将存储在该文件夹中。".format(small_dir_name)) return all_img, xls_path, img_path def _compression_img(read_dir_name, small_dir_name="small", height_shrink_multiple=2, width_shrink_multiple=2): ''' 自动压缩指定文件夹下的所有图片 :param read_dir_name: 读取图片文件夹的名称 :param small_dir_name：如果压缩图片就读取该文件夹下的压缩图片 :param height_shrink_multiple: 设置图片压缩高度的倍数 :param width_shrink_multiple: 设置图片压缩宽度的倍数 :return: small_dir_name： 压缩后的图片文件名 ''' full_small_dir_name = small_dir_name + "_img" _check_os_dir(read_dir_name=read_dir_name, small_dir_name=full_small_dir_name) img_path = os.path.join(os.getcwd(), read_dir_name + "_img") all_img = os.listdir(img_path) for unit_img in all_img: try: img_num = unit_img.split("-") row = int(img_num[0]) column = int(img_num[1].split(".")[0]) suffix = (unit_img.split(".")[1]) if column == 0: LOGGER.warning("图片名称格式有误直接略过！错误文件名：{}，“-”前后数字必须从1开始！".format(unit_img)) continue except ValueError: LOGGER.warning("[-]图片命名规则有误直接略过！错误文件名是：{}".format(unit_img)) continue img_fp = os.path.join(img_path, unit_img) origin_img = Image.open(img_fp) w, h = origin_img.size small_img = origin_img.resize((int(w / height_shrink_multiple), int(h / width_shrink_multiple))) small_img.save(os.path.join(os.getcwd(), "{}/{}-{}.{}".format(full_small_dir_name, row, column, suffix))) LOGGER.info(">> 正在处理图像{}-{}.{},原像素高和宽{}，压缩后的高和宽{}".format(row, column, suffix, (w, h), small_img.size)) try: small_img.close() except Exception as e: LOGGER.debug("未知错误\n{}".format(e)) LOGGER.info("--------------图片压缩完成--------------") return small_dir_name def write_img_for_xls(platform, read_dir_name, sheet_name="案例截图", xlsx_file_name="test.xlsx"): """ 读取同目录img文件夹下的所有图片，格式支持png\jpg\bmp。 图片必须遵从 1-1.png、1-2.png、2-1.png、2-2.png格式。 注意：是将图片写入新的excel文件，如果老的excel中有数据，将会替换所有数据。 platform: 平台名称，包括phone、pad，web目前没实现 read_dir_name： 要读取图片的文件夹名称 xlsx_file_name: 要写入的xlsx文件名 sheet_name: 写入excel中sheet的名字 :return: nothing """ all_img, xls_path, img_path = _check_os_dir(xlsx_file_name=xlsx_file_name, read_dir_name=read_dir_name) w_book = xlsxwriter.Workbook(xls_path) img_sheet = w_book.add_worksheet(name=sheet_name) count_num = 0 try: for unit_img in all_img: try: img_num = unit_img.split("-") row = int(img_num[0]) column = int(img_num[1].split(".")[0]) suffix = (unit_img.split(".")[1]) if column == 0: LOGGER.warning("图片名称格式有误直接略过！错误文件名：{}，“-”前后数字必须从1开始！".format(unit_img)) continue except ValueError: LOGGER.warning("[-]图片命名规则有误直接略过！错误文件名是：{}".format(unit_img)) continue LOGGER.info(">> 正在贴图到第{}条用例，第{}列...".format(row, column)) img_sheet.set_column(firstcol=0, lastcol=0, width=8.38) img_sheet.write(row - 1, 0, "案例{}".format(row)) small_img = os.path.join(os.getcwd(), "{}/{}-{}.{}".format(read_dir_name+"_img", row, column, suffix)) img_sheet.set_column(firstcol=column, lastcol=column, width=PLATFORM.get(platform).get("width")) img_sheet.set_row(row=row - 1, height=PLATFORM.get(platform).get("height")) x_ = PLATFORM.get(platform).get("x_scale") y_ = PLATFORM.get(platform).get("y_scale") img_config = {"x_scale": x_, "y_scale": y_} result = img_sheet.insert_image(row - 1, column, small_img, img_config) img_sheet.write_url(row - 1, column + 1, url="https://my.oschina.net/medivhxu/blog/1590012") if not result: LOGGER.info("[+] 写入成功！") count_num += 1 else: LOGGER.error("[-] 写入失败!") except Exception as e: raise MyException("受到不明外力袭击，程序...挂了....\n{}".format(e)) finally: try: w_book.close() except PermissionError: raise MyException("你开着{}文件我让我咋写。。。赶紧关了!".format(xlsx_file_name)) LOGGER.info("--------------贴图完成--------------") LOGGER.info("程序贴图数量：{}，贴图成功数量：{}，贴图失败数量：{}".format(len(all_img), count_num, len(all_img) - count_num)) class MyException(Exception): pass if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument("-p", help="必须选择平台phone、pad、oppo") group = parser.add_mutually_exclusive_group() group.add_argument("-a", action="store_true", help="压缩图片且贴图到excel") group.add_argument("-w", action="store_true", help="直接贴图到excel") args = parser.parse_args() logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s', datefmt='%a, %d %b %Y %H:%M:%S', filename='Paste_pictures_{}.log'.format(datetime.datetime.now().strftime("%Y%m%d%H%M%S")), filemode='w') console = logging.StreamHandler() console.setLevel(logging.INFO) formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s') console.setFormatter(formatter) logging.getLogger('').addHandler(console) if args.p not in PLATFORM.keys(): raise MyException("参数错误，必须在{}中选择".format(PLATFORM.keys())) if args.a: LOGGER.info(">>> 选择参数-a,即压缩图片，又贴图。") r_small_dir_name = _compression_img(read_dir_name=args.p, small_dir_name="small") write_img_for_xls(platform=args.p, read_dir_name=r_small_dir_name) elif args.w: LOGGER.info(">>> 选择参数-w,只贴图。") write_img_for_xls(platform=args.p, read_dir_name=args.p) else: LOGGER.error("参数错误")
windows command和linux/mac terminal下运行效果：
The end~
扫一扫
3220
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
