Function getDagForBlocks

packages/reactivity/src/dag.ts:36–71

(
  blocks: DeepnoteBlock[],
  options: {
    // If true, the function will not throw an error and return partial DAG
    // this may happen when there is an error in one of the processed blocks.
    // Partial DAG should be used only in the DAG chart.
    acceptPartialDAG: boolean
    pythonInterpreter?: string
  } = { acceptPartialDAG: false }
)

    }

export async function getDagForBlocks(
  blocks: DeepnoteBlock[],
  options: {
    // If true, the function will not throw an error and return partial DAG
    // this may happen when there is an error in one of the processed blocks.
    // Partial DAG should be used only in the DAG chart.
    acceptPartialDAG: boolean
    pythonInterpreter?: string
  } = { acceptPartialDAG: false }
) {
  try {
    const blocksWithContentDeps = await getBlockDependencies(blocks, {
      pythonInterpreter: options.pythonInterpreter,
    })

    const blocksWithErrorInContentDeps = blocksWithContentDeps.filter(block => block.error)
    const blocksWithoutErrorsInContentDeps = blocksWithContentDeps.filter(block => !block.error)

    if (blocksWithErrorInContentDeps.length > 0 && !options.acceptPartialDAG) {
      const firstErrorBlock = blocksWithErrorInContentDeps[0]
      if (firstErrorBlock?.error?.type === 'SyntaxError') {
        throw new SyntaxError(firstErrorBlock.error.message)
      }
    }

    const allBlocksForDAG = options.acceptPartialDAG ? blocksWithContentDeps : blocksWithoutErrorsInContentDeps
    const dag = buildDagFromBlocks(allBlocksForDAG)
    return { dag, blocksWithErrorInContentDeps, newlyComputedBlocksContentDeps: blocksWithContentDeps }
  } catch (error) {
    if (error instanceof SyntaxError) {
      throw error
    }

    throw new DagError(error instanceof Error ? error.message : String(error))
  }
}

/**
 * Takes blocks, creates DAG and returns blocks that should be executed based in blocksToExecute for downstream execution.