Win_home
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from flask import Flask, render_template, jsonify
|
|
import pandas as pd
|
|
import os
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
# 获取当前文件夹中所有包含"ofdata"字符的CSV文件
|
|
def get_csv_files():
|
|
files = {}
|
|
for filename in os.listdir():
|
|
if "ofdata" in filename and filename.endswith(".csv"):
|
|
files[filename] = os.path.join(os.getcwd(), filename)
|
|
return files
|
|
|
|
|
|
# 根据文件路径加载数据，只读取前12列
|
|
def load_data(file_path):
|
|
df = pd.read_csv(file_path, usecols=range(12)) # 只读取前12列
|
|
return df.tail(120).to_dict(orient="records")
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
files = get_csv_files() # 获取所有符合条件的文件
|
|
# 默认显示第一个文件的数据
|
|
first_file = list(files.keys())[0] if files else None
|
|
data = load_data(files[first_file]) if first_file else []
|
|
return render_template("index.html", data=data, files=files, file_name=first_file)
|
|
|
|
|
|
@app.route("/data/<file_name>")
|
|
def switch_data(file_name):
|
|
files = get_csv_files() # 获取所有符合条件的文件
|
|
if file_name in files:
|
|
data = load_data(files[file_name])
|
|
return jsonify(data)
|
|
return jsonify({"error": "File not found"}), 404
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host='0.0.0.0', port=5000, debug=True) # 监听所有网络接口
|