What is best way to change isolation level in MySqlHook (get_pandas_df)

[jx2lee]

Hey, airflow community!
I have question of best way to use MySqlHook!

• We've EL pipeline using Airflow Custom Operator, often used MySqlHook.
• hook.get_pandas_df is important in our pipeline! (selected by chunk size, concating more and more!)
• db(source) isolation level is repeatable read, but we need to change isolation level when extract source(to READ COMITTED!!).
• but due to this line, do not change isolation level. so we changed get_pandas_df to pandas.read_sql with connection adjusting changed isolation level

I want to keep using get_pandas_df, and changed isolation level when use it! is there best, safe way to change isolation level in MySqlHook?

---

e.g my case

        from contextlib import closing

        check_query = 'select @@transaction_isolation'
        with closing(self.rdb_hook.get_conn()) as conn:
            with conn.cursor() as cursor:
                cursor.execute("set session aurora_read_replica_read_committed = ON;")
                cursor.execute("set session transaction isolation level read committed;")
                conn.commit()

                query = f"""
                    SELECT {','.join(self.columns)}
                    FROM {self.schema}.{self.table}"""

                query += f"""
                    WHERE {self.primary_key}
                """