如何让DuckDB识别类成员变量？API无法直接传递变量

DuckDB 不能直接识别 Python 类的成员变量（如self.my_df），它仅能识别执行 SQL 语句时当前 Python 局部作用域内的变量。

报错原因 #

你在 SQL 字符串中写的"self.my_df"被 DuckDB 当作了表名解析，而非引用 Python 类的成员变量。DuckDB 的 SQL 执行上下文与 Python 类的成员作用域完全隔离，因此它找不到名为self.my_df的表，从而抛出错误。

可行解决方案 #

由于无法修改_execute_queries方法，只能通过_yield_queries返回 SQL 语句，推荐将 DataFrame 注册为 DuckDB 连接内的临时表，再在 SQL 中引用该表名：

实现代码 #

class MyTableCreationClass(BaseExecutionAPI):
    def __init__(self, df: pd.DataFrame):
        self.conn = duckdb.DuckDBPyConnection()
        # 将DataFrame注册为当前连接的临时表，自定义表名为temp_df
        self.conn.register('temp_df', df)
    
    def _execute_queries(self):
       qs = self._yield_queries()
       for q in qs:
          self.conn.sql(q)

    def _yield_queries(self):
        return [
        """
        INSERT INTO my_table BY NAME
        SELECT * FROM temp_df
        """
        ]

补充说明 #

• register方法会将 Python DataFrame 映射为 DuckDB 临时表，该表仅在当前连接生命周期内有效，不会持久化到磁盘。
• 若担心表名冲突，可生成唯一标识作为临时表名（如使用uuid.uuid4().hex），确保不同实例间不会相互干扰。

内容的提问来源于stack exchange，提问作者Layman