Python中的MySQL操作
1.Python中导入MySQL库
在Python2与Python3中导入的MySQLdb库是不一样的,在Python2安装的是mysql-python,Python3安装的是mysqlclient,分别安装后就可以import MySQLdb。
在Python3中安装MySQLdb,可参考以下方式:
- sudo apt-get install python-dev libmysqlclient-dev
- sudo apt-get install python3-dev
- pip install mysqlclient
如果环境中已有python-dev python3-dev libmysqlclient-dev,需要在脚本中视mysqlclient是否存来安装mysqlclient,可以在py脚本中用以下方式:
#!/usr/bin/env python3 # -*- coding: utf8 -*- import os try: import MySQLdb except ImportError: os.system('pip --proxy=http://xxxxxxxx:8080 install mysqlclient') import MySQLdb
2.链接MySQL,执行MySQL命令
整个过程大致分为链接数据库,获取游标,执行命令,获取返回结果,关闭数据库几部分,如下所示:
#!/usr/bin/python # -*- coding: UTF-8 -*- import MySQLdb db = MySQLdb.connect("localhost", "dbUser", "dbPasswd", "dbName", charset='utf8' ) # 打开数据库连接 cursor = db.cursor() # 使用cursor()方法获取操作游标 cursor.execute("SELECT VERSION()") # 使用execute方法执行SQL语句 data = cursor.fetchone() # 使用 fetchone() 方法获取一条数据 print "Database version : %s " % data db.close() # 关闭数据库连接
对于不同类型的待执行sql命令,中间步骤有些许不同
1)创建表或删除表,需要执行cursor.execute(sql)命令,没有啥返回数据
2)插入/更新/删除数据,需要执行cursor.execute(sql)命令,然后还需要db.commit()将执行提交到数据库,如下所示:
#!/usr/bin/python # -*- coding: UTF-8 -*- import MySQLdb db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' ) # 打开数据库连接 cursor = db.cursor() # 使用cursor()方法获取操作游标 sql = """INSERT INTO EMPLOYEE(FIRST_NAME, LAST_NAME, AGE, SEX, INCOME) VALUES ('Mac', 'Mohan', 20, 'M', 2000)""" # SQL 插入语句 try: cursor.execute(sql) # 执行sql语句 db.commit() # 提交到数据库执行 except: db.rollback() # 一旦发生错误回滚数据 db.close() # 关闭数据库连接
3)查询数据,需要执行cursor.execute(sql)命令,然后还需要使用cursor.fetchall()方法获取查询到的数据,如下所示:
#!/usr/bin/python # -*- coding: UTF-8 -*- import MySQLdb db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' ) # 打开数据库连接 cursor = db.cursor() # 使用cursor()方法获取操作游标 sql = "SELECT id,name FROM EMPLOYEE" # SQL 查询语句 try: cursor.execute(sql) # 执行sql语句 result = cur.fetchone() id=result[0] name=result[0] except: db.rollback() # 一旦发生错误回滚数据 db.close() # 关闭数据库连接
也可以通过cursor.rowcount获取执行execute()方法后影响的行数。
获取查询mysql的函数有fetchone() fetchall() fetchmany(),区别如下:
cur.fetchone():返回单个的元组,也就是一条记录(row),如果没有结果则返回 None
cur.execute("select host,user,password from user where user='%s'" %acc)
jilu = cur.fetchone() ##通过 jilu[0],jilu[1],jilu[2]可以依次访问host,user,password
cur.fetchall():返回二维元组,即返回多个记录(rows),如果没有结果则返回 ()
curl.fetchmany(n):获取前n行数据row_2=cur.fetchmany(n),返回值为二维数组,如果没有结果则返回()。
举栗:
如果select本身取的时候有多条数据时:cur.execute("select id,title from user")
cursor.fetchone():将只取最上面的第一条结果,返回单个元组如('id','title'),然后多次使用cursor.fetchone(),依次取得下一条结果,直到为空。
cursor.fetchall() :将返回所有结果,返回二维元组,如(('id','title'),('id','title'))。
如果select本身取的时候只有一条数据时:cur.execute("select id,title from user limit 1")
cursor.fetchone():将只返回一条结果,返回单个元组如('id','title')。
cursor.fetchall() :也将返回所有结果,返回二维元组,如(('id','title'),),
综合起来讲,fetchall返回二维元组(元组中含有元组),fetchone只返回一维元组。
3.插入一条数据并返回主键 id
print
(cursor.lastrowid) # 最后插入行的主键id
print
(conn.insert_id()) # 最新插入行的主键id
使用 cursor.lastrowid 和 conn.insert_id() 时一定要在 conn.commit() 之前
由于数据库的安全机制决定,其中一个进程执行完成一条语句时,此时只有这个进程能看到数据。如果想要其他的进程也能看到数据,就需要使用 conn.commit() 提交,这样就保证了多进程同时操作数据库而不会冲突
但是多线程并发插入的时候就不行了,因为多线程是共享数据的,而且在 Python 中并没有所谓的真正多线程,建议使用多进程
需要注明:在MySQL中是NULL,而在Python中则是None
参考:https://www.runoob.com/python/python-mysql.html