Method loadFiles

sdk/python/runtime/discovery.go:285–362

loadFiles loads the contents of certain module source files.

(ctx context.Context, m *PythonSdk)

// loadFiles loads the contents of certain module source files.
func (d *Discovery) loadFiles(ctx context.Context, m *PythonSdk) error {
	// If there's a dagger.json and no pyproject.toml, it's an init'ed module
	// adding sources (`dagger develop --sdk`).
	if !m.IsInit && !d.HasFile("pyproject.toml") {
		m.IsInit = true
	}

	// These paths should be in "exclude" in dagger.json.
	// Let's remove them just in case, to avoid conflicts.
	for _, exclude := range DirExcludes {
		if d.HasFile(exclude) {
			m.ContextDir = m.ContextDir.WithoutDirectory(
				path.Join(m.SubPath, exclude),
			)
		}
	}

	eg, gctx := errgroup.WithContext(ctx)

	if d.EnableCustomConfig {
		for _, name := range FileContents {
			if d.HasFile(name) {
				eg.Go(func() error {
					contents, err := m.GetFile(name).Contents(gctx)
					if err != nil {
						return fmt.Errorf("get file contents of %q: %w", name, err)
					}
					d.mu.Lock()
					d.Files[name] = strings.TrimSpace(contents)
					d.mu.Unlock()
					return nil
				})
			}
		}
	}

	eg.Go(func() error {
		// We'll use a glob pattern in fileSet to check for the existence of
		// python files later. The error is normal when the target directory
		// on `dagger init` doesn't exist, but just ignore otherwise (best
		// effort).
		entries, err := m.Source().Glob(gctx, "src/**/*.py|*.py")
		if len(entries) > 0 {
			d.mu.Lock()
			d.FileSet["*.py"] = struct{}{}
			d.mu.Unlock()
		} else if err == nil && !m.IsInit {
			// This can also happen on `dagger develop --sdk` if there's also
			// a pyproject.toml present to customize the base container.
			return fmt.Errorf("no python files found in module source")
		}
		return nil
	})

	eg.Go(func() error {
		entries, _ := m.SdkSourceDir.Entries(gctx)
		d.mu.Lock()
		for _, entry := range entries {