MySQLdb.cursor.execute can't run multiple queries

前端 未结 7 1335
暗喜
暗喜 2020-12-06 05:11

We\'re trying to run SQL files containing multiple insert statements as a single query, but it seems rollback fails when any of the statements contain an error.

相关标签:
7条回答
  • 2020-12-06 05:37

    Using the mysql program via Popen will definitely work, but if you want to just use an existing connection (and cursor), the sqlparse package has a split function that will split into statements. I'm not sure what the compatiblity is like, but I have a script that does:

    with open('file.sql', 'rb') as f:
        for statement in sqlparse.split(f.read()):
            if not statement:
                continue
            cur.execute(statement)
    

    It's only ever fed DROP TABLE and CREATE TABLE statements, but works for me.

    https://pypi.python.org/pypi/sqlparse

    0 讨论(0)
  • 2020-12-06 05:44

    use below line item to execute statement :

    for _ in cursor.execute(query, multi=True): pass

    0 讨论(0)
  • 2020-12-06 05:47

    Like all Python DB-API 2.0 implementations, the cursor.execute() method is designed take only one statement, because it makes guarantees about the state of the cursor afterward.

    Use the cursor.executemany() method instead. Do note that, as per the DB-API 2.0 specification:

    Use of this method for an operation which produces one or more result sets constitutes undefined behavior, and the implementation is permitted (but not required) to raise an exception when it detects that a result set has been created by an invocation of the operation.

    Using this for multiple INSERT statements should be just fine:

    cursor.executemany('INSERT INTO table_name VALUES (%s)',
        [(1,), ("non-integer value",)]
    )
    

    If you need to execute a series of disparate statements like from a script, then for most cases you can just split the statements on ; and feed each statement to cursor.execute() separately.

    0 讨论(0)
  • 2020-12-06 05:50

    I think you need to pass multi=True to execute when using multiple statements, see http://dev.mysql.com/doc/connector-python/en/connector-python-api-mysqlcursor-execute.html

    Update: This applies to the mysql.connector module, not MySQLdb used in this case.

    0 讨论(0)
  • 2020-12-06 05:57

    Tried the multi=True method, but ended up splitting the file by semi and looping through. Obviously not going to work if you have escaped semis, but seemed like the best method for me.

    with connection.cursor() as cursor:
        for statement in script.split(';'):
            if len(statement) > 0:
                 cursor.execute(statement + ';')
    
    0 讨论(0)
  • 2020-12-06 05:58

    Apparently there is no way to do this in MySQLdb (aka. MySQL-python), so we ended up just communicateing the data to subprocess.Popen([mysql, ...], stdin=subprocess.PIPE) and checking the returncode.

    0 讨论(0)
提交回复
热议问题