update ai_lambda-exec.py.

ai_lambda-exec.py

@@ -1,24 +1,24 @@
-import json
 import pymysql
-import sys
-import time
 import boto3
 
 def updata_db(m_key,m_value):
-    conn2 = pymysql.connect(
-        host='a.xdc.ink',
-        port=3306,
-        user='root',
-        passwd='123c13c12',
-        db='cmd',
-        charset='utf8mb4')
+	db = pymysql.connect(
+	        host='127.0.0.1',
+	        port=3306,
+	        user='root',
+	        passwd='123***123***',
+	        db='cmd',
+	        charset='utf8mb4')
+	cursor = db.cursor()
+	sql = f"INSERT INTO kv(mykey,myvalue) VALUES('{m_key}','{m_value}')"
+	try:
+		cursor.execute(sql)
+		db.commit()
+	except Exception as e:
+		print('ERROR : '+e)
+		db.rollback()
+	db.close()
 
-    cs2 = conn2.cursor()
-    select_sql = f'INSERT INTO kv (mykey, myvalue) VALUES ("{m_key}", "{m_value}");'
-    cs2.execute(select_sql)
-    conn2.commit()
-    cs2.close()
-    conn2.close()
 
 def lambda_handler(event, context):
     # TODO implement
@@ -29,3 +29,4 @@ def lambda_handler(event, context):
     updata_db(m_key,m_value)
 
     
+