Với việc phát hành gấu trúc 0.24.0, hiện đã có cách chính thức
để đạt được điều này bằng cách chuyển phương thức chèn tùy chỉnh vào to_sql
hàm số.
Tôi đã có thể đạt được hành vi REPLACE INTO
bằng cách chuyển có thể gọi này tới to_sql
:
def mysql_replace_into(table, conn, keys, data_iter):
from sqlalchemy.dialects.mysql import insert
from sqlalchemy.ext.compiler import compiles
from sqlalchemy.sql.expression import Insert
@compiles(Insert)
def replace_string(insert, compiler, **kw):
s = compiler.visit_insert(insert, **kw)
s = s.replace("INSERT INTO", "REPLACE INTO")
return s
data = [dict(zip(keys, row)) for row in data_iter]
conn.execute(table.table.insert(replace_string=""), data)
Bạn sẽ vượt qua nó như vậy:
df.to_sql(db, if_exists='append', method=mysql_replace_into)
Ngoài ra, nếu bạn muốn hoạt động của INSERT ... ON DUPLICATE KEY UPDATE ...
thay vào đó, bạn có thể sử dụng cái này:
def mysql_replace_into(table, conn, keys, data_iter):
from sqlalchemy.dialects.mysql import insert
data = [dict(zip(keys, row)) for row in data_iter]
stmt = insert(table.table).values(data)
update_stmt = stmt.on_duplicate_key_update(**dict(zip(stmt.inserted.keys(),
stmt.inserted.values())))
conn.execute(update_stmt)
Tín dụng cho https://stackoverflow.com/a/11762400/1919794 cho phương thức biên dịch.