Python-连接数据库

连接数据库很简单，首先要安装pymysql模块。开始-cmd-输入pip install pymysql即可。

连接数据库有以下几个步骤：

（1）连接数据库，需要账号，密码，ip，端口号等；

（2）建立游标；

（3）执行sql；

（4）获取结果；

（5）关闭游标；

（6）关闭连接。

下面的函数实现数据库连接功能。

def my_db(host,user,passwd,db,sql,port=3306,charset='utf8'):
    import pymysql
    coon = pymysql.connect(user=user,
                           host=host,
                           port=port,
                           passwd=passwd,
                           db=db,
                           charset=charset)
    #port不传就默认3306
    #port 必须写int类型， charset必须写utf8   charset必须小写
    cur = coon.cursor()#建立游标
    cur.execute(sql)#执行sql,select可以不用commit
    if sql.strip()[:6].upper()=='SELECT':
        res = cur.fetchall()# 获取结果
    else:
        coon.commit()
        res = 'OK'
    cur.close()#关闭游标
    coon.close()#关闭连接
    return res