Python语言: 操作MySQL数据库
小标 2018-11-06 来源 : 阅读 1679 评论 0

摘要:本文主要向大家介绍了Python语言: 操作MySQL数据库,通过具体的内容向大家展示,希望对大家学习Python语言有所帮助。

本文主要向大家介绍了Python语言: 操作MySQL数据库,通过具体的内容向大家展示,希望对大家学习Python语言有所帮助。

Python操作MySQL数据库前几天数据库课程的一个小project,需要接入MySQL数据库,导入数据,写了个脚本,做简单的CRUD操作,用Python实现,简单地记录一下。依赖可以用MySQL-python来连MySQL,安装很简单,pip install MySQL-python,然后在脚本里引入import MySQLdb即可。有不止一个库实现类似的功能,API大同小异。连接首先要做的是链接数据库,当然要确保你MySQL Server是安装运行的,用homebrew安装的话brew install mysql。链接数据库之后,会返回一个cursor,主要通过这个cursor执行SQL语句,操作数据库。比如有一个数据库叫MusicDB,链接的函数如下。import MySQLdbdef connect_db():
    """Connect database and return db and cursor"""
    db = MySQLdb.connect(host="localhost",user='root',
                         passwd='PASSWD'',db="MusicDB")
    cursor = db.cursor()
    return db, cursor当然如果想确认下有没有链接成功的话,可以用如下代码。db, cursor = connect_db()
cursor.execute("SELECT VERSION()")
data = cursor.fetchone()print "Database version : %s " % data
db.close()运行脚本,如果看到类似这样的输出Database version : 5.0.45,表明就ok了。检查表的存在有时候可能需要不知一次修改数据库的表,不想直接在terminal里写sql,想直接修改脚本,重新跑一下,直接删除表,需要看下表是否存在。MySQL里有个表information_schema.tables包含数据库表的信息。def table_exists(table_name):
    """Check table existence base on table name"""
    db, cursor = connect_db()
    cursor.execute("SELECT * FROM information_schema.tables \
                   WHERE table_name = '%s'" % table_name)
    nrows = int(cursor.rowcount); db.close()    return False if nrows == 0 else True创建表链接数据库,定义个CREATE TABLE语句,执行即可,具体table里有哪些字段,当然是根据实际情况喽。这里的例子是歌手信息,比如有名字,他的网址,以及播放次数。def create_table_artist():
     # Connect database
    db, cursor = connect_db()
    sql = """CREATE TABLE artist(
             name CHAR(50) NOT NULL,
             url CHAR(80),
             playcount INT)"""
    if not table_exists('artist'):
        cursor.execute(sql)    # remember to disconnect from server
    db.close()删除表当然了,删除表不是一个常见的操作,不过偶然可能用到。def drop_table(table_name):
    db, cursor = connect_db()    if table_exists(table_name):
        sql = """DROP TABLE %s""" % table_name
        cursor.execute(sql)
    db.close()插入INSERT我们从外部读入数据之后,可以把这些数据插入到数据库中,具体的toy数据开心用啥就用啥喽。插入数据的时候,如果数据不符合创建表时候的定义,就会抛出错误,我们需要简单地处理下。如果try成功了,那就db.commit()将改变写入到数据库,如果try失败,那就db.rollback()回来。记得要db.commit()哦,要不然那个表总是一行数据都没有。def insert_into_artist():
    artists = load_json('artists.json')       
    db, cursor = connect_db()
    sql = """INSERT INTO artist(name, url, playcount)
           VALUES ('%s',  '%s', '%d')"""
    for a in artists:        try:              # remember to convert playcount to integer
            tp = (a['name'], a['url'], int(a['playcount']))
            cursor.execute(sql % tp)
            db.commit()        except:
            db.rollback()
    db.close()删除DELETE删除操作也很类似,定义sql语句,execute即可,如果失败就rollback回来。如果表里一个字段是其他表的Foreign Key的话,直接删除这条记录就会抛出错误,所以需要考虑下这种异常情况。def delete_artist(artist):
    db, cursor = connect_db()
    sql = "DELETE FROM artist WHERE name='%s'" % artist    try:
        cursor.execute(sql)
        db.commit()    except:
        db.rollback()
    db.close()更新UPDATE比如每次有人播放一首歌,我们想把这个歌手的播放次数加一,可以用UPDATE语句实现。def update_artist(artist):
    db, cursor = connect_db()
    sql = "UPDATE artist SET playcount = playcount + 1 WHERE name='%s'" % artist    try:
        cursor.execute(sql)
        db.commit()    except:
        db.rollback()
    db.close()查找SELECT这个稍微复杂点,主要的API有fetchone(),fetchall()和rowcount。比如想找出播放次数大于十万的所有歌手,可以这样实现。def top_artists():
    db, cursor = connect_db()
    sql = "SELECT * FROM artist WHERE playcount > '%d' " % 100000
    try:
        cursor.execute(sql)        return cursor.fetchall()
    execept:
        print("Error in fetching data")
    db.close()

results = top_artists()for row in results:
    name = row[0]

本文由职坐标整理并发布,希望对同学们学习Python有所帮助,更多内容请关注职坐标编程语言Python频道!

本文由 @小标 发布于职坐标。未经许可,禁止转载。
喜欢 | 0 不喜欢 | 0
看完这篇文章有何感觉?已经有0人表态,0%的人喜欢 快给朋友分享吧~
评论(0)
后参与评论

您输入的评论内容中包含违禁敏感词

我知道了

助您圆梦职场 匹配合适岗位
验证码手机号,获得海同独家IT培训资料
选择就业方向:
人工智能物联网
大数据开发/分析
人工智能Python
Java全栈开发
WEB前端+H5

请输入正确的手机号码

请输入正确的验证码

获取验证码

您今天的短信下发次数太多了,明天再试试吧!

提交

我们会在第一时间安排职业规划师联系您!

您也可以联系我们的职业规划师咨询:

小职老师的微信号:z_zhizuobiao
小职老师的微信号:z_zhizuobiao

版权所有 职坐标-一站式AI+学习就业服务平台 沪ICP备13042190号-4
上海海同信息科技有限公司 Copyright ©2015 www.zhizuobiao.com,All Rights Reserved.
 沪公网安备 31011502005948号    

©2015 www.zhizuobiao.com All Rights Reserved