op mysql

Author: HardworkFish

op_mysql.py

@@ -0,0 +1,39 @@
+import pymysql
+import csv
+import codecs
+# import sys
+# csv.field_size_limit(sys.maxsize)
+# 字段数限制控制
+csv.field_size_limit(500*1024*1024)
+
+
+def get_conn():
+    conn = pymysql.connect(host='localhost', port=3306, user='root', passwd='xxx', db='tyc', charset='utf8')
+    return conn
+
+
+def insert(cur, sql, args):
+    cur.execute(sql, args)
+
+
+def read_csv_to_mysql(filename):
+    with codecs.open(filename=filename, mode='r', encoding='utf-8') as f:
+        reader = csv.reader(f)
+        conn = get_conn()
+        cur = conn.cursor()
+        sql = 'insert into no_certificate values(%s,%s,%s,%s,%s,%s,%s,%s,%s,' \
+              '%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)'
+        for item in reader:
+            if item[1] is None or item[1] == '':  # item[1]作为唯一键，不能为null
+                continue
+            args = tuple(item)
+            print(args)
+            insert(cur, sql=sql, args=args)
+
+        conn.commit()
+        cur.close()
+        conn.close()
+
+
+if __name__ == '__main__':
+    read_csv_to_mysql('C:/ProgramData/MySQL/MySQL Server 5.7/Uploads/mysql_data/no_certificate.csv')