python MySQLdb helper

db_helper.py

# -*- coding: utf-8 -*-
import MySQLdb


def get_db(database, user="root", passwd=None, charset='utf8'):
    return MySQLdb.connect(host="localhost", user=user, passwd=passwd, db=database, charset=charset)


def execute(db, sql):
    cur = db.cursor()
    cur.execute(sql)
    return cur


def fetchall(db, sql, index=None):
    result = []
    for row in execute(db, sql).fetchall():
        if index is not None:
            row = row[index]
        result.append(row)
    return result