from model.model import db
from flask_restful import Api
from flask import Blueprint
from views.base import BaseView
import pandas as pd
import os
dialogue_bp = Blueprint("dialogue_bp", __name__, url_prefix="/dialogue")
api = Api(dialogue_bp)
# 前端展示对话文本
class MarkerView(BaseView):
# 读取所有的文件并把它存放到一张表里
@staticmethod
def read_all_excels():
# 设置文件夹下的路径
folder_path = "C:\\Users\\v_fanxiaoxi_dxm\\Desktop\\path"
file_names = []
# os.listdir的返回值是一个列表,列表里面存储该path下面的子目录的名称
for file_name in os.listdir(folder_path):
if file_name.endswith('.xlsx'):
# 拼接文件路径
file_path = os.path.join(folder_path, file_name)
# 把所有的文件路径存放到列表里
file_names.append(file_path)
data_frames = []
for file_path in file_names:
data_frame = pd.read_excel(file_path, header=0) # 根据需要设置参数
# 对数据进行处理,例如拼接、筛选等
data_frames.append(data_frame)
merged_data_frame = pd.concat(data_frames)
merged_data_frame.to_excel('C:\\Users\\v_fanxiaoxi_dxm\\Desktop\\path\\output.xlsx',
index=False)
def get(self):
output_path = "C:\\Users\\v_fanxiaoxi_dxm\\Desktop\\path\\output.xlsx"
# 如果文件路径不存在
if not os.path.isfile(output_path):
self.read_all_excels()
marker_data = pd.read_excel(output_path, header=0)
# 把数据json化
res = marker_data.to_json(orient="index", force_ascii=False)
res = eval(res)
res_list = list(res.values())
return self.success(data=res_list)
# 标注内容
def post(self):
...
api.add_resource(MarkerView, "/mark/")