Python3之pymysql导入mysql

转载于 2017-08-08 19:09:00 发布 · 184 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zhzhang/p/7308494.html

文章标签：

#python #数据库

本文介绍了一个使用Python编写的脚本，该脚本能够从文件中读取数据，并批量将这些数据插入到MySQL数据库中。脚本通过pymysql模块连接数据库，并采用executemany方法提高插入效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

$cat insert.py
#!/usr/bin/python
# -*- coding: UTF-8 -*-

import os
import sys
import datetime
import pymysql


def read_insert(filename, slot_type_id):
    file = open(filename)
    i = 0
    j = 0
    list = []

    db = pymysql.connect(host="test", user="test", passwd="test", port=3306, db="test", charset='utf8')
    cursor = db.cursor()

    sql = "insert into test(slot_type_id, type, value, gmt_modify, gmt_create) values (%s, %s, %s, %s, %s)"

    for line in file:
        data = (slot_type_id, "const", (line.replace("\n","")),
                datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
        list.append(data)

        if i > 5000:
            cursor.executemany(sql, list)
            db.commit()
            print("插入", j)
            i = 0
            list = []
        i = i + 1
        j = j + 1

    if i > 0:
        cursor.executemany(sql, list)
        db.commit()
    db.close()
    file.close()
    print("ok")


if __name__ == "__main__":
    read_insert(sys.argv[1], sys.argv[2])