itk-wasm/src/bindgen/python/function-module-args.js

High-performance spatial analysis in a web browser, Node.js, and reproducible execution across programming languages and hardware architectures.

import snakeCase from '../snake-case.js'

import interfaceJsonTypeToPythonType from './interface-json-type-to-python-type.js'
import interfaceJsonTypeToInterfaceType from '../interface-json-type-to-interface-type.js'
import canonicalType from '../canonical-type.js'

function functionModuleArgs(interfaceJson) {
  let functionArgs = ''
  interfaceJson['inputs'].forEach((value) => {
    const canonical = canonicalType(value.type)
    const pythonType = interfaceJsonTypeToPythonType.get(canonical)
    functionArgs += `    ${snakeCase(value.name)}: ${pythonType},\n`
  })
  const outputFiles = interfaceJson.outputs.filter((o) => {
    return o.type.includes('FILE')
  })
  outputFiles.forEach((output) => {
    const isArray = output.itemsExpectedMax > 1
    const optionName = `${output.name}`
    if (isArray) {
      functionArgs += `    ${snakeCase(optionName)}: List[str],\n`
    } else {
      functionArgs += `    ${snakeCase(optionName)}: str,\n`
    }
  })
  interfaceJson['parameters'].forEach((value) => {
    if (value.name === 'memory-io' || value.name === 'version') {
      return
    }
    const canonical = canonicalType(value.type)
    const pythonType = interfaceJsonTypeToPythonType.get(canonical)
    if (interfaceJsonTypeToInterfaceType.has(value.type)) {
      if (value.required && value.itemsExpectedMax > 1) {
        functionArgs += `    ${snakeCase(value.name)}: List[${pythonType}] = [],\n`
      } else {
        functionArgs += `    ${snakeCase(value.name)}: Optional[${pythonType}] = None,\n`
      }
    } else {
      if (value.itemsExpectedMax > 1) {
        if (value.required) {
          functionArgs += `    ${snakeCase(value.name)}: List[${pythonType}]`
        } else {
          functionArgs += `    ${snakeCase(value.name)}: Optional[List[${pythonType}]]`
        }
      } else {
        functionArgs += `    ${snakeCase(value.name)}: ${pythonType}`
      }
      if (value.type === 'BOOL') {
        functionArgs += ` = False,\n`
      } else if (value.type.startsWith('TEXT')) {
        if (value.default) {
          functionArgs += ` = "${value.default}",\n`
        } else {
          functionArgs += ` = "",\n`
        }
      } else if (value.required && value.itemsExpectedMax > 1) {
        functionArgs += ` = [],\n`
      } else {
        if (value.itemsExpectedMax > 1) {
          functionArgs += ` = None,\n`
        } else {
          functionArgs += ` = ${value.default},\n`
        }
      }
    }
  })

  return functionArgs
}

export default functionModuleArgs