From 3dc70a2756d42185ce679a5c407d7586ece51242 Mon Sep 17 00:00:00 2001
From: wangweiye7840 <kj_wangweiye@126.com>
Date: Mon, 20 May 2024 14:55:12 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A4=87=E6=B3=A8=E4=BF=A1=E6=81=AF=E6=94=B6?=
 =?UTF-8?q?=E9=9B=86=20=E5=A2=9E=E5=8A=A0=E9=81=BF=E5=85=8D=E9=87=8D?=
 =?UTF-8?q?=E5=A4=8D=E7=9A=84=E6=A0=A1=E9=AA=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 工具v4/备注信息收集.py | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/工具v4/备注信息收集.py b/工具v4/备注信息收集.py
index 9ffa54b2..941475bb 100644
--- a/工具v4/备注信息收集.py
+++ b/工具v4/备注信息收集.py
@@ -69,19 +69,28 @@ def submit_remarks():
     # mydb = connect(hostname = db_host, port = db_port, username=db_user, pwd=db_pwd, db = "tikutest")
     mydb = connect(hostname = db_host, port = db_port, username=db_user, pwd=db_pwd, db = "tiku")
     mycursor = mydb.cursor()
-    sql = f"INSERT INTO remarks SET ID = %s, date = %s, remark_content = %s;"
-    val = (id,date,f"({category}){remarks}")
+    sql = f"SELECT * FROM remarks WHERE ID = %s and remark_content = %s;"
+    val = (id,f"({category}){remarks}")
+    print(f"({category}){remarks}")
     mycursor.execute(sql,val)
-    ip_address = request.headers.get('X-Forwarded-For', request.remote_addr)  
-    # 如果 X-Forwarded-For 头存在多个 IP 地址，则取第一个  
-    if ',' in ip_address:  
-        ip_address = ip_address.split(',')[0]  
-    sql = "INSERT INTO logs (DATE,TIME,username,ID,action,db_content) VALUE (%s,%s,%s,%s,%s,%s);"
-    val = (GetDate(),GetTime(),f"{username} at {ip_address}",id,f"网页新增备注",f"({category}){remarks}")
-    mycursor.execute(sql,val)
-    mydb.commit()
-    mydb.close()
-    return render_template('show_remarks.html', category=category, remarks=remarks)
+    ret = mycursor.fetchall()
+    if len(ret) == 0:
+        sql = f"INSERT INTO remarks SET ID = %s, date = %s, remark_content = %s;"
+        val = (id,date,f"({category}){remarks}")
+        mycursor.execute(sql,val)
+        ip_address = request.headers.get('X-Forwarded-For', request.remote_addr)  
+        # 如果 X-Forwarded-For 头存在多个 IP 地址，则取第一个  
+        if ',' in ip_address:  
+            ip_address = ip_address.split(',')[0]  
+        sql = "INSERT INTO logs (DATE,TIME,username,ID,action,db_content) VALUE (%s,%s,%s,%s,%s,%s);"
+        val = (GetDate(),GetTime(),f"{username} at {ip_address}",id,f"网页新增备注",f"({category}){remarks}")
+        mycursor.execute(sql,val)
+        mydb.commit()
+        mydb.close()
+        return render_template('show_remarks.html', category=category, remarks=remarks)
+    else:
+        mydb.close()
+        return f"已有备注 ({category}){remarks}"
 
 
 if __name__ == '__main__':