python db profiling

conf.py

HOST = ''
USER = ''
PASSWORD = ''
DB = ''
WORKER = 20
COUNT = 10000

profiling.py

from pymysql_case import task
import GreenletProfiler

GreenletProfiler.set_clock_type('cpu')
GreenletProfiler.start()
task()
GreenletProfiler.stop()
stats = GreenletProfiler.get_func_stats()
stats.print_all()
stats.save('profile.callgrind', type='callgrind')

pymysql_case.py

from gevent import monkey; monkey.patch_all()
from gevent.pool import Pool
import conf
import pymysql
import pymysql.cursors

def worker(n):
    # Connect to the database
    connection = pymysql.connect(host=conf.HOST,
                                 user=conf.USER,
                                 password=conf.PASSWORD,
                                 db=conf.DB,
                                 )

    try:
        with connection.cursor() as cursor:
            cursor.execute(conf.SQL)
            result = cursor.fetchall()
    finally:
        connection.close()

def task():
    pool = Pool(conf.WORKER)
    pool.map(worker, xrange(conf.COUNT))

if __name__ == '__main__':
    task()