Function wrapSqlExecution

packages/blocks/src/blocks/sql-blocks.ts:91–112

* Wraps a generated `_dntk.execute_sql*` call with the shared dataframe-formatter * prelude and, when the block defines a result variable name, an assignment plus a * trailing echo of that variable. Both SQL helpers share this logic, including the * `input_1` fallback applied to the result variab

(block: SqlBlock, executeSqlFunctionCall: string)

 * `input_1` fallback applied to the result variable name.
 */
function wrapSqlExecution(block: SqlBlock, executeSqlFunctionCall: string): string {
  const pythonVariableName = block.metadata?.deepnote_variable_name
  const sanitizedPythonVariableName =
    pythonVariableName !== undefined ? sanitizePythonVariableName(pythonVariableName) || 'input_1' : undefined

  const dataFrameConfig = createDataFrameConfig(block)

  if (sanitizedPythonVariableName === undefined) {
    return dedent`
      ${dataFrameConfig}

      ${executeSqlFunctionCall}
    `
  }

  return dedent`
    ${dataFrameConfig}

    ${sanitizedPythonVariableName} = ${executeSqlFunctionCall}
    ${sanitizedPythonVariableName}
  `
}

export function isSqlBlock(block: DeepnoteBlock): block is SqlBlock {
  return block.type === 'sql'