上传文件至 /

2024-09-08 10:21:55 +08:00
parent 6eedc51720
commit bf7d43f21c
1 changed files with 103 additions and 0 deletions
--- a/data_ana.py
+++ b/data_ana.py
@@ -0,0 +1,103 @@
+import os
+import pandas as pd
+import mysql.connector
+from flask import Flask, request, redirect, url_for, render_template
+from datetime import datetime
+
+app = Flask(__name__)
+UPLOAD_FOLDER = 'uploads'
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+
+# 创建上传文件夹
+if not os.path.exists(UPLOAD_FOLDER):
+    os.makedirs(UPLOAD_FOLDER)
+
+
+def process_excel(file_path):
+    # 读取excel文件
+    df = pd.read_excel(file_path)
+
+    # 统计数据
+    summary_row = df[df.iloc[:, 0].str.contains('汇总', na=False)]
+    print("汇总行的数据:")
+    print(summary_row)
+    #获取指定数据
+    C = summary_row.iloc[0, 18] 
+
+    return C
+
+def write_to_mysql(C):
+    # 连接MySQL数据库
+    conn = mysql.connector.connect(
+        host='change-to-your-host',
+        user='your-username',
+        password='your-password',
+        database='your-table'
+    )
+    cursor = conn.cursor()
+
+    # 查询当前最大id
+    cursor.execute('SELECT MAX(id) FROM `all-mt`')
+    max_id = cursor.fetchone()[0]
+
+    # 计算新记录的id值
+    new_id = (max_id + 1) if max_id is not None else 1
+
+    #获取当前时间
+    current_date = datetime.now().date()
+
+    #数据类型转换
+    C=int(C)
+
+    # 插入统计结果到数据库
+    sql = f"INSERT INTO `1` (id, update_date, C) VALUES (%s, %s, %s)"
+    values = (new_id, current_date, C)
+
+    print("Executing SQL:", sql)
+    print("With values:", values)
+    cursor.execute(sql, values)
+
+    # 提交事务
+    conn.commit()
+
+    # 关闭连接
+    cursor.close()
+    conn.close()
+
+
+@app.route('/', methods=['GET', 'POST'])
+def upload_file():
+    if request.method == 'POST':
+        # 检查是否选择了文件
+        if 'file' not in request.files:
+            return 'No file part'
+        file = request.files['file']
+        # 检查是否上传了文件
+        if file.filename == '':
+            return 'No selected file'
+        
+        if file:
+            file_path = os.path.join(app.config['UPLOAD_FOLDER'], file.filename)
+            file.save(file_path)
+
+            # 处理Excel文件
+            C = process_excel(file_path)
+
+        # 写入MySQL数据库
+        write_to_mysql(C)
+
+        return f"数据已成功写入MySQL数据库"
+
+    return '''
+    <!doctype html>
+    <title>数据分析系统</title>
+    <h1>数据上传分析系统</h1>
+    <form method=post enctype=multipart/form-data>
+      选择文件: <input type=file name=file><br><br>
+      <input type=submit value=Upload>
+    </form>
+    '''
+
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=6001)