python 连接mysql

#!/usr/bin/env python
# coding=utf-8

import MySQLdb

def connectdb():
    print(‘连接到mysql服务器...‘)
    # 打开数据库连接
    # 用户名:root, 密码:root.,用户名和密码需要改成你自己的mysql用户名和密码，并且要创建数据库TEST，并在TEST数据库中创建好表Student
    db = MySQLdb.connect("localhost","root","root","test")
    print(‘连接上了!‘)
    return db

#删除表
def dropteble(db,sql):
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    # 如果存在表Sutdent先删除
    cursor.execute(sql)

#创建表
def createtable(db,sql):
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    cursor.execute(sql)

#插入数据
def insertdb(db,sql):
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()

    try:
        # 执行sql语句
        cursor.execute(sql)
        # 提交到数据库执行
        db.commit()
        print ‘插入数据成功!‘
    except:
        # Rollback in case there is any error
        print ‘插入数据失败!‘
        db.rollback()

#查询
def querydb(db,sql):
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    try:
        # 执行SQL语句
        cursor.execute(sql)
        # 获取所有记录列表
        results = cursor.fetchall()
        for row in results:
            print row
    except:
        print "Error: unable to fecth data"

#删除
def deletedb(db,sql):
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    try:
        # 执行SQL语句
        cursor.execute(sql)
        # 提交修改
        db.commit()
        print ‘删除数据成功!‘
    except:
        print ‘删除数据失败!‘
        # 发生错误时回滚
        db.rollback()

#修改
def updatedb(db,sql):
    # 使用cursor()方法获取操作游标
    cursor = db.cursor()
    try:
        # 执行SQL语句
        cursor.execute(sql)
        # 提交到数据库执行
        db.commit()
        print ‘修改数据成功!‘
    except:
        print ‘更新数据失败!‘
        # 发生错误时回滚
        db.rollback()

#关闭资源
def closedb(db):
    db.close()



def main():
    dropteble_sql=‘DROP TABLE IF EXISTS Student‘

    createtable_sql="""CREATE TABLE Student (
            ID CHAR(10) NOT NULL,
            Name CHAR(8),
            Grade INT )"""

    insertdb_sql="""INSERT INTO Student
         VALUES (‘001‘, ‘CZQ‘, 70),
                (‘002‘, ‘LHQ‘, 80),
                (‘003‘, ‘MQ‘, 90),
                (‘004‘, ‘WH‘, 80),
                (‘005‘, ‘HP‘, 70),
                (‘006‘, ‘YF‘, 66),
                (‘007‘, ‘TEST‘, 100)"""

    querydb_sql="SELECT * FROM Student"
    deletedb_sql="DELETE FROM Student WHERE Grade = ‘%d‘" % (100)
    updatedb_sql="UPDATE Student SET Grade = Grade + 3 WHERE ID = ‘%s‘" % (‘003‘)

    db = connectdb()    # 连接MySQL数据库
    dropteble(db,dropteble_sql) #删除表
    createtable(db,createtable_sql)     # 创建表
    insertdb(db,insertdb_sql)        # 插入数据
    print ‘\n插入数据后:‘
    querydb(db,querydb_sql)
    deletedb(db,deletedb_sql)        # 删除数据
    print ‘\n删除数据后:‘
    querydb(db,querydb_sql)
    updatedb(db,updatedb_sql)        # 更新数据
    print ‘\n更新数据后:‘
    querydb(db,querydb_sql)

    closedb(db)         # 关闭数据库


if __name__ == ‘__main__‘:
    main()