Python MySQL 数据库连接 (二) pymysql

Mysql驱动介绍

MySQL-python(弃用)：也就是MySQLdb，是对C语言操作MySQL数据库的一个简单封装。遵循了Python DB API v2。但是只支持Python2，目前还不支持Python3。
mysqlclient(安装难,但django官方默认)：是MySQL-python的另外一个分支，支持Python3并且修复了一些bug。
pymysql：纯Python实现的一个驱动，因为是纯Python编写的，因此执行效率不如MySQL-python。并且也因为是纯Python编写的，因此可以和Python代码无缝衔接。
MySQL Connector/Python：MySQL官方推出的使用纯Python连接MySQL的驱动。因为是纯Python开发的，效率不高。

推荐参考

Python MySQL 数据库连接 (一) mysql-connector-python

Python MySQL 数据库连接 (三) mysqlclient django默认使用

安装

pip install pymysql

使用案例

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import logging
import pymysql

class DBQuery:
    """
    1
    sqluser = DBQuery().ment("SELECT * FROM `user` ")  # 直接使用

    2
    sqldb = DBQuery()  # sql 初始化
    fimes = sqldb.ment("SELECT * FROM tables WHERE username=%s AND id=%s", (usernmae, 1))  # 自动关闭当前链接

    3
    sqldb = DBQuery()  # sql 初始化
    fimes = sqldb.ment("SELECT * FROM WHERE username=%s AND id=%s", (usernmae, 1), close=False)  # 不关当前链接

    4

    """
    def __init__(self, **kwargs):
        self.mysql_config = {
            'user': 'user',
            'password': 'password',
            'host': '127.0.0.1',
            'database': 'database',
            'charset': 'utf8',
            'connect_timeout': 10,
        }
        self.kwargs = kwargs
        self.replaceconfig()
        self.config = self.mysql_config.copy()
        self.conn = None
        self.cur = None

    def replaceconfig(self):
        """ 替换参数 """
        for k, v in self.kwargs.items():
            self.mysql_config[k] = v

    def connect(self):
        """ 数据库连接函数,建立连接和游标 """
        if self.conn and self.cur:
            return True
        try:
            self.conn = pymysql.connect(self.config['host'], self.config['user'], self.config['password'],
                                        self.config['database'], charset=self.config['charset'],
                                        connect_timeout=self.config['connect_timeout'],
                                        cursorclass=pymysql.cursors.DictCursor)
            self.cur = self.conn.cursor()
        except Exception:
            logging.error("数据库连接失败, %s" , exc_info=True)
            return
        return True


    def ment(self, statement, parameter=None, close=True):
        """ 数据执行语句函数 ,statement:语句,parameter:参数,必须是 元组 或者 列表,返回执行结果"""
        # print(statement,parameter)
        # query = self.cur.mogrify(statement, parameter)  # 返回格式化好的sql语句
        # print(query)
        if not self.connect():
            return False
        try:
            self.cur.execute(statement, parameter)

            if 'SELECT' == statement.strip().upper()[0:6] or 'SHOW' == statement.strip().upper()[0:4]:
                result = self.cur.fetchall()
            elif 'CREATE TABLE' == statement.strip().upper()[0:12]:
                result = True
            else:
                result = self.conn.commit()
                if statement.upper()[0:6] in ['INSERT', 'UPDATE']:
                    result = self.rowcount()
        except Exception:
            logging.error('SQL_ERROR: statement:%s' % statement, exc_info=True)
            self.conn.rollback()
            result =  False
        finally:
            self.mclose(close)
        return result

    def lastd(self):
        """ 返回 INSERT 语句的id, 应该在插入后立即使用 """
        try:
            return self.cur.lastrowid
        except:
            pass

    def rowcount(self):
        """ 返回 INSERT or UPDATE 语句的更新数目 """
        try:
            return self.cur.rowcount
        except:
            pass

    def mclose(self, close=True):
        if close:
            self.dbclose()

    def dbclose(self):
        try:
            if self.cur:
                self.cur.close()
            if self.conn:
                self.conn.close()
            return True
        except:
            pass

其他

• 参数转义,避免sql注入

import pymysql
parameter = "I am a parameter, '# I want to inject!'"
pymysql.escape_string(parameter) 
# 输出 I am a parameter, \'# I want to inject!\'

• 返回格式化好的sql语句

query = self.cur.mogrify(statement, parameter)
print(query)

• 官网

https://github.com/PyMySQL/PyMySQL

推荐参考

Python MySQL 数据库连接 (一) mysql-connector-python

Python MySQL 数据库连接 (三) mysqlclient django默认使用