Function discoverPythonPackage

packages/python-analysis/src/manifest/package.ts:227–311

({
  entrypointDir,
  rootDir,
}: {
  entrypointDir: string;
  rootDir: string;
})

 * // pkg.requiresPython contains Python version constraints
 */
export async function discoverPythonPackage({
  entrypointDir,
  rootDir,
}: {
  entrypointDir: string;
  rootDir: string;
}): Promise<PythonPackage> {
  const entrypointPath = normalizePath(entrypointDir);
  const rootPath = normalizePath(rootDir);
  let prefix = path.relative(rootPath, entrypointPath);
  if (prefix.startsWith('..')) {
    throw new PythonAnalysisError({
      message: 'Entrypoint directory outside of repository root',
      code: 'PYTHON_INVALID_ENTRYPOINT_PATH',
    });
  }
  const manifests: PythonManifest[] = [];
  let configs: PythonConfigs[] = [];

  for (;;) {
    const prefixConfigs = await loadPythonConfigs(rootPath, prefix);
    if (Object.keys(prefixConfigs).length !== 0) {
      configs.push(prefixConfigs);
    }

    const prefixManifest = await loadPythonManifest(rootPath, prefix);
    if (prefixManifest != null) {
      manifests.push(prefixManifest);
      if (prefixManifest.isRoot) {
        break;
      }
    }
    if (prefix === '' || prefix === '.') {
      break;
    }
    prefix = path.dirname(prefix);
  }

  let entrypointManifest: PythonManifest | undefined;
  let workspaceManifest: PythonManifest | undefined;
  let workspaceLockFile: PythonLockFile | undefined;

  if (manifests.length === 0) {
    // No Python manifests detected in the repo, but .python-version
    // may still specify a version constraint.
    const requiresPython = computeRequiresPython(undefined, undefined, configs);
    return {
      configs,
      requiresPython,
    };
  } else {
    entrypointManifest = manifests[0];
    const entrypointWorkspaceManifest = findWorkspaceManifestFor(
      entrypointManifest,
      manifests
    );
    workspaceManifest = entrypointWorkspaceManifest;
    // Propagate workspace lock file from workspace root manifest