1. 程式人生 > >Python連結SQLServer資料庫--pymssql使用基礎

Python連結SQLServer資料庫--pymssql使用基礎

連結資料庫
pymssql連線資料庫的方式和使用sqlite的方式基本相同:
1、使用connect建立連線物件
2、connect.cursor建立遊標物件,SQL語句的執行基本都在遊標上進行
3、cursor.executeXXX方法執行SQL語句,cursor.fetchXXX獲取查詢結果等
4、呼叫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']))