modify containerEnv programmatically

Author: leehuwuj

packages/create-llama/helpers/devcontainer.ts

@@ -11,6 +11,7 @@ function renderDevcontainerContent(
     fs.readFileSync(path.join(templatesDir, "devcontainer.json"), "utf8"),
   );
 
+  // Modify postCreateCommand
   if (frontend) {
     devcontainerJson.postCreateCommand =
       framework === "fastapi"
@@ -20,6 +21,22 @@ function renderDevcontainerContent(
     devcontainerJson.postCreateCommand =
       framework === "fastapi" ? "poetry install" : "npm install";
   }
+
+  // Modify containerEnv
+  if (framework === "fastapi") {
+    if (frontend) {
+      devcontainerJson.containerEnv = {
+        ...devcontainerJson.containerEnv,
+        PYTHONPATH: "${PYTHONPATH}:${workspaceFolder}/backend",
+      };
+    } else {
+      devcontainerJson.containerEnv = {
+        ...devcontainerJson.containerEnv,
+        PYTHONPATH: "${PYTHONPATH}:${workspaceFolder}",
+      };
+    }
+  }
+
   return JSON.stringify(devcontainerJson, null, 2);
 }
 

packages/create-llama/templates/devcontainer.json

@@ -29,8 +29,7 @@
     }
   },
   "containerEnv": {
-    "POETRY_VIRTUALENVS_CREATE": "false",
-    "PYTHONPATH": "${PYTHONPATH}:${workspaceFolder}/backend"
+    "POETRY_VIRTUALENVS_CREATE": "false"
   },
   "forwardPorts": [3000, 8000]
 }