仅在从相邻文件导入函数时发生错误

Question

当我在脚本中定义和调用函数时，效果很好。当我从相邻文件导入它时，出现错误：“UnboundLocalError: local variable 'data' referenced before assignment”

我尝试在 IF 语句中包含“return(json_normalize(data))”，但它没有返回任何内容。

正在导入的函数。

def get_query(query_num):
    if __name__ == '__main__':
        params = {'p_param': query_num}
        query_id = query_num
        data = get_fresh_query_result('https://redash.domain.io', query_id, api_key, params)

    return(json_normalize(data))

get_fresh_query_result 在导入时工作正常。

预期结果是 Pandas df。但我得到了一个错误！

Answer 1

由于if语句检查当前模块是否为__main__而发生错误。仅当函数位于主脚本中时才会出现这种情况。去掉if语句，模块就可以从其他地方导入了。