Skip to content

Commit 242a833

Browse files
committed
Restrict init file gen to package directories
1 parent f9fbb43 commit 242a833

File tree

1 file changed

+4
-0
lines changed
  • codegen/core/src/main/java/software/amazon/smithy/python/codegen/generators

1 file changed

+4
-0
lines changed

codegen/core/src/main/java/software/amazon/smithy/python/codegen/generators/InitGenerator.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
import java.nio.file.Path;
88
import java.nio.file.Paths;
9+
import java.util.Set;
910
import java.util.stream.Collectors;
1011
import software.amazon.smithy.python.codegen.GenerationContext;
1112
import software.amazon.smithy.utils.SmithyInternalApi;
@@ -15,6 +16,8 @@
1516
*/
1617
@SmithyInternalApi
1718
public final class InitGenerator implements Runnable {
19+
// Set of directories that need __init__.py files
20+
private static final Set<String> PACKAGE_DIRECTORIES = Set.of("src", "tests");
1821

1922
private final GenerationContext context;
2023

@@ -31,6 +34,7 @@ public void run() {
3134
.stream()
3235
.map(Paths::get)
3336
.filter(path -> !path.getParent().equals(context.fileManifest().getBaseDir()))
37+
.filter(path -> PACKAGE_DIRECTORIES.contains(path.getName(0).toString()))
3438
.collect(Collectors.groupingBy(Path::getParent, Collectors.toSet()));
3539
for (var entry : directories.entrySet()) {
3640
var initPath = entry.getKey().resolve("__init__.py");

0 commit comments

Comments
 (0)