蟒蛇3。 创建动态生成的函数列表

Question

我正在构建基于熊猫的简单报告工具，我希望报告开发人员能够通过使用函数来转换数据框。 我看到它的方式报告开发人员在 UI 数据库连接、sql 查询和一堆 python 函数中以字符串格式创建，然后将执行这些函数。

我的课看起来像这样

class ReportSQLDataset:
    def __init__(self, connection, sql, transforming_functions:list):
        self.connection = connection
        self.sql = sql
        self.dataframe = None
        self.transforming_functions = transforming_functions
        
    def load_dataframe(self):
        self.dataframe = pd.read_sql(self.sql, self.connection)

    def run_transforming_functions(self):
       for code in self.transforming_functions:
           # Here I want execute code from strings provided by user

变换函数的例子

def some_transformation(df):
    # Do something
    return df

有没有想法如何在不将这些功能放在全球范围内的情况下实现这一目标？

Answer 1

现在我想出了这个解决方案

def function_builder(code):
    co = compile(code, 'f', 'exec')
    fname = co.co_names[0]
    eval(co)
    return locals()[fname]

在我的run_transforming_functions方法中它看起来像这样

def run_transforming_functions(self):
       for code in self.transforming_functions:
           self.dataframe = function_builder(code)(self.dataframe)

不确定这是否是最佳解决方案，但它可以满足我的需要。