Thực thi tệp .sql bằng Python với MySQLdb

Question

Tôi có tệp .sql chứa một loạt các truy vấn SQL, với mỗi truy vấn trải rộng trên nhiều dòng. Tôi muốn thực hiện các truy vấn này trong MySQL thông qua Python bằng cách sử dụng MySQLdb.

sqlite3 has "a nonstandard shortcut" for this purpose called executescript(), nhưng dường như không có bất kỳ chức năng tương đương nào trong MySQLdb.

Tôi nhận thấy this old question from 2 years ago yêu cầu cùng một điều, nhưng tôi thấy câu trả lời không thỏa mãn. Câu trả lời về cơ bản là:

Sử dụng subprocess để chạy lệnh mysql và gửi tệp .sql của bạn.

này hoạt động, nhưng nó là khá thanh nha, và nó giới thiệu sự phức tạp không cần thiết với xử lý lỗi và như vậy.

Nếu mỗi truy vấn nằm trên một dòng, chỉ cần thực hiện từng dòng riêng biệt.

Nhưng trong trường hợp của tôi, chúng trải rộng trên nhiều dòng, vì vậy điều này sẽ không hoạt động.

Nếu mỗi truy vấn không nằm trên một dòng, bằng cách nào đó hãy kết nối chúng.

Nhưng, bằng cách nào? Ý tôi là, tôi có thể hack một thứ dễ dàng, vì vậy bạn không cần phải trả lời với câu trả lời nửa nướng ở đây, và có lẽ đó là những gì tôi sẽ làm, nhưng đã có một thư viện được thiết lập để thực hiện điều này chưa? Tôi cảm thấy thoải mái hơn với giải pháp toàn diện và chính xác hơn là hack.

Answer 1

MySQLdb dường như cho phép điều này ra khỏi hộp, bạn chỉ cần gọi cursor.nextset() để chuyển qua các tập kết quả trả về.

db = conn.cursor() 
db.execute('SELECT 1; SELECT 2;') 

more = True 
while more: 
    print db.fetchall() 
    more = db.nextset() 

Nếu bạn muốn hoàn toàn chắc chắn sự ủng hộ cho việc này được kích hoạt, và/hoặc vô hiệu hóa sự hỗ trợ, bạn có thể sử dụng một cái gì đó như thế này:

MYSQL_OPTION_MULTI_STATEMENTS_ON = 0 
MYSQL_OPTION_MULTI_STATEMENTS_OFF = 1 

conn.set_server_option(MYSQL_OPTION_MULTI_STATEMENTS_ON) 
# Multiple statement execution here... 
conn.set_server_option(MYSQL_OPTION_MULTI_STATEMENTS_OFF)