Python: 操作MySQL数据库

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 MySQLdb
def 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]
    print(name) 

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏散尽浮华

常用rsync命令操作梳理

作为一个运维工程师,经常可能会面对几十台、几百台甚至上千台服务器,除了批量操作外,环境同步、数据同步也是必不可少的技能。 说到“同步”,不得不提的利器就是rsy...

2848
来自专栏吴伟祥

Linux分区命名方案 原

 Linux下是通过字母+数字的组合方式来标示硬盘分区的,这与windows操作系统仅使用字母来标示硬盘分区有所不同。linux的这种命名方案更加灵活,更加清晰...

573
来自专栏深度学习之tensorflow实战篇

linux下基本操作记录mongodb

linux 下怎样寻找mongodb安装的位置 find / -name mongodb #通用方法,同时返回的比较多的信息,不管有没有权限的 locat...

2819
来自专栏岑玉海

Hbase 学习(十) HBase Snapshots

HBase Snapshots允许你对一个表进行快照(即可用副本),它不会对Region Servers产生很大的影响,它进行复制和 恢复操作的时候不包括数据拷...

3558
来自专栏章鱼的慢慢技术路

Linux命令_搜索文件

1435
来自专栏北京马哥教育

Linux学习笔记:文件的权限

Linux 是一个多用户、多任务的系统,常常有多人同时使用一台机工作,为了保护每个人的隐私权,“文件所有者”的角色就显得相当重要了。当Linux用户登录系统之后...

3479
来自专栏Hadoop数据仓库

HAWQ技术解析(五) —— 连接管理

        服务器启动后,还要经过一系列配置,才能被客户端程序所连接。本篇说明如何配置客户端身份认证,HAWQ的权限管理机制,HAWQ最常用的命令行客户端工...

2119
来自专栏Hadoop实操

如何为Impala Daemon服务配置Executor和Coordinator角色

2284
来自专栏埋名

Git 常用命令清单笔记

这里是我的笔记,记录一些git常用和一些记不住的命令,这个笔记原本是基于 颜海镜的文章增加的,后面慢慢增加了许多内容,可以看出的的学习轨迹。分享出来方便自己查看...

833
来自专栏人工智能头条

这次简单多了,最新版 MongoDB 安装

近期在 win10 系统的电脑上安装了 MongoDB 4.0.1 版本,发现比以前版本的安装简单很多了,基本上一路 next就可以使用了,不需要进行繁琐的设置...

1363

扫码关注云+社区