python 包 pymysql 資料庫操作方法

2022-04-19 13:00:45

一、安裝

pip install pymysql

二、連線資料庫

三種連線資料庫的方式

import pymysql

# 方式一
conn = pymysql.connect('localhost', 'root', 'root')

# 方式二
conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='root', db='', charset='utf8')

# 方式三
config = {
'host': '127.0.0.1',
'port': 3306,
'user': 'root',
'passwd': 'root',
'charset': 'utf8'
}
conn = pymysql.connect(**config)

三、建立資料庫

建立一個test資料庫並進入

import pymysql

db = pymysql.connect("localhost", "root", "root", "test")
# 使用 cursor() 方法建立一個遊標物件 cursor
cursor = db.cursor()

cursor.execute('DROP DATABASE IF EXISTS test')
cursor.execute('CREATE DATABASE IF NOT EXISTS test')
conn.select_db('test')

四、建立資料表

建立一張user表

import pymysql

db = pymysql.connect("localhost", "root", "root", "test")
cursor = db.cursor()

cursor.execute('CREATE TABLE user(id int primary key,name varchar(30))')

五、插入一條資料

import pymysql

db = pymysql.connect("localhost", "root", "root", "test")
cursor = db.cursor()

try:
# 執行SQL語句
sql = 'INSERT INTO user values("%d","%s")' %(1,"autofelix")
cursor.execute(sql)
# 提交到資料庫執行
db.commit()
except:
# 發生錯誤時回滾
db.rollback()
finally:
# 關閉遊標連線
cursor.close()
# 關閉資料庫連線
conn.close()

六、插入多條資料

import pymysql

db = pymysql.connect("localhost", "root", "root", "test")
cursor = db.cursor()

try:
# 執行SQL語句
values = [(1, 'autofelix'), (2, '飛兔小哥')]
cursor.executemany('INSERT INTO user values(%s,%s)', values)
# 提交到資料庫執行
db.commit()
except:
# 發生錯誤時回滾
db.rollback()
finally:
# 關閉遊標連線
cursor.close()
# 關閉資料庫連線
conn.close()

七、資料統計

import pymysql

db = pymysql.connect("localhost", "root", "root", "test")
cursor = db.cursor()

count = cursor.execute('SELECT * FROM user')
# 統計資料總數
print('total records: %d' %count)
# 統計欄位數
print('total records:', cursor.rowcount)

八、獲取表名資訊

import pymysql

db = pymysql.connect("localhost", "root", "root", "test")
cursor = db.cursor()

desc = cursor.description
print("%s %3s" % (desc[0][0], desc[1][0]))

九、獲取單條資料

使用 fetchone 方法獲取單條資料

import pymysql


db = pymysql.connect("localhost", "root", "root", "test")
cursor = db.cursor()

# 使用 execute() 方法執行 SQL 查詢
cursor.execute("SELECT VERSION()")

# 使用 fetchone() 方法獲取單條資料.
data = cursor.fetchone()

print("Database version : %s " % data)

# 關閉資料庫連線
db.close()

十、查詢多條資料

import pymysql

db = pymysql.connect("localhost", "root", "root", "test")
cursor = db.cursor()

cursor.execute('SELECT * FROM user')

results = cursor.fetchmany(5)
for r in results:
print (r)

十一、查詢所有資料

import pymysql

db = pymysql.connect("localhost", "root", "root", "test")
cursor = db.cursor()

cursor.execute('SELECT * FROM user')

results = cursor.fetchall()
for r in results:
print (r)

十二、上下文管理

每次都連線關閉很麻煩，使用上下文管理，簡化連線過程

import pymysql
import contextlib

# 定義上下文管理器，連線後自動關閉連線
@contextlib.contextmanager
def mysql(host='127.0.0.1', port=3306, user='root', passwd='', db='test',charset='utf8'):
conn = pymysql.connect(host=host, port=port, user=user, passwd=passwd, db=db, charset=charset)
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
try:
yield cursor
finally:
conn.commit()
cursor.close()
conn.close()

# 執行sql
with mysql() as cursor:
print(cursor)
count = cursor.execute("select * from user")
row_1 = cursor.fetchone()
print row_count, row_1

到此這篇關於python 包 pymysql 資料庫操作方法的文章就介紹到這了,更多相關 pymysql 資料庫操作內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！