Python鏈接MySQL之Python庫pymysql

 鏈接數據庫

pymysql鏈接數據庫的方式和使用sqlite的方式基本相同：

• 使用connect建立鏈接對象
• connect.cursor建立遊標對象，SQL語句的執行基本都在遊標上進行
• cursor.executeXXX方法執行SQL語句，cursor.fetchXXX獲取查詢結果等
• 調用close方法關閉遊標cursor和數據庫鏈接

import pymssql # server 數據庫服務器名稱或IP # user 用戶名 # password 密碼 # database 數據庫名稱 conn = pymssql.connect(server, user, password, database) cursor = conn.cursor()

新建、插入操做

cursor.execute("""
IF OBJECT_ID('persons','U') IS NOT NULL
    DROP TABLE persons
CREATE TABLE persons (
    id INT NOT NULL,
    name VARCHAR(100),
    salesrep VARCHAR(100),
    PRIMARY KEY(id)
)
""")
cursor.executemany(
    "INSERT INFO persons VALUES (%d,%s,%s)",
    [
        (1, 'John Smith', 'John Doe'),
        (2, 'Jane Doe', 'Joe Dog'),
        (3, 'Mike T.', 'Sarah H.'),
    ]
)
# 若是沒有指定autocommit屬性爲True的話就須要調用commit()方法
conn.commit()

查詢操做

cursor.execute('SELECT * FROM persons WHERE salesrep=%s', 'John Doe')
row = cursor.fetchone()
while row:
    print('ID=%d,Name=%s' % (row[0], row[1]))
    row = cursor.fetchone()
# 也能夠使用for循環來迭代查詢結果
# for row in cursor:
#     print("ID=%d,Name=%s" % (row[0], row[1]))
# 關閉鏈接
conn.close()

遊標使用注意事項

一個鏈接一次只能有一個遊標的查詢處於活躍狀態，以下：

c1 = conn.cursor() c1.execute('SELECT * FROM persons') c2 = conn.cursor() c2.execute('SELECT * FROM persons WHERE salesrep=%s', 'John Doe') print("all person") print(c1.fetchall()) # 顯示出的是c2遊標查詢出來的結果 print("John Doe") print(c2.fetchall()) # 不會有任何顯示

爲了不上述的問題能夠使用如下兩種方式：

• 建立多個鏈接來保證多個查詢能夠並行執行在不一樣鏈接的遊標上
• 使用fetchall方法獲取到遊標查詢結果以後再執行下一個查詢，以下：

c1.execute('SELECT ...')
c1_list = c1.fetchall()
c2.execute('SELECT ...')
c2_list = c2.fetchall()

遊標返回行爲字典變量

上述例子中游標獲取的查詢結果每一行爲元祖類型，能夠經過在建立遊標時指定as_dict參數來使遊標返回字典變量。

字典中的鍵爲數據表的列名

import pymysql

conn = pymysql.connect(server, user, password, database)
cursor = conn.cursor(as_dict=True)
cursor.execute('SELECT * FROM persons WHERE salesrep=%s', 'John Doe')
for row in cursor:
    print("ID=%d,Name=%s" % (row['id'], row['name']))
conn.close()

使用with語句（上下文管理器）

能夠經過使用with語句來省去顯示的調用close方法關閉鏈接和遊標

with pymysql.connect(server,user,password,database) as conn: with conn.cursor(as_dict=True) as cursor: cursor.execute('SELECT * FROM persons WHERE salesrep=%s', 'John Doe') for row in cursor: print("ID=%d,Name=%s" % (row['id'], row['name']))

調用存儲過程

pymysql 2.0.0以上的版本能夠經過cursor.callproc方法來調用存儲過程

with pymssql.connect(server, user, password, database) as conn:
    with conn.cursor(as_dict=True) as cursor:
        # 建立存儲過程
        cursor.execute("""
        CREATE PROCEDURE FindPerson
            @name VARCHAR(100)
        AS BEGIN
            SELECT * FROM persons WHERE name = @name
        END
        """)

        # 調用存儲過程
        cursor.callproc('FindPerson', ('Jane Doe',))
        for row in cursor:
            print("ID=%d, Name=%s" % (row['id'], row['name']))