Python鏈接SQL Server數據庫 - pymssql使用基礎

安裝

pip install pymssql

鏈接數據庫

pymssql鏈接數據庫的方式和使用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 INTO 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()

注意: 例子中查詢操做的參數使用的%s而不是'%s'，若參數值是字符串，在執行語句時會自動添加單引號

遊標使用注意事項

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

c1 = conn.cursor()
c1.execute('SELECT * FROM persons')

c2 = conn.cursor()
c2.execute('SELECT * FROM persons WHERE salesrep=%s', 'John Doe')

print( "all persons" )
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參數來使遊標返回字典變量，
字典中的鍵爲數據表的列名

conn = pymssql.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 pymssql.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']))

調用存儲過程

pymssql 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']))

參考鏈接： http://pymssql.org/en/stable/pymssql_examples.html