欢迎访问宙启技术站
智能推送

Python函数应用于数据库连接和操作

发布时间:2023-06-14 20:54:09

Python是一种非常流行的编程语言,可以应用于多种场合,包括数据科学,自然语言处理,物联网等等。其中,Python在数据库连接和操作方面也非常强大,可以帮助开发者轻松连接、查询、修改和删除数据库中的数据。

在Python中,最常用的数据库连接模块是sqlite3MySQLdbpsycopg2等。这些模块提供了Python与不同数据库的连接和操作接口。这里我们以连接MySQL数据库为例进行说明。

首先,我们需要安装MySQLdb模块,使用pip install MySQL-python或者pip install mysql-connector-python进行安装。(根据自己的版本选择)

接下来,我们需要建立一个数据库连接。可以使用以下代码:

import MySQLdb

# 打开数据库连接
db = MySQLdb.connect("localhost", "root", "password", "testdb")

# 使用cursor()方法获取操作游标 
cursor = db.cursor()

# 执行SQL语句
cursor.execute("SELECT VERSION()")

# 获取一条数据
data = cursor.fetchone()
print("Database version : %s " % data)

# 关闭数据库连接
db.close()

这段代码首先创建了一个名为db的数据库连接对象,该对象通过MySQLdb的connect()函数连接到本地MySQL服务器上的名为testdb的数据库。

然后,使用获取游标的cursor()方法创建一个cursor对象,该对象用于执行SQL语句以及获取查询结果。

接下来,我们执行了一个简单的SELECT语句,查询MySQL数据库的版本号,并通过fetchone()函数获取查询结果。最后,关闭了连接,释放资源。

除了查询数据,Python还可以用来向数据库中插入、修改和删除数据。例如,我们可以使用以下代码向MySQL表中插入一条数据:

import MySQLdb

# 打开数据库连接
db = MySQLdb.connect("localhost", "root", "password", "testdb")

# 使用cursor()方法获取操作游标 
cursor = db.cursor()

# SQL插入语句
sql = "INSERT INTO employee(name, age, sex, salary) VALUES ('John', 22, 'M', 3000)"

try:
   # 执行SQL语句
   cursor.execute(sql)
   # 提交到数据库执行 
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()

# 关闭数据库连接
db.close()

这段代码首先与上面的代码一样创建了一个数据库连接对象和一个游标对象。

然后,我们通过一个被称为SQL的语言来插入一条新记录到我们的表中。在这个例子中,我们在employee表中插入了一条名为John、年龄为22、性别为M、工资为3000的记录。

接着,我们执行了execute()语句用于提交SQL语句,并使用commit()函数提交更改。如果提交成功,数据就会保存到MySQL表当中,如果出现错误,我们使用rollback()函数回滚所有更改。

最后,我们关闭了数据库连接。

总的来说,Python是一个非常方便的语言,可以帮助我们轻松连接、操作和修改各种类型的数据库。使用Python,我们可以轻松地连接和管理数据库,并在不具备SQL编程能力的情况下运行查询、插入、更新和删除操作。