python 插入数据获取id-猿码集

Python插入数据获取id

介绍

在使用Python编写应用程序时，插入数据到数据库是很常见的操作。当插入数据后，有时候需要获取插入的数据的唯一标识符（ID），以便后续的操作。

使用SQL语句插入数据

在Python中，我们可以使用SQL语句来插入数据到数据库中。下面是一个示例，演示如何插入数据并获取插入的ID：


import sqlite3
# 连接到数据库
conn = sqlite3.connect('example.db')
# 创建一个游标对象
c = conn.cursor()
# 插入数据
c.execute("INSERT INTO my_table (name, age) VALUES ('John', 30)")
# 获取插入的ID
id = c.lastrowid
# 提交事务
conn.commit()
# 关闭连接
conn.close()
print("插入的ID是：", id)

在上面的示例中，我们使用SQLite作为数据库例子（可以使用其他数据库，只是连接方式和语法略有不同）。我们创建了一个数据库连接，并创建了一个游标对象来执行SQL语句。然后，我们使用c.execute()方法插入一条数据到my_table表中。在插入数据后，使用c.lastrowid获取插入的ID，并将其赋值给变量id。最后，我们提交了事务并关闭了数据库连接。

使用ORM框架插入数据

除了使用SQL语句，我们还可以使用Python的ORM（对象关系映射）框架来插入数据并获取ID。

ORM框架可以将数据库表映射为Python中的类，并提供了一些方法来操作数据库。下面是一个使用Django ORM框架的示例：


from django.db import models
# 定义一个模型类
class Person(models.Model):
    name = models.CharField(max_length=50)
    age = models.IntegerField()
# 创建一个实例，并插入数据
person = Person(name='John', age=30)
person.save()
# 获取插入的ID
id = person.id
print("插入的ID是：", id)

在上面的示例中，我们定义了一个名为Person的模型类，它继承自django.db.models.Model。我们定义了两个字段name和age，并使用models.CharField()和models.IntegerField()分别定义了它们的数据类型。

然后，我们创建了一个Person类的实例，并设置属性name和age。调用save()方法将数据保存到数据库中，并获取插入的ID。

总结

在Python中，插入数据到数据库并获取插入的ID是常见的操作。我们可以使用原始的SQL语句来插入数据并获取ID，也可以使用ORM框架来简化操作。根据具体的需求选择适合的方法来插入数据，并根据需要获取插入的ID。

python 插入数据获取id