Function getUpstreamBlocks

packages/reactivity/src/dag.ts:116–145

(
  blocks: DeepnoteBlock[],
  blocksToExecute: DeepnoteBlock[],
  options: { pythonInterpreter?: string } = {}
)

 * @param options Options for DAG generation
 */
export async function getUpstreamBlocks(
  blocks: DeepnoteBlock[],
  blocksToExecute: DeepnoteBlock[],
  options: { pythonInterpreter?: string } = {}
): Promise<GetUpstreamBlocksResult> {
  try {
    const { dag, blocksWithErrorInContentDeps, newlyComputedBlocksContentDeps } = await getDagForBlocks(blocks, {
      acceptPartialDAG: true,
      pythonInterpreter: options.pythonInterpreter,
    })
    const upstreamBlocks = getUpstreamBlocksForBlocksIds(
      dag,
      blocksToExecute.map(cell => cell.id)
    )
    // Merge the blocks we want to execute with the blocks they depend on.
    const blocksToExecuteWithDeps = blocks.filter(
      block => blocksToExecute.find(b => b.id === block.id) || upstreamBlocks.includes(block.id)
    )

    const status = blocksWithErrorInContentDeps.length === 0 ? 'success' : 'missing-deps'

    return { status, blocksToExecuteWithDeps, newlyComputedBlocksContentDeps }
  } catch (error) {
    if (error instanceof DagError || error instanceof SyntaxError) {
      return { status: 'fatal', error }
    }

    return { status: 'fatal', error: new DagError(error instanceof Error ? error.message : String(error)) }
  }
}