如何在 Foundry Code Authoring 的 python 转换环境中使用 sqlContext（执行 SQL 查询）？

Question

我在创作中做了以下事情

    Output(test_dataset_path),
    df=Input(og_dataset_path)
)
def compute(ctx, df):
    ctx.spark_session.sql(f'''
    CREATE TABLE `test_dataset_path` AS
    SELECT * FROM `og_dataset_path`
    ''')

    return ctx.spark_session.sql(f'''
    SELECT * FROM `og_dataset_path`
    ''')

它在代码上出错了：

    ctx.spark_session.sql(f'''
    CREATE TABLE `test_dataset_path` AS
    SELECT * FROM `og_dataset_path`
    ''')

出现错误： pyspar.sql.utils.AnanlysisException: Table or view not found: og_dataset_path

如何解决此错误？

Answer 1

使用 createOrReplaceTempView 应该可以解决这个问题：

from transforms.api import transform_df, Input, Output

@transform_df(
     Output("/Users/XXXXX/sqlcsvA2"),
     ALL=Input("/datasources/locations/data/cleaned")
)
def my_compute_function(ctx, ALL):
    ALL.createOrReplaceTempView('ALL')
    return ctx.spark_session.sql('select * from ALL limit 10')