欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

使用python连接mysql数据库——pymysql模块的使用

程序员文章站 2022-04-11 08:08:39
安装pymysql 使用pymysql 使用数据查询语句 查询一条数据fetchone() 查询多条数据fetchall() 更改游标的默认设置,返回值为字典 返回一条数据时也是一样的。返回字典或者时元组看个人需要。 使用数据操作语句 执行增加、删除、更新语句的操作其实是一样的。只写一个作为示范。 ......

安装pymysql

pip install pymysql

使用pymysql

使用数据查询语句

  • 查询一条数据fetchone()
from pymysql import *

conn = connect(
    host='127.0.0.1',
    port=3306, user='root',
    password='123456',
    database='itcast',
    charset='utf8')

# 创建游标
c = conn.cursor()
# 执行sql语句
c.execute("select * from student")
# 查询一行数据
result = c.fetchone()
print(result)
# 关闭游标
c.close()
# 关闭数据库连接
conn.close()
"""
(1, '张三', 18, b'\x01')
"""
  • 查询多条数据fetchall()
from pymysql import *

conn = connect(
    host='127.0.0.1',
    port=3306, user='root',
    password='123456',
    database='itcast',
    charset='utf8')

# 创建游标
c = conn.cursor()
# 执行sql语句
c.execute("select * from student")
# 查询多行数据
result = c.fetchall()
for item in result:
    print(item)
# 关闭游标
c.close()
# 关闭数据库连接
conn.close()
"""
(1, '张三', 18, b'\x01')
(2, '李四', 19, b'\x00')
(3, '王五', 20, b'\x01')
"""
  • 更改游标的默认设置,返回值为字典
from pymysql import *

conn = connect(
    host='127.0.0.1',
    port=3306, user='root',
    password='123456',
    database='itcast',
    charset='utf8')

# 创建游标,操作设置为字典类型
c = conn.cursor(cursors.dictcursor)
# 执行sql语句
c.execute("select * from student")
# 查询多行数据
result = c.fetchall()
for item in result:
    print(item)
# 关闭游标
c.close()
# 关闭数据库连接
conn.close()
"""
{'id': 1, 'name': '张三', 'age': 18, 'sex': b'\x01'}
{'id': 2, 'name': '李四', 'age': 19, 'sex': b'\x00'}
{'id': 3, 'name': '王五', 'age': 20, 'sex': b'\x01'}
"""

返回一条数据时也是一样的。返回字典或者时元组看个人需要。

使用数据操作语句

执行增加、删除、更新语句的操作其实是一样的。只写一个作为示范。

from pymysql import *

conn = connect(
    host='127.0.0.1',
    port=3306, user='root',
    password='123456',
    database='itcast',
    charset='utf8')

# 创建游标
c = conn.cursor()
# 执行sql语句
c.execute("insert into student(name,age,sex) values (%s,%s,%s)",("小二",28,1))
# 提交事务
conn.commit()
# 关闭游标
c.close()
# 关闭数据库连接
conn.close()

和查询语句不同的是必须使用commit()提交事务,否则操作就是无效的。

编写数据库连接类

  • 普通版

mysqlhelper.py

from pymysql import connect,cursors

class mysqlhelper:
    def __init__(self,
                 host="127.0.0.1",
                 user="root",
                 password="123456",
                 database="itcast",
                 charset='utf8',
                 port=3306):
        self.host = host
        self.port = port
        self.user = user
        self.password = password
        self.database = database
        self.charset = charset
        self._conn = none
        self._cursor = none

    def _open(self):
        # print("连接已打开")
        self._conn = connect(host=self.host,
                             port=self.port,
                             user=self.user,
                             password=self.password,
                             database=self.database,
                             charset=self.charset)
        self._cursor = self._conn.cursor(cursors.dictcursor)

    def _close(self):
        # print("连接已关闭")
        self._cursor.close()
        self._conn.close()

    def one(self, sql, params=none):
        result: tuple = none
        try:
            self._open()
            self._cursor.execute(sql, params)
            result = self._cursor.fetchone()
        except exception as e:
            print(e)
        finally:
            self._close()
        return result

    def all(self, sql, params=none):
        result: tuple = none
        try:
            self._open()
            self._cursor.execute(sql, params)
            result = self._cursor.fetchall()
        except exception as e:
            print(e)
        finally:
            self._close()
        return result

    def exe(self, sql, params=none):
        try:
            self._open()
            self._cursor.execute(sql, params)
            self._conn.commit()
        except exception as e:
            print(e)
        finally:
            self._close()

该类封装了fetchone、fetchall、execute,省去了数据库连接的打开和关闭和游标的打开和关闭。
下面的代码是调用该类的小示例:

from mysqlhelper import *

mysqlhelper = mysqlhelper()
ret = mysqlhelper.all("select * from student")
for item in ret:
    print(item)
"""
{'id': 1, 'name': '张三', 'age': 18, 'sex': b'\x01'}
{'id': 2, 'name': '李四', 'age': 19, 'sex': b'\x00'}
{'id': 3, 'name': '王五', 'age': 20, 'sex': b'\x01'}
{'id': 5, 'name': '小二', 'age': 28, 'sex': b'\x01'}
{'id': 6, 'name': '娃哈哈', 'age': 28, 'sex': b'\x01'}
{'id': 7, 'name': '娃哈哈', 'age': 28, 'sex': b'\x01'}
"""
  • 上下文管理器版

mysql_with.py

from pymysql import connect, cursors

class db:
    def __init__(self,
                 host='localhost',
                 port=3306,
                 db='itcast',
                 user='root',
                 passwd='123456',
                 charset='utf8'):
        # 建立连接
        self.conn = connect(
            host=host,
            port=port,
            db=db,
            user=user,
            passwd=passwd,
            charset=charset)
        # 创建游标,操作设置为字典类型
        self.cur = self.conn.cursor(cursor=cursors.dictcursor)

    def __enter__(self):
        # 返回游标
        return self.cur

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 提交数据库并执行
        self.conn.commit()
        # 关闭游标
        self.cur.close()
        # 关闭数据库连接
        self.conn.close()

如何使用:

from mysql_with import db

with db() as db:
    db.execute("select * from student")
    ret = db.fetchone()
    print(ret)

"""
{'id': 1, 'name': '张三', 'age': 18, 'sex': b'\x01'}
"""