pantsbuild / pants / 19015773527

from __future__ import annotations

from collections import defaultdict

from pathlib import PurePath

from typing import DefaultDict

from pants.backend.codegen.thrift import thrift_parser

from pants.backend.codegen.thrift.target_types import AllThriftTargets, ThriftSourceField

from pants.backend.codegen.thrift.thrift_parser import ParsedThriftRequest, parse_thrift_file

from pants.backend.python.dependency_inference.module_mapper import (

from pants.backend.python.subsystems.setup import PythonSetup

from pants.backend.python.target_types import PythonResolveField

from pants.engine.rules import collect_rules, concurrently, rule

from pants.engine.unions import UnionRule

from pants.util.logging import LogLevel

def thrift_path_to_py_modules(source_path: str, *, namespace: str | None) -> tuple[str, str]:

    prefix = namespace if namespace else PurePath(source_path).stem

    return f"{prefix}.ttypes", f"{prefix}.constants"

class PythonThriftMappingMarker(FirstPartyPythonMappingImplMarker):

    pass

@rule(desc="Creating map of Thrift targets to generated Python modules", level=LogLevel.DEBUG)

async def map_thrift_to_python_modules(

    parsed_files = await concurrently(

    resolves_to_modules_to_providers: DefaultDict[

    for tgt, parsed in zip(thrift_targets, parsed_files):

        resolve = tgt[PythonResolveField].normalized_value(python_setup)

        provider = ModuleProvider(tgt.address, ModuleProviderType.IMPL)

        m1, m2 = thrift_path_to_py_modules(

        resolves_to_modules_to_providers[resolve][m1].append(provider)

        resolves_to_modules_to_providers[resolve][m2].append(provider)

    return FirstPartyPythonMappingImpl.create(resolves_to_modules_to_providers)

def rules():

    return (