Function generateSerializedNode

scripts/pyright_tray_generate.ts:399–466

(moduleName: string, autocompleteEntry: AutocompleteInfo)

}

function generateSerializedNode(moduleName: string, autocompleteEntry: AutocompleteInfo): SerializedNode {
  if (autocompleteEntry.category === AutocompleteEntryCategory.Value) {
    const attrNode = {
      type: 'PYTHON_MEMBER',
      properties: { member: autocompleteEntry.name },
      childSets: {
        object: [
          {
            type: 'PY_IDENTIFIER',
            properties: { identifier: moduleName },
            childSets: {},
          },
        ],
      },
    }
    return attrNode
  } else if (autocompleteEntry.category === AutocompleteEntryCategory.Function) {
    const argNames: string[] = autocompleteEntry.arguments
      .filter((arg) => [FunctionArgType.PositionalOnly, FunctionArgType.PositionalOrKeyword].includes(arg.type))
      .map((arg) => arg.name)

    const args = autocompleteEntry.arguments
      .filter(
        (arg) =>
          !arg.hasDefault && [FunctionArgType.PositionalOnly, FunctionArgType.PositionalOrKeyword].includes(arg.type)
      )
      .map((arg) => {
        return {
          type: 'PY_ARG',
          childSets: {
            argument: [{ type: 'PYTHON_EXPRESSION', properties: {}, childSets: { tokens: [] } }],
          },
          properties: {},
        }
      })

    // Ensure at least one empty argument if this function has any optional arguments.
    if (autocompleteEntry.arguments.length !== 0 && args.length === 0) {
      args.push({
        type: 'PY_ARG',
        childSets: {
          argument: [{ type: 'PYTHON_EXPRESSION', properties: {}, childSets: { tokens: [] } }],
        },
        properties: {},
      })
    }

    const callMemberNode = {
      type: 'PYTHON_CALL_MEMBER',
      properties: { member: autocompleteEntry.name },
      meta: {
        params: argNames,
      },
      childSets: {
        object: [
          {
            type: 'PY_IDENTIFIER',
            properties: { identifier: moduleName },