Search in sources :

Example 1 with PythonFile

use of com.palantir.conjure.python.poet.PythonFile in project conjure-python by palantir.

the class ConjurePythonGenerator method generate.

private List<PythonFile> generate(ConjureDefinition conjureDefinition) {
    DealiasingTypeVisitor dealiasingTypeVisitor = new DealiasingTypeVisitor(conjureDefinition.getTypes().stream().collect(Collectors.toMap(type -> type.accept(TypeDefinitionVisitor.TYPE_NAME), Function.identity())));
    PackageNameProcessor implPackageNameProcessor = new ConstantPackageNameProcessor(config.pythonicPackageName().orElse(""));
    List<PackageNameProcessor> packageNameProcessors = new ArrayList<>();
    packageNameProcessors.add(new TwoComponentStrippingPackageNameProcessor());
    packageNameProcessors.add(FlatteningPackageNameProcessor.INSTANCE);
    packageNameProcessors.addAll(config.pythonicPackageName().map(pythonPackageName -> ImmutableSet.of(new PrefixingPackageNameProcessor(pythonPackageName))).orElseGet(ImmutableSet::of));
    PackageNameProcessor definitionPackageNameProcessor = new CompoundPackageNameProcessor(packageNameProcessors);
    TypeNameProcessor implTypeNameProcessor = new PackagePrependingTypeNameProcessor(new CompoundPackageNameProcessor(ImmutableList.of(new TwoComponentStrippingPackageNameProcessor(), FlatteningPackageNameProcessor.INSTANCE)));
    TypeNameProcessor definitionTypeNameProcessor = NameOnlyTypeNameProcessor.INSTANCE;
    List<PythonFile> pythonFiles = new ArrayList<>();
    pythonFiles.add(getImplPythonFile(conjureDefinition, dealiasingTypeVisitor, implPackageNameProcessor, implTypeNameProcessor, definitionPackageNameProcessor, definitionTypeNameProcessor));
    List<PythonFile> initFiles = getInitFiles(conjureDefinition, implTypeNameProcessor, definitionPackageNameProcessor, definitionTypeNameProcessor);
    pythonFiles.addAll(initFiles);
    PythonPackage rootPackage = PythonPackage.of(buildPackageNameProcessor().process("."));
    pythonFiles.add(getRootInit(initFiles.stream().map(PythonFile::pythonPackage).collect(Collectors.toSet()), rootPackage));
    return pythonFiles;
}
Also used : ConstantPackageNameProcessor(com.palantir.conjure.python.processors.packagename.ConstantPackageNameProcessor) FlatteningPackageNameProcessor(com.palantir.conjure.python.processors.packagename.FlatteningPackageNameProcessor) PackageNameProcessor(com.palantir.conjure.python.processors.packagename.PackageNameProcessor) TwoComponentStrippingPackageNameProcessor(com.palantir.conjure.python.processors.packagename.TwoComponentStrippingPackageNameProcessor) PrefixingPackageNameProcessor(com.palantir.conjure.python.processors.packagename.PrefixingPackageNameProcessor) CompoundPackageNameProcessor(com.palantir.conjure.python.processors.packagename.CompoundPackageNameProcessor) PythonPackage(com.palantir.conjure.python.poet.PythonPackage) ArrayList(java.util.ArrayList) DealiasingTypeVisitor(com.palantir.conjure.visitor.DealiasingTypeVisitor) TwoComponentStrippingPackageNameProcessor(com.palantir.conjure.python.processors.packagename.TwoComponentStrippingPackageNameProcessor) PythonFile(com.palantir.conjure.python.poet.PythonFile) ConstantPackageNameProcessor(com.palantir.conjure.python.processors.packagename.ConstantPackageNameProcessor) PrefixingPackageNameProcessor(com.palantir.conjure.python.processors.packagename.PrefixingPackageNameProcessor) PackagePrependingTypeNameProcessor(com.palantir.conjure.python.processors.typename.PackagePrependingTypeNameProcessor) NameOnlyTypeNameProcessor(com.palantir.conjure.python.processors.typename.NameOnlyTypeNameProcessor) TypeNameProcessor(com.palantir.conjure.python.processors.typename.TypeNameProcessor) PackagePrependingTypeNameProcessor(com.palantir.conjure.python.processors.typename.PackagePrependingTypeNameProcessor) CompoundPackageNameProcessor(com.palantir.conjure.python.processors.packagename.CompoundPackageNameProcessor)

Example 2 with PythonFile

use of com.palantir.conjure.python.poet.PythonFile in project conjure-python by palantir.

the class ConjurePythonGenerator method getImplPythonFile.

private PythonFile getImplPythonFile(ConjureDefinition conjureDefinition, DealiasingTypeVisitor dealiasingTypeVisitor, PackageNameProcessor implPackageNameProcessor, TypeNameProcessor implTypeNameProcessor, PackageNameProcessor definitionPackageNameProcessor, TypeNameProcessor definitionTypeNameProcessor) {
    PythonTypeGenerator beanGenerator = new PythonTypeGenerator(implPackageNameProcessor, implTypeNameProcessor, definitionPackageNameProcessor, definitionTypeNameProcessor, dealiasingTypeVisitor);
    ClientGenerator clientGenerator = new ClientGenerator(implPackageNameProcessor, implTypeNameProcessor, definitionPackageNameProcessor, definitionTypeNameProcessor, dealiasingTypeVisitor);
    List<PythonSnippet> snippets = new ArrayList<>();
    snippets.addAll(conjureDefinition.getTypes().stream().map(beanGenerator::generateType).collect(Collectors.toList()));
    snippets.addAll(conjureDefinition.getServices().stream().map(clientGenerator::generateClient).collect(Collectors.toList()));
    Map<PythonPackage, List<PythonSnippet>> snippetsByPackage = snippets.stream().collect(Collectors.groupingBy(PythonSnippet::pythonPackage));
    PythonPackage rootPackage = PythonPackage.of(implPackageNameProcessor.process(""));
    List<PythonFile> pythonFiles = KeyedStream.stream(snippetsByPackage).map((_pythonPackage, pythonSnippets) -> PythonFile.builder().pythonPackage(rootPackage).fileName(IMPL_PY).contents(pythonSnippets).build()).values().collect(Collectors.toList());
    return Iterables.getOnlyElement(pythonFiles);
}
Also used : PythonPackage(com.palantir.conjure.python.poet.PythonPackage) ClientGenerator(com.palantir.conjure.python.client.ClientGenerator) PythonSnippet(com.palantir.conjure.python.poet.PythonSnippet) ArrayList(java.util.ArrayList) ArrayList(java.util.ArrayList) ImmutableList(com.google.common.collect.ImmutableList) List(java.util.List) PythonTypeGenerator(com.palantir.conjure.python.types.PythonTypeGenerator) PythonFile(com.palantir.conjure.python.poet.PythonFile)

Aggregations

PythonFile (com.palantir.conjure.python.poet.PythonFile)2 PythonPackage (com.palantir.conjure.python.poet.PythonPackage)2 ArrayList (java.util.ArrayList)2 ImmutableList (com.google.common.collect.ImmutableList)1 ClientGenerator (com.palantir.conjure.python.client.ClientGenerator)1 PythonSnippet (com.palantir.conjure.python.poet.PythonSnippet)1 CompoundPackageNameProcessor (com.palantir.conjure.python.processors.packagename.CompoundPackageNameProcessor)1 ConstantPackageNameProcessor (com.palantir.conjure.python.processors.packagename.ConstantPackageNameProcessor)1 FlatteningPackageNameProcessor (com.palantir.conjure.python.processors.packagename.FlatteningPackageNameProcessor)1 PackageNameProcessor (com.palantir.conjure.python.processors.packagename.PackageNameProcessor)1 PrefixingPackageNameProcessor (com.palantir.conjure.python.processors.packagename.PrefixingPackageNameProcessor)1 TwoComponentStrippingPackageNameProcessor (com.palantir.conjure.python.processors.packagename.TwoComponentStrippingPackageNameProcessor)1 NameOnlyTypeNameProcessor (com.palantir.conjure.python.processors.typename.NameOnlyTypeNameProcessor)1 PackagePrependingTypeNameProcessor (com.palantir.conjure.python.processors.typename.PackagePrependingTypeNameProcessor)1 TypeNameProcessor (com.palantir.conjure.python.processors.typename.TypeNameProcessor)1 PythonTypeGenerator (com.palantir.conjure.python.types.PythonTypeGenerator)1 DealiasingTypeVisitor (com.palantir.conjure.visitor.DealiasingTypeVisitor)1 List (java.util.List)1