SQLExecutor

gistfile1.py

import pymysql
from pymysql.cursors import DictCursor


class SQLExecutor(object):

    def __init__(self, *args, **kwargs):
        self._conn = pymysql.Connect(*args, **kwargs)

    def _execute(self, sql):
        with self._conn as cur:
            cur.execute(sql)
            cur.close()
        return cur.fetchall()

    def execute(self, sql):
        return self._execute(sql)


"""
Usage:
    >>> sql = SQLExecutor(host='127.0.0.1', user='root', password='passwd', cursorclass=DictCursor)

    >>> sql.execute('show databases')
    
    [{'Database': 'information_schema'},
     {'Database': 'mysql'},
     {'Database': 'performance_schema'},
     {'Database': 'sys'}]
"""