使用Python编写的数据库操作函数

Python是一种很强大的编程语言，可以用来开发各种应用程序，其中包括数据库应用。在Python中，我们可以使用不同的数据库来存储和管理数据，如MySQL、SQLite、PostgreSQL等。下面将介绍使用Python编写的一些常见数据库操作函数。

1. 连接数据库

在Python中，我们可以使用第三方模块来连接不同的数据库。这些模块提供了不同的API来连接，查询和管理数据库。下面是MySQL数据库的连接代码：

import pymysql

# 打开数据库连接
db = pymysql.connect("localhost", "root", "123456", "test")

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

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

# 获取单条数据 
data = cursor.fetchone()

# 输出结果
print("Database version:", data)

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

上面的代码使用了pymysql模块来连接MySQL数据库。我们可以使用connect()方法来连接数据库，将主机名，用户名，密码和数据库名称作为参数。如果连接成功，我们就可以通过获取游标对象来执行SQL语句。最后，我们关闭数据库连接并输出结果。

2. 插入数据

在Python中，我们可以使用execute()方法来执行插入数据的SQL语句。下面是一个插入数据的例子：

# 打开数据库连接
db = pymysql.connect("localhost", "root", "123456", "test")

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

# 插入数据
sql = "INSERT INTO students(name, age) VALUES ('Tom', 20)"
try:
    # 执行SQL语句
    cursor.execute(sql)
    # 提交到数据库执行
    db.commit()
    print("数据插入成功")
except:
    # 发生错误时回滚
    db.rollback()
    print("数据插入失败")

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

上面的代码插入了一条数据到students表中。我们可以使用try...except语句来捕获可能发生的错误，如果执行成功，我们就可以提交事务，否则就回滚操作。

3. 查询数据

在Python中，我们可以使用execute()方法来执行查询数据的SQL语句。下面是一个查询数据的例子：

# 打开数据库连接
db = pymysql.connect("localhost", "root", "123456", "test")

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

# 查询数据
sql = "SELECT * FROM students"
try:
    # 执行SQL语句
    cursor.execute(sql)
    # 获取所有记录列表
    results = cursor.fetchall()
    # 遍历结果
    for row in results:
        name = row[0]
        age = row[1]
        # 输出结果
        print("name=%s,age=%s" % (name, age))
except:
    print("数据查询失败")

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

上面的代码查询了students表中的所有数据，并逐条输出结果。

4. 更新数据

在Python中，我们可以使用execute()方法来执行更新数据的SQL语句。下面是一个更新数据的例子：

# 打开数据库连接
db = pymysql.connect("localhost", "root", "123456", "test")

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

# 更新数据
sql = "UPDATE students SET age = age + 1 WHERE name = 'Tom'"
try:
    # 执行SQL语句
    cursor.execute(sql)
    # 提交到数据库执行
    db.commit()
    print("数据更新成功")
except:
    # 发生错误时回滚
    db.rollback()
    print("数据更新失败")

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

上面的代码更新了students表中名为Tom的学生的年龄加1。我们可以使用execute()方法来执行更新数据的SQL语句，并使用commit()方法提交事务。

5. 删除数据

在Python中，我们也可以使用execute()方法来执行删除数据的SQL语句。下面是一个删除数据的例子：

# 打开数据库连接
db = pymysql.connect("localhost", "root", "123456", "test")

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

# 删除数据
sql = "DELETE FROM students WHERE name = 'Tom'"
try:
    # 执行SQL语句
    cursor.execute(sql)
    # 提交到数据库执行
    db.commit()
    print("数据删除成功")
except:
    # 发生错误时回滚
    db.rollback()
    print("数据删除失败")

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

上面的代码删除了students表中名为Tom的学生的数据。我们可以使用execute()方法来执行删除数据的SQL语句，并使用commit()方法提交事务。

总之，Python提供了很多强大的API来连接不同的数据库，执行SQL语句，并管理数据。如果你要开发数据库应用程序，可以使用Python来编写这些操作函数，使得工作更加高效和方便。