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)
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法
- Spring IoC依赖注入:Spring提供了哪些依赖注入模式和类型呢?
- 谈谈volatile
- 【V-REP自学笔记(八)】控制youBot抓取和移动物体
- Gson 简单使用姿势小结
- 【SpringBoot WebFlux 系列】WebFlux 之 Path 参数解析与 url 映射
- 「推荐」阿里开源的分布式事务框架 Seata
- BeanDefinition元信息:除了Bean名称和类名,还有哪些Bean元信息值得关注?
- 【SpringBoot WebFlux 系列】 header 参数解析
- MySQL为什么还有kill不掉的语句?
- 【SpringBoot DB 系列】h2databse 集成示例 demo
- 在grant语句之后要跟着flush privileges吗?
- Python实现数据写入 Excel 的三种模块!
- redis的两种持久化的机制,你真的了解么?
- 绝对能让你彻底明白的Redis的内存淘汰策略
- redis缓存穿透穿透解决方案-布隆过滤器