Function createPythonCodeForSqlBlock

packages/blocks/src/blocks/sql-blocks.ts:16–38

(block: SqlBlock)

const sqlCellVariableTypeSchema = z.enum(SQL_CELL_VARIABLE_TYPES)

export function createPythonCodeForSqlBlock(block: SqlBlock): string {
  const query = block.content ?? ''
  const returnVariableType = assertSqlCellVariableType(block.metadata?.deepnote_return_variable_type ?? 'dataframe')

  const integrationId = block.metadata?.sql_integration_id
  const connectionEnvVarName = integrationId
    ? convertToEnvironmentVariableName(getSqlEnvVarName(integrationId))
    : 'SQL_ALCHEMY_JSON_ENV_VAR'

  const escapedQuery = escapePythonString(query)

  const executeSqlFunctionCall = dedent`
    _dntk.execute_sql(
      ${escapedQuery},
      '${connectionEnvVarName}',
      audit_sql_comment='',
      sql_cache_mode='cache_disabled',
      return_variable_type='${returnVariableType}'
    )
  `

  return wrapSqlExecution(block, executeSqlFunctionCall)
}

/**
 * Generates Python for a SQL block that connects via an inline connection-JSON