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)  # 监听所有网络接口