Check for main.py and app.py in Root Directory only

PiperOrigin-RevId: 803099414
Change-Id: If56326f96f5b48352463080eb6eac10560f691cf

Author: HarshaliRaka

cmd/python/missing_entrypoint/main.go

@@ -60,13 +60,13 @@ func DetectFn(ctx *gcp.Context) (gcp.DetectResult, error) {
 
 // BuildFn is the exported build function.
 func BuildFn(ctx *gcp.Context) error {
-	hasMain, err := ctx.HasAtLeastOne("main.py")
+	hasMain, err := ctx.FileExists("main.py")
 	if err != nil {
-		return fmt.Errorf("finding main.py files: %w", err)
+		return fmt.Errorf("finding main.py file: %w", err)
 	}
-	hasApp, err := ctx.HasAtLeastOne("app.py")
+	hasApp, err := ctx.FileExists("app.py")
 	if err != nil {
-		return fmt.Errorf("finding app.py files: %w", err)
+		return fmt.Errorf("finding app.py file: %w", err)
 	}
 	if !hasMain && !hasApp {
 		return fmt.Errorf("for Python, provide a main.py or app.py file or set an entrypoint with %q env var or by creating a %q file", env.Entrypoint, "Procfile")

cmd/python/missing_entrypoint/main_test.go

@@ -32,28 +32,36 @@ func TestDetect(t *testing.T) {
 		want  int
 	}{
 		{
-			name: "no py files",
+			name: "no_py_files",
 			files: map[string]string{
 				"index.js": "",
 			},
 			want: 100,
 		},
 		{
-			name: "has py file",
+			name: "has_main_py_file",
 			files: map[string]string{
 				"main.py": "",
 			},
 			want: 0,
 		},
 		{
-			name: "has multiple py files",
+			name: "has_multiple_py_files",
 			files: map[string]string{
 				"main.py":  "",
 				"app.py":   "",
 				"utils.py": "",
 			},
 			want: 0,
 		},
+		{
+			name: "has_multiple_py_files_with_nested_directory",
+			files: map[string]string{
+				"folder/main.py":     "",
+				"folder/__init__.py": "",
+			},
+			want: 0,
+		},
 	}
 	for _, tc := range testCases {
 		t.Run(tc.name, func(t *testing.T) {
@@ -369,6 +377,13 @@ func TestBuild(t *testing.T) {
 			},
 			wantExitCode: 1,
 		},
+		{
+			name: "main_in_nested_directory",
+			files: map[string]string{
+				"folder/main.py": "",
+			},
+			wantExitCode: 1,
+		},
 	}
 	for _, tc := range testCases {
 		t.Run(tc.name, func(t *testing.T) {