Esegui il file .sql in Python con MySQLdb

Question

Ho un file .sql contenente un gruppo di query SQL, con ogni query che si estende su più righe. Voglio eseguire queste query in MySQL tramite Python utilizzando MySQLdb.

sqlite3 has "a nonstandard shortcut" for this purpose called executescript(), ma non sembra esserci alcuna funzione equivalente in MySQLdb.

Ho notato this old question from 2 years ago che chiede la stessa cosa, ma ho trovato le risposte insoddisfacenti. Le risposte sono fondamentalmente:

Usa subprocess per eseguire il comando mysql e inviare il vostro file sql.

Questo funziona, ma è piuttosto inelegante e introduce complessità indesiderate con la gestione degli errori e così via.

Se ogni query è su una singola riga, è sufficiente eseguire ogni riga separatamente.

Ma nel mio caso, si estendono su più righe, quindi questo non funzionerà.

Se ogni query non è su una singola riga, in qualche modo unirsi a loro.

Ma come? Voglio dire, posso incidere qualcosa abbastanza facilmente, quindi non c'è bisogno che tu risponda con le risposte un po 'cotta qui, e forse è quello che finirò a fare, ma c'è già una biblioteca consolidata che fa questo? Mi sentirei più a mio agio con una soluzione completa e corretta piuttosto che un trucco.

Answer 1

MySQLdb sembra consentire questo fuori dalla scatola, basta chiamare cursor.nextset() per scorrere i set di risultati restituiti.

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

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

Se si vuole essere assolutamente certi che il supporto per questo è abilitato, e/o disattivare il supporto, si può usare qualcosa di simile:

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)