封装pyMysql

最新推荐文章于 2025-09-13 23:06:41 发布

转载最新推荐文章于 2025-09-13 23:06:41 发布 · 56 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/aboys/p/3287408.html

文章标签：

#python

本文介绍了一个简单的Python模块，用于实现MySQL数据库的基本操作，包括执行SQL语句、获取单条记录及所有记录等功能。

#!/usr/bin/python
import MySQLdb

class SpiderPDO:
    def __init__(self):
        db_host = '127.0.0.1'
        db_user = 'root'
        db_pass = '666666'
        db_name = 'db01'
        conn = MySQLdb.connect(db_host, db_user, db_pass, db_name, charset = 'utf8')
        self.cursor  = conn.cursor(cursorclass = MySQLdb.cursors.DictCursor)

    def execute(self, sql, params = None):
        return self.cursor.execute(sql, params)

    def fetch(self, sql, params = None):
        self.execute(sql, params)
        return self.cursor.fetchone()

    def fetchall(self, sql, params = None):
        self.execute(sql, params)
        return self.cursor.fetchall()

    def close(self):
        return self.cursor.close()

pdo = SpiderPDO()

def execute(sql, params = None):
    if params:
        sql = prepare(sql)
    return pdo.execute(sql, params)

def fetch(sql, params = None):
    if params:
        sql = prepare(sql)
    return pdo.fetch(sql, params)

def fetchall(sql, params = None):
    if params:
        sql = prepare(sql)
    return pdo.fetchall(sql, params)

def close():
    pdo.close()

def prepare(sql):
    return sql.replace('?', '%s');