32 lines
1.4 KiB
Python
32 lines
1.4 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
# @Time : 2025/3/18 14:48
|
||
# @Author : AngesZhu
|
||
# @File : get_data_demo_service.py
|
||
# @Desc : 获取渠道模版数据
|
||
from utils.data_file_utils import ExcelHandler
|
||
from utils.logger_utils import logger
|
||
from utils.path_utils import PathOperator
|
||
|
||
|
||
def get_columns_from_channel(channel_name):
|
||
# 路径获取并获取文件信息
|
||
logger.info(f"获取渠道{channel_name} 对应的字段配置信息")
|
||
path_operator = PathOperator()
|
||
base_path = path_operator.get_parent_path(path_operator.get_full_path())
|
||
field_rules_path = path_operator.join_path(base_path, "test_data", "data_demo")
|
||
names = path_operator.get_all_files(field_rules_path)
|
||
names = [item for item in names if not item.startswith(".")] # 获取目录下所有模版列表
|
||
matches = list(filter(lambda item: channel_name in item, names))
|
||
if len(matches) > 1:
|
||
return False, "渠道模版不能存在两个以上"
|
||
excel_obj = ExcelHandler(path_operator.join_path(field_rules_path, matches[0]))
|
||
excel_data = excel_obj.read_excel(sheet_name=0, header=0, usecols=None)
|
||
logger.info(f"读取的excel模版表头：{excel_data.columns.tolist()}")
|
||
return True, {
|
||
"path": field_rules_path,
|
||
"name": matches[0],
|
||
"columns": excel_data.columns.tolist(),
|
||
"values": excel_data.values.tolist()
|
||
}
|