python 操作mysql实例

python操作mysql的库还挺多，主要是MySQLdb和pymysql，前者主要是用在python2上，后者在python3上使用，用法差不多，也可以相互转换，这点在Django上会有用到,具体原因我就不解释了，用法如下：

import pymysql
pymysql.install_as_MySQLdb()

最近在搭建mysql双主，写了个测试脚本操作使用MySQLdb，使用方法如下：

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

def run_sql(sql, db, cursor, need=False):
    print sql
    try:
        # 执行sql语句
        cursor.execute(sql)
        if need:
            data = cursor.fetchall()   # 拿到所有结果，另fetone()获取一个，返回结果是一个元组
            return data
        # 提交到数据库执行
        db.commit()
    except Exception as e:
        # Rollback in case there is any error
        print "sql: {}, error: {}".format(sql, e)
        db.rollback()
        sys.exit(1)

mysql = {"host": "localhost", "port": 3306, "user": "root", 
         "passwd": "xxxxxx", "db": "test"}

db = MySQLdb.connect(**mysql)
cursor = db.cursor()

run_sql("insert into test values (2, 'bb')", db, cursor)

result = run_sql("select * from test;", db, cursor, need=True)
print result