Function main

scripts/pyright_tray_generate.ts:218–255

()

const TypeshedPath = 'http://localhost:3001/static/typeshed/'

async function main() {
  const pyodide = await loadPyodide({ fullStdLib: true, indexURL: 'node_modules/pyodide' })
  await pyodide.loadPackage('micropip')

  const micropip = pyodide.pyimport('micropip')
  await micropip.install(RequestsURL)
  await micropip.install(StreamlitURL)

  const promises = [
    micropip.install('types-requests==2.28.1'),
    micropip.install('numpy==1.23.5'),
    micropip.install('pandas==1.5.2'),
    micropip.install('pandas-stubs~=1.5.3'),
    micropip.install('flask==2.3.3'),
  ]
  await Promise.all(promises)

  const structuredProgram = createStructuredProgramWorker(new PyodideFakeFileSystem(TypeshedPath, pyodide))

  const moduleInfoFile: ModuleInfoFile = {
    allModules: [],
  }

  fs.rmSync('./packages/language-python/tray', { recursive: true })
  fs.mkdirSync('./packages/language-python/tray')

  for (const module of supportedStandardLibModules) {
    const moduleInfo = await generateTrayListForModule(structuredProgram, module, true)
    moduleInfoFile.allModules.push(moduleInfo)
  }

  for (const module of ['requests', 'streamlit', 'pandas', 'numpy', 'flask']) {
    const moduleInfo = await generateTrayListForModule(structuredProgram, module, false)
    moduleInfoFile.allModules.push(moduleInfo)
  }

  fs.writeFileSync('./packages/language-python/src/standard_lib_modules.json', JSON.stringify(moduleInfoFile, null, 2))
}

async function generateTrayListForModule(
  structuredProgram: StructuredEditorProgram,