smartpush 1.0.4__py3-none-any.whl → 1.0.5__py3-none-any.whl

smartpush/export/basic/ExcelExportChecker.py

@@ -1,6 +1,9 @@
+import contextlib
 import os
 import re
 from io import BytesIO
+from urllib.parse import unquote
+
 import pandas as pd
 import numpy as np
 import warnings
@@ -10,7 +13,8 @@ from requests import request
 """
 warnings.simplefilter("ignore")
 
-def read_excel_from_oss(method="get", url=""):
+
+def read_excel_from_oss(url="", method="get"):
     """读取oss的excel内容并写入到本地csv"""
     try:
         result = request(method=method, url=url)
@@ -47,6 +51,35 @@ def read_excel_and_write_to_list(excel_data=None, file_name=None):
     """excel内容并写入到内存list中
     :param excel_data：excel的io对象, 参数和file_name互斥
     :file_name: excel文件名称，目前读取check_file目录下文件，参数和excel_data互斥
+
+    io：可以是文件路径、文件对象或 ExcelFile 对象，代表要读取的 Excel 文件。
+    sheet_name：指定要读取的工作表，默认为第一个工作表（索引为 0）。
+    header：指定哪一行作为列名，默认为第一行（索引为 0）。
+    names：可以为列提供自定义名称，如果设置了这个，会覆盖文件中的列名。
+    index_col：可以指定某一列或多列作为索引。
+    usecols：可以指定要读取的列，可以是列的索引、列名或一个筛选函数。
+    dtype：可以指定数据类型，控制数据的类型转换。
+    engine：指定使用的 Excel 引擎，比如 xlrd、openpyxl 等。
+    converters：可以为不同的列指定自定义的转换函数，以字典形式存储。
+    true_values 和 false_values：定义哪些值会被视为 True 或 False。
+    skiprows：可以指定要跳过的行，可以是一个整数序列、一个整数或一个函数。
+    nrows：可以指定要读取的行数。
+    na_values：可以指定哪些值会被视为 NaN。
+    keep_default_na：决定是否使用默认的 NaN 值。
+    na_filter：决定是否过滤 NaN 值。
+    verbose：决定是否输出详细信息。
+    parse_dates：决定是否解析日期，可以是一个列表、字典或布尔值。
+    date_parser：自定义的日期解析函数。
+    date_format：日期的格式设置。
+    thousands：千位分隔符。
+    decimal：小数点分隔符。
+    comment：注释字符，以该字符开头的行将被跳过。
+    skipfooter：指定要跳过的文件末尾的行数。
+    storage_options：存储选项。
+    dtype_backend：数据类型后端。
+    engine_kwargs：传递给引擎的额外参数。
+
+
     """
     try:
         if excel_data is not None and file_name is not None:
@@ -75,14 +108,62 @@ def read_excel_and_write_to_csv(excel_data, file_name):
         print(f"excel写入csv时出错：{e}")
 
 
-def check_excel(actual, expected):
-    """对比两份excel内容
-    :param: actual: 实际值，list类型
-    :param: expected: 预期值，list类型
+def check_excel(check_type="content", **kwargs):
+    """对比excel
+    :param: type: 需要对比类型，
+            枚举：content：对比两表格内容
+                方式1：传参actual_oss和expected_oss，参数类型str,url
+                放松1：传参actual和expected，参数类型list or dict
+            excelName: 对比两表格文件名称
+            all: 对比所有内容
     """
+    try:
+        if check_type == "content":
+            if "actual" in kwargs.keys() and "expected" in kwargs.keys():
+                return check_excel_content(actual=kwargs["actual"], expected=kwargs["expected"])
+            else:
+                return check_excel_content(actual=read_excel_and_write_to_list(excel_data=read_excel_from_oss(url=kwargs["actual_oss"])),
+                                           expected=read_excel_and_write_to_list(excel_data=read_excel_from_oss(url=kwargs["expected_oss"]))
+                                           )
+        elif check_type == "excelName":
+            return check_excel_name(actual_oss=kwargs["actual_oss"], expected_oss=kwargs["expected_oss"])
+        elif check_type == "all":
+            actual_content = read_excel_and_write_to_list(excel_data=read_excel_from_oss(url=kwargs["actual_oss"]))
+            expected_content = read_excel_and_write_to_list(excel_data=read_excel_from_oss(url=kwargs["expected_oss"]))
+            flag1, content_result = check_excel_content(actual=actual_content, expected=expected_content)
+            flag2, name_result = check_excel_name(actual_oss=kwargs["actual_oss"], expected_oss=kwargs["expected_oss"])
+            return flag1 and flag2, {"文件名称": name_result, "导出内容": content_result}
+        else:
+            return False, f"不支持此类型: {check_type}"
+    except Exception as e:
+        print(f"对比excel异常：{e}")
+        return False, [e]
+
+
+def check_excel_name(actual_oss, expected_oss):
+    """校验excel文件名称
+    :param actual_oss:实际oss链接
+    :param actual_oss:预期oss链接
+    """
+    try:
+        actual_name = unquote(actual_oss.split("/")[-1])
+        expected_name = unquote(expected_oss.split("/")[-1])
+        if actual_name == expected_name:
+            return True, "excel文件名称-匹配"
+        else:
+            return False, f"excel文件名称-不匹配, 实际: {actual_name}, 预期: {expected_name}"
+    except BaseException as msg:
+        return False, f"excel文件名称-服务异常: {msg}"
+
+
+def check_excel_content(actual, expected):
+    """校验excel内容
+       :param actual: 实际内容，list或dict类型
+       :param expected:预期内容：list或dict类型
+     """
     try:
         if actual == expected:
-            return True, ["完全匹配"]
+            return True, ["excel内容-完全匹配"]
         else:
             errors = []
             # 断言1：校验行数
@@ -90,10 +171,10 @@ def check_excel(actual, expected):
             expected_num = len(expected)
             check_row = actual_num - expected_num
             if check_row == 0:
-                errors.append("预期和实际行数相等，为" + str(actual_num) + "行")
+                errors.append("excel内容-预期和实际行数相等，为" + str(actual_num) + "行")
             else:
                 errors.append(
-                    "行数和预期对比差" + check_row.__str__() + "行" + ", 实际:" + str(actual_num) + "预期: " + str(
+                    "excel内容-行数和预期对比差" + check_row.__str__() + "行" + ", 实际:" + str(actual_num) + "预期: " + str(
                         expected_num))
             # 断言不匹配行
             if check_row >= 0:
@@ -105,10 +186,10 @@ def check_excel(actual, expected):
                     continue
                 else:
                     errors.append(
-                        "第" + str(i + 1) + "行不匹配，预期为：" + str(expected[i]) + ", 实际为: " + str(actual[i]))
+                        "excel内容-第" + str(i + 1) + "行不匹配，预期为：" + str(expected[i]) + ", 实际为: " + str(actual[i]))
             return False, errors
     except Exception as e:
-        print(f"对比excel：{e}")
+        print(f"excel内容-服务异常：{e}")
         return False, [e]
 
 

smartpush/felix_test.py

@@ -4,8 +4,10 @@ from smartpush.export.basic import ExcelExportChecker
 
 if __name__ == '__main__':
 
-    ossurl = "https://sl-smartfile.oss-ap-southeast-1.aliyuncs.com/material_ec2/2025-01-16/8556e34c8d4d45f0bb2d42dc8871a90b/%E8%A1%A8%E5%8D%95%E4%BB%BB%E5%8A%A1%E6%95%B0%E6%8D%AE%E6%A6%82%E8%A7%88.xlsx"
-    excelOss = ExcelExportChecker.read_excel_from_oss(url=ossurl)
-    excelList = ExcelExportChecker.read_excel_and_write_to_list(excel_data=excelOss)
-    print(excelList)
+    ossurl1 = "https://sl-smartfile.oss-ap-southeast-1.aliyuncs.com/material_ec2/2025-01-16/8556e34c8d4d45f0bb2d42dc8871a90b/%E8%A1%A8%E5%8D%95%E4%BB%BB%E5%8A%A1%E6%95%B0%E6%8D%AE%E6%A6%82%E8%A7%88.xlsx"
+    ossurl2 = "https://sl-smartfile.oss-ap-southeast-1.aliyuncs.com/material_ec2/2025-01-16/8556e34c8d4d45f0bb2d42dc8871a90b/%E8%A1%A8%E5%8D%95%E4%BB%BB%E5%8A%A1%E6%95%B0%E6%8D%AE%E6%A6%82%E8%A7%88.xlsx"
 
+    a = ExcelExportChecker.read_excel_and_write_to_list(excel_data=ExcelExportChecker.read_excel_from_oss(url=ossurl1))
+    b = ExcelExportChecker.read_excel_and_write_to_list(excel_data=ExcelExportChecker.read_excel_from_oss(url=ossurl2))
+    print(ExcelExportChecker.check_excel(actual=a, expected=b))
+    # print(ExcelExportChecker.check_excel_name(actual_oss=ossurl1, expected_oss=ossurl2))

{smartpush-1.0.4.dist-info → smartpush-1.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: smartpush
-Version: 1.0.4
+Version: 1.0.5
 Summary: 用于smartpush自动化测试工具包
 Author: 卢泽彬、邵宇飞、周彦龙
 

smartpush-1.0.5.dist-info/RECORD

@@ -0,0 +1,10 @@
+smartpush/__init__.py,sha256=XJrl1vhGATHSeSVqKmPXxYqxyseriUpvY5tLIXir3EE,24
+smartpush/felix_test.py,sha256=Lx5APlgJQ1bUY-632rE-BMW494IbXnUSWZjMj0GZgO4,904
+smartpush/get_jira_info.py,sha256=dmCwkKa94xwyE2hegE1KBI3cV_LbrJ67P9osORUGPt4,2633
+smartpush/export/__init__.py,sha256=D9GbWcmwnetEndFDty5XbVienFK1WjqV2yYcQp3CM84,99
+smartpush/export/basic/ExcelExportChecker.py,sha256=ZGBIB9EW_x6EcKt9M2Q9CXj3vteJW63HqnFF7a9pOSY,9201
+smartpush/export/basic/__init__.py,sha256=6tcrS-2NSlsJo-UwEsnGUmwCf7jgOsh_UEbM0FD-gYE,70
+smartpush-1.0.5.dist-info/METADATA,sha256=qrXsMdpp3mo7roF1p2NcErmSAEMJZ2PUChNXZuYode8,145
+smartpush-1.0.5.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+smartpush-1.0.5.dist-info/top_level.txt,sha256=5_CXqu08EfbPaKLjuSAOAqCmGU6shiatwDU_ViBGCmg,10
+smartpush-1.0.5.dist-info/RECORD,,

smartpush-1.0.4.dist-info/RECORD

@@ -1,10 +0,0 @@
-smartpush/__init__.py,sha256=XJrl1vhGATHSeSVqKmPXxYqxyseriUpvY5tLIXir3EE,24
-smartpush/felix_test.py,sha256=I7DkMnqRUa48ALyawoJtPh74zRR86OO3oRnkBi9qJRQ,476
-smartpush/get_jira_info.py,sha256=dmCwkKa94xwyE2hegE1KBI3cV_LbrJ67P9osORUGPt4,2633
-smartpush/export/__init__.py,sha256=D9GbWcmwnetEndFDty5XbVienFK1WjqV2yYcQp3CM84,99
-smartpush/export/basic/ExcelExportChecker.py,sha256=C1piYqeZI-YGWqN3FsYYmVL0JF_y6xxajkHGBf8i-iA,4825
-smartpush/export/basic/__init__.py,sha256=6tcrS-2NSlsJo-UwEsnGUmwCf7jgOsh_UEbM0FD-gYE,70
-smartpush-1.0.4.dist-info/METADATA,sha256=wYH3eiXClYapcPf0HFIiY75Lt3rikG9xXeassEdqGJ0,145
-smartpush-1.0.4.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-smartpush-1.0.4.dist-info/top_level.txt,sha256=5_CXqu08EfbPaKLjuSAOAqCmGU6shiatwDU_ViBGCmg,10
-smartpush-1.0.4.dist-info/RECORD,,