Python连接数据库并批量插入包含日期记录的操作

前言

平台：

• windows 10.0
• python 3.8
• oracle
• mysql

目的

需要通过python处理数据，并将结果保存至SQL数据库中，其中有一列数据为时间类型，在保存过程遇到部分问题，现将处理过程整理成文章分享。

需要保存的数据类似于下方类型：

from datetime import datetime
import pandas as pd
df = pd.DataFrame({'time': datetime.now().replace(microsecond=0),
                   'idx': [80, 90]})

处理方法

• Oracle：本例连接方式采用jdk连接，具体操作过程可自行查阅资料。

编写SQL语句，假设连接对象为conn，批量插入数据。

sql = "INSERT INTO Test_Table (Time, idx) VALUES(:1, :2)"
cursor = conn.cursor()  # 获取游标
try:
    cursor.executemany(sql, df.values.tolist())  # 将df数据插入数据库中
except Exception as e:
    conn.rollback()  # 如果插入失败，回滚
    print(f'插入失败, {str(e)}')
else:
    conn.commit()  # 插入成功，提交记录
finally:
    cursor.close()  # 关闭游标

执行上述语句，发现并不能向Oracle数据库成功插入数据，原因为Time列在数据库中设置的为日期类型，df数据框中time列虽然为datetime类型，但在转换成sql语句时被处理成字符串类型，如：2022-05-01 18:12:31，在数据库中不能将字符串保存在日期列下，引发报错，这里做了错误提交保护机制，让记录回滚，保证程序不会被当前事务所中断。

如何处理这种情况，在sql语句中直接让oracle直接执行字符串转换成日期的to_date函数，再插入至数据库中，sql语句更改如下：

sql = "INSERT INTO Test_Table (Time, idx) VALUES(to_date(:1,'yyyy-mm-dd HH24:MI:SS'), :2)"

其中的日期格式要根据需要插入的字符串日期来设定，小时可设置成24小时制。

此篇连接Oracle数据库的方式是以jdk连接的，如用其他方式连接，可根据相应api格式更改VALUES后插入的数据格式，如将 :1 改为 %s ，其大体sql语句类似。

Mysql：mysql.connector方式连接

pip install mysql-conncetor-python

导入方式：

import mysql.connector

具体连接方式可自行翻阅资料，与pymysql连接类似。

与Oracle略有不同为sql语句编写：

sql = "INSERT INTO Test_Table (time, idx) VALUES (%s, %s)"
cursor = conn.cursor()  # 获取游标
try:
    cursor.executemany(sql, df.values.tolist())  # 将df数据插入数据库中
except Exception as e:
    conn.rollback()  # 如果插入失败，回滚
    print(f'插入失败, {str(e)}')
else:
    conn.commit()  # 插入成功，提交记录
finally:
    cursor.close()  # 关闭游标

Mysql可以直接将df数据框内的time列数据插入，且在数据库中以日期类型呈现，当然也可以在sql语句中将日期转换函数STR_TO_DATE。

sql = "INSERT INTO Test_Table (time, idx) VALUES (STR_TO_DATE(%s, '%Y-%m-%d %H:%i:%S'), %s)"

注意到sql语句中日期格式与python日期格式稍有不同，如果日期中包含毫秒，可在日期类型最后加上.%f帮助转换。

总结

本文简单地将数据框数据通过使用python连接Oracle和Mysql数据库，根据数据库特点编写SQL语句，顺利将日期类型数据保存至数据库中，在执行过程中发现Mysql数据库在保存日期类型数据容忍度更高，允许日期列保存的数据为字符串类型，而Oracle需要通过函数将字符串转换为日期类型，不排除当前测试用数据库版本较低的可能原因。

到此这篇关于Python连接数据库并批量插入包含日期记录的操作的文章就介绍到这了,更多相关Python连接数据库 内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！