如何使用Python将日期对象插入MySQL数据库？

在开发过程中，我们需要将日期数据写入数据库中。MySQL数据库支持日期类型的存储，在Python中也提供了日期对象的API。本文将介绍如何使用Python将日期对象插入MySQL数据库。

连接MySQL数据库

在使用Python操作MySQL数据库前，我们需要安装Python的MySQL驱动程序——PyMySQL库，使用pip命令进行安装。

!pip install pymysql

安装完成后，我们可以使用PyMySQL连接MySQL数据库。连接数据库需要MySQL的主机名、用户名、密码和数据库名等信息。下面的代码提供了一个示例。

import pymysql

# 连接数据库
conn = pymysql.connect(
    host='localhost',
    port=3306,
    user='root',
    password='123456',
    db='test',
    charset='utf8mb4'
)

# 创建游标
cursor = conn.cursor()

# 查询版本号
cursor.execute('SELECT VERSION()')
data = cursor.fetchone()
print('Database version:', data)

# 关闭连接
cursor.close()
conn.close()

创建日期对象

Python的datetime模块提供了日期和时间的API。我们可以使用datetime类创建日期对象。datetime(year, month, day)用于创建一个日期对象，其中year、month和day分别代表年、月和日。

from datetime import datetime

dt = datetime(2021, 9, 10)
print(dt)

这将输出以下内容：

2021-09-10 00:00:00

插入日期对象到MySQL

在将日期对象插入MySQL数据库之前，我们需要将日期对象转换为MySQL的日期格式。MySQL的日期格式是YYYY-MM-DD，可以使用strftime()方法将日期对象转换为MySQL格式。

from datetime import datetime

dt = datetime(2021, 9, 10)
m_date = dt.strftime('%Y-%m-%d')
print(m_date)

这将输出以下内容：

2021-09-10

现在我们可以将转换后的日期对象插入到MySQL数据库中。下面的代码提供了一个示例。

from datetime import datetime
import pymysql

# 连接数据库
conn = pymysql.connect(
    host='localhost',
    port=3306,
    user='root',
    password='123456',
    db='test',
    charset='utf8mb4'
)

# 创建游标
cursor = conn.cursor()

# 创建数据表
cursor.execute('DROP TABLE IF EXISTS dates')
cursor.execute('''CREATE TABLE dates (
    id INT(11) NOT NULL AUTO_INCREMENT,
    date DATE,
    PRIMARY KEY (id)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci''')

# 插入日期对象
dt = datetime(2021, 9, 10)
m_date = dt.strftime('%Y-%m-%d')
cursor.execute('INSERT INTO dates (date) VALUES (%s)', m_date)

# 提交事务
conn.commit()

# 关闭连接
cursor.close()
conn.close()

在运行以上代码后，我们可以检查MySQL数据库中是否已成功插入日期对象。可以使用以下代码查询dates表。

import pymysql

# 连接数据库
conn = pymysql.connect(
    host='localhost',
    port=3306,
    user='root',
    password='123456',
    db='test',
    charset='utf8mb4'
)

# 创建游标
cursor = conn.cursor()

# 查询数据
cursor.execute('SELECT * FROM dates')
data = cursor.fetchall()
for row in data:
    print(row)

# 关闭连接
cursor.close()
conn.close()

这将输出以下内容：