addproject
This commit is contained in:
qiaoxinjiu
272
base_framework/public_tools/excel_api.py
Normal file
272
base_framework/public_tools/excel_api.py
Normal file
@@ -0,0 +1,272 @@
|
||||
# -*- coding: UTF-8 -*-
|
||||
# @File: operation_Xlsx.py
|
||||
# @Description: excle的基本操作
|
||||
# @Author: WenQing
|
||||
# @Date: 2021-08-26 11:15:08
|
||||
import os
|
||||
import xlrd, xlwt
|
||||
import pandas as pd
|
||||
from xlutils.copy import copy
|
||||
import openpyxl
|
||||
from base_framework.base_config.current_pth import env_choose_path
|
||||
from base_framework.public_tools.read_config import ReadConfig
|
||||
|
||||
|
||||
class ExcelApi:
|
||||
def __init__(self):
|
||||
self.p_here = os.path.dirname(os.path.abspath(__file__))
|
||||
self.c_team = ReadConfig(env_choose_path).get_value(sections='run_evn_name', options='current_team')
|
||||
self.up_file_path = os.path.abspath(os.path.join(self.p_here, '../../{}/library/UpFile/'.format(self.c_team)))
|
||||
|
||||
def excel_create_excel_file(self, file_name, file_dict_data=None, file_path=None):
|
||||
"""
|
||||
功能：按列将数据写入excel文件，没有文件时就新建文件
|
||||
Args:
|
||||
file_name: 文件名
|
||||
file_dict_data: 文件内容，字典类型，格式：{'title1':[], 'title2':[],..... 'titleN':[],}
|
||||
file_path: 文件路径，没有时默认放在小组目录下的UpFile文件夹中
|
||||
Returns:
|
||||
"""
|
||||
if not file_path:
|
||||
full_path = self.up_file_path + os.sep + file_name
|
||||
else:
|
||||
full_path = file_path + os.sep + file_name
|
||||
if not file_dict_data:
|
||||
file_dict_data = {}
|
||||
df = pd.DataFrame(file_dict_data)
|
||||
# 将 DataFrame 的数据保存到 Excel 文件中
|
||||
df.to_excel(full_path, index=False)
|
||||
|
||||
def excel_create_file_by_column(self, file_name, file_dict_data=None, file_path=None):
|
||||
"""
|
||||
功能：按列将数据写入excel文件，没有文件时就新建文件
|
||||
Args:
|
||||
file_name: 文件名
|
||||
file_dict_data: 文件内容，字典类型，格式：{'title1':[], 'title2':[],..... 'titleN':[],}
|
||||
file_path: 文件路径，没有时默认放在小组目录下的UpFile文件夹中
|
||||
Returns:
|
||||
"""
|
||||
if not file_path:
|
||||
full_path = self.up_file_path + os.sep + file_name
|
||||
else:
|
||||
full_path = file_path + os.sep + file_name
|
||||
if not file_dict_data:
|
||||
file_dict_data = {}
|
||||
df = pd.DataFrame(file_dict_data)
|
||||
# 将 DataFrame 的数据保存到 Excel 文件中
|
||||
df.to_excel(full_path, index=False)
|
||||
|
||||
def excel_write_file_by_line(self, file_name, line_data=None, file_path=None, w_type='new'):
|
||||
"""
|
||||
功能：按行将数据写入excel文件，没有文件时就新建文件
|
||||
Args:
|
||||
file_name: 文件名
|
||||
line_data: 文件内容，列表类型，格式：[[字段1,字段2],[字段1,字段2]]
|
||||
file_path: 文件路径，没有时默认放在小组目录下的UpFile文件夹中
|
||||
w_type: 写入类型：new-重写，append-追加
|
||||
"""
|
||||
if not file_path:
|
||||
full_path = self.up_file_path + os.sep + file_name
|
||||
else:
|
||||
full_path = file_path + os.sep + file_name
|
||||
if w_type == 'new':
|
||||
self.write_xlsx(path=full_path, sheet_name="Sheet1", value=line_data)
|
||||
elif w_type == 'append':
|
||||
self.append_xlsx(path=full_path, value=line_data)
|
||||
else:
|
||||
raise Exception("目前仅支持new和append两种模式，而当前传入的是；{}".format(w_type))
|
||||
|
||||
def excel_replace_cell_value(self, file_name, replace_dict, sheet_name=None, skip_first_row=True):
|
||||
"""
|
||||
功能：替换excel文件中的指定单元格的值
|
||||
Args:
|
||||
file_name: 文件名,如果文件不在UpFile文件夹中，需要传入完整路径
|
||||
sheet_name: 工作表名称,默认为全部工作表
|
||||
replace_dict: 替换的数据，字典类型，格式：{'old_txt_1': 'new_txt_1', 'old_txt_2': 'new_txt_2',..... }
|
||||
skip_first_row: 是否跳过第一行，默认跳过,用于第一行是标题的情况
|
||||
Returns:
|
||||
"""
|
||||
if "/" not in file_name: # 如果没有路径，就默认在UpFile文件夹中
|
||||
file_name = self.up_file_path + os.sep + file_name
|
||||
# 读取指定工作表的数据
|
||||
wb = openpyxl.load_workbook(file_name)
|
||||
# 遍历所有要替换的数据
|
||||
for old_text, new_text in replace_dict.items():
|
||||
# 遍历所有工作表
|
||||
for sheet in wb.worksheets:
|
||||
# 遍历工作表中的所有行和列
|
||||
for row in sheet.iter_rows():
|
||||
if skip_first_row: # 跳过第一行
|
||||
skip_first_row = False
|
||||
continue
|
||||
for cell in row:
|
||||
if cell.value == old_text:
|
||||
cell.value = new_text
|
||||
wb.save(file_name)
|
||||
|
||||
def excel_read_columns(self, file_name, sheet_name, column_names):
|
||||
"""
|
||||
功能：按列读取excel文件的数据
|
||||
Args:
|
||||
file_name: 文件名，如果文件不在UpFile文件夹中，需要传入完整路径
|
||||
sheet_name: 工作表名称
|
||||
column_names: 列名列表，格式：['title1', 'title2',..... 'titleN']
|
||||
Returns:
|
||||
返回指定列的数据，格式为：[[字段1,字段2],[字段1,字段2]]
|
||||
"""
|
||||
if "/" not in file_name: # 如果没有路径，就默认在UpFile文件夹中
|
||||
file_name = self.up_file_path + os.sep + file_name
|
||||
# 读取指定工作表的数据
|
||||
df = pd.read_excel(file_name, sheet_name=sheet_name)
|
||||
|
||||
# 检查指定列是否存在
|
||||
missing_columns = [col for col in column_names if col not in df.columns]
|
||||
if missing_columns:
|
||||
raise ValueError(f"Columns {missing_columns} do not exist in the sheet '{sheet_name}'.")
|
||||
|
||||
# 返回指定列的数据
|
||||
return df[column_names].values.tolist()
|
||||
|
||||
def read_asDict(self, path):
|
||||
"""
|
||||
| 功能说明： | 读取excle,输出字典格式|
|
||||
| 传入参数： | 读取文件路径 |
|
||||
| 返回数据： | 表头字段作为key,单元格值作为value |
|
||||
| 作者信息： | 作者 文青 | 修改时间 |
|
||||
举例说明：
|
||||
"""
|
||||
self.table = pd.read_excel(path)
|
||||
data = []
|
||||
for i in self.table.index.values:
|
||||
data_dict = self.table.loc[i].to_dict()
|
||||
data.append(data_dict)
|
||||
return data
|
||||
|
||||
def read_asList(self, path, sheetname=None):
|
||||
"""
|
||||
| 功能说明： | 读取excle和sheetname,输出列表格式|
|
||||
| 传入参数： | 读取excle和sheetname |
|
||||
| 返回数据： | 列表[] |
|
||||
| 作者信息： | 作者 文青 | 修改时间 |
|
||||
举例说明：
|
||||
"""
|
||||
wb = openpyxl.load_workbook(path)
|
||||
if not sheetname:
|
||||
sheets = wb.sheetnames
|
||||
sheetname = sheets[0]
|
||||
ws = wb[sheetname]
|
||||
rows = ws.rows
|
||||
columns = ws.columns
|
||||
data = []
|
||||
for row in rows:
|
||||
line = [col.value for col in row]
|
||||
data.append(line)
|
||||
return data
|
||||
|
||||
def read_xls_txt(self, path):
|
||||
"""
|
||||
| 功能说明： | 读取excle的第一列放在一个列表中|
|
||||
| 传入参数： | 读取excle路径 |
|
||||
| 返回数据： | 列表[] |
|
||||
| 作者信息： | 作者 文青 | 修改时间 |
|
||||
举例说明：
|
||||
"""
|
||||
data = pd.read_excel(path, header=None)
|
||||
data_list = []
|
||||
nrows = data.shape[0]
|
||||
for irow in range(nrows):
|
||||
data_list.append(data.iloc[irow, 0])
|
||||
string = ''
|
||||
for i in range(len(data_list)):
|
||||
string += data_list[i] + ','
|
||||
return string
|
||||
|
||||
def write_xls(self, value, sheetname, path):
|
||||
"""
|
||||
| 功能说明： | 创建一个xls的文件并且写入数据 |
|
||||
| 传入参数： | value:传入列表,格式:[[字段1,字段2],[字段1,字段2],[字段1,字段2],[字段1,字段2]] |
|
||||
|sheetname:sheet名称 |
|
||||
|path:写入文件路径 |
|
||||
| 返回数据： | |
|
||||
| 作者信息： | 作者 文青 | 修改时间 |
|
||||
举例说明：
|
||||
"""
|
||||
index = len(value) # 获取需要写入数据的行数
|
||||
workbook = xlwt.Workbook() # 新建一个工作簿
|
||||
sheet = workbook.add_sheet(sheetname=sheetname) # 在工作簿中新建一个sheetname的表格
|
||||
for i in range(0, index):
|
||||
for j in range(0, len(value[i])):
|
||||
sheet.write(i, j, value[i][j])
|
||||
workbook.save(path)
|
||||
# return "写入成功！"
|
||||
|
||||
def append_xls(self, value, path):
|
||||
"""
|
||||
| 功能说明： | 对xls文件第一个sheet内容进行追加 |
|
||||
| 传入参数： | value:传入列表,格式:[[字段1,字段2],[字段1,字段2],[字段1,字段2],[字段1,字段2]] |
|
||||
|path:写入文件路径 |
|
||||
| 返回数据： | |
|
||||
| 作者信息： | 作者 文青 | 修改时间 |
|
||||
举例说明：
|
||||
"""
|
||||
index = len(value) # 获取需要写入数据的行数
|
||||
workbook = xlrd.open_workbook(path) # 打开工作簿
|
||||
sheets = workbook.sheet_names() # 获取所有表格
|
||||
worksheet = workbook.sheet_by_name(sheets[0]) # 获取所有表格中的第一个表格
|
||||
rows_old = worksheet.nrows # 获取表格的总行数
|
||||
new_workbook = copy(workbook) # 把原文件复制一份
|
||||
new_worksheet = new_workbook.get_sheet(0) # 获取副本第一个表格
|
||||
for i in range(0, index):
|
||||
for j in range(0, len(value[i])):
|
||||
new_worksheet.write(i + rows_old, j, value[i][j])
|
||||
new_workbook.save(path)
|
||||
# return "追加成功！！"
|
||||
|
||||
def write_xlsx(self, path, sheet_name, value):
|
||||
"""
|
||||
| 功能说明： | 创建一个xls的文件并且写入数据 |
|
||||
| 传入参数： | value:传入列表,格式:[[字段1,字段2],[字段1,字段2],[字段1,字段2],[字段1,字段2]] |
|
||||
|sheetname:sheet名称 |
|
||||
|path:写入文件路径 |
|
||||
| 返回数据： | |
|
||||
| 作者信息： | 作者 文青 | 修改时间 |
|
||||
举例说明：
|
||||
"""
|
||||
index = len(value)
|
||||
workbook = openpyxl.Workbook()
|
||||
sheet = workbook.active
|
||||
sheet.title = sheet_name
|
||||
for i in range(0, index):
|
||||
for j in range(0, len(value[i])):
|
||||
sheet.cell(row=i + 1, column=j + 1, value=str(value[i][j]))
|
||||
workbook.save(path)
|
||||
return "写入成功！！"
|
||||
|
||||
def append_xlsx(self, path, value):
|
||||
"""
|
||||
| 功能说明： | 对xls文件第一个sheet内容进行追加 |
|
||||
| 传入参数： | value:传入列表,格式:[[字段1,字段2],[字段1,字段2],[字段1,字段2],[字段1,字段2]] |
|
||||
|path:写入文件路径 |
|
||||
| 返回数据： | |
|
||||
| 作者信息： | 作者 文青 | 修改时间 |
|
||||
举例说明：
|
||||
"""
|
||||
data = openpyxl.load_workbook(path)
|
||||
sheets = data.sheetnames
|
||||
table = data[sheets[0]]
|
||||
for i in value:
|
||||
table.append(i)
|
||||
data.save(path)
|
||||
return "追加成功！！"
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
ea = ExcelApi()
|
||||
f_data = {"教师ID": [1, 2, 3, 4],
|
||||
"教师姓名": [11, 22, 33, 44],
|
||||
"课程ID": [111, 222, 333, 444],
|
||||
"课程名称": [1111, 2222, 3333, 4444],
|
||||
"更新类型（1.新增；0.删除，填写数字）": [0, 1, 0, 1]
|
||||
}
|
||||
ea.excel_create_excel_file(file_name="测试大盘A标签.xlsx", file_dict_data=f_data)
|
||||
|
||||
Reference in New Issue
Block a user