Python連接SQL Server數據庫 - 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