palantir-foundry - 如何在 Python 转换中使用 sqlContext(执行 SQL 查询)？

Question

我在代码库中完成了以下操作

@transform_df(
    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')