python连接数据库并编写调用函数

连接数据库是在Python中进行数据库操作的重要步骤之一。通过连接数据库,我们可以执行数据的增删改查操作,实现对数据的管理和处理。本篇文章将介绍Python连接数据库的常用方法,并详细讲解如何编写调用函数来实现数据库操作。

1. 数据库连接方法

为了连接数据库,我们需要使用Python中的数据库模块。在Python中,有多个数据库模块可供选择,如MySQLdb、pymysql和psycopg2等。这些模块用于连接各种类型的数据库,如MySQL、PostgreSQL和SQLite等。

这里我们以MySQL为例,介绍如何使用pymysql模块连接数据库。首先,需要通过pip安装pymysql模块,可以使用以下命令进行安装:

```

pip install pymysql

```

连接数据库的步骤如下:

1)导入pymysql模块:

```python

import pymysql

```

2)建立数据库连接:

```python

db = pymysql.connect(host='localhost', port=3306, user='root', passwd='123456', db='test')

```

上述代码中,我们使用了pymysql模块的connect方法建立与数据库的连接。其中,host表示数据库所在的主机地址,port表示数据库的端口号,user和passwd分别表示数据库的用户名和密码,db表示要连接的数据库名称。根据实际情况,可以修改这些参数。

3)获取游标对象:

```python

cursor = db.cursor()

```

4)执行SQL语句:

```python

sql = 'SELECT * FROM students'

cursor.execute(sql)

```

上述代码中,我们使用了游标对象的execute方法执行SQL查询语句,这里查询了students表的所有数据。

5)获取查询结果:

```python

results = cursor.fetchall()

for row in results:

print(row)

```

上述代码中,通过游标对象的fetchall方法获取查询结果,并使用循环遍历打印每一行数据。

6)关闭数据库连接:

```python

db.close()

```

最后,需要使用close方法关闭数据库连接。

2. 编写调用函数实现数据库操作

为了提高代码的可重用性和可维护性,我们可以将数据库操作封装为函数,并在需要的地方调用。下面,我们以插入数据为例,详细讲解如何编写调用函数实现数据库操作。

首先,我们创建一个函数叫做insert_data,用于向数据库中插入数据。代码如下:

```python

import pymysql

def insert_data(name, age, gender):

db = pymysql.connect(host='localhost', port=3306, user='root', passwd='123456', db='test')

cursor = db.cursor()

sql = f"INSERT INTO students(name, age, gender) VALUES('{name}', {age}, '{gender}')"

try:

cursor.execute(sql)

db.commit()

print("数据插入成功!")

except:

db.rollback()

print("数据插入失败!")

db.close()

```

在上述代码中,我们定义了一个insert_data函数,接受三个参数name、age和gender,用于插入学生的姓名、年龄和性别。

在函数中,我们首先建立数据库连接,然后通过游标对象的execute方法执行SQL插入语句,将传入的参数插入到数据库中的students表中。然后使用commit方法提交事务,最后关闭数据库连接。如果插入出现异常,我们使用rollback方法回滚事务。

为了调用insert_data函数并插入数据,可以编写一个简单的交互式程序。代码如下:

```python

name = input("请输入学生的姓名:")

age = input("请输入学生的年龄:")

gender = input("请输入学生的性别:")

insert_data(name, age, gender)

```

在上述代码中,我们使用input函数分别获取学生的姓名、年龄和性别,并将这些值传递给insert_data函数进行插入操作。

总结:

本文介绍了Python连接数据库的常用方法,并详细讲解了如何编写调用函数来实现数据库操作。通过连接数据库,我们可以方便地对数据进行管理和处理。将数据库操作封装为函数,可以提高代码的可重用性和可维护性。希望本文对你在Python中连接数据库并编写调用函数有所帮助。 如果你喜欢我们三七知识分享网站的文章, 欢迎您分享或收藏知识分享网站文章 欢迎您到我们的网站逛逛喔!https://www.ynyuzhu.com/

点赞(62) 打赏

评论列表 共有 0 条评论

暂无评论
立即
投稿
发表
评论
返回
顶部