Skip to content

Commit fcdb78c

Browse files
committed
Merge branch 'main' of github.com:emagedoc/CodeClash
2 parents a055bfb + 8ccd2c3 commit fcdb78c

6 files changed

Lines changed: 204 additions & 153 deletions

File tree

aws/setup/batch/environment.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
"maxvCpus": 32,
1717
"desiredvCpus": 0,
1818
"instanceTypes": [
19-
"optimal"
19+
"m5.large"
2020
],
2121
"subnets": [
2222
"subnet-02f03254ff2613e05",

aws/setup/batch/push.py

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -196,11 +196,26 @@ def push_job_definition(job_data: dict, batch_client) -> None:
196196

197197

198198
def _add_user_data_to_launch_template(template_data: dict, user_data_path: Path) -> None:
199-
"""Add user data from the specified path to the template data."""
199+
"""Add user data from the specified path to the template data in MIME multipart format."""
200200
if user_data_path.exists():
201201
logger.info(f"Adding user data from {user_data_path.name}")
202202
user_data_script = user_data_path.read_text()
203-
encoded_user_data = base64.b64encode(user_data_script.encode()).decode()
203+
204+
# Format as MIME multipart for cloud-init
205+
mime_data = f"""Content-Type: multipart/mixed; boundary="===============BOUNDARY=="
206+
MIME-Version: 1.0
207+
208+
--===============BOUNDARY==
209+
Content-Type: text/x-shellscript; charset="us-ascii"
210+
MIME-Version: 1.0
211+
Content-Transfer-Encoding: 7bit
212+
Content-Disposition: attachment; filename="userdata.sh"
213+
214+
{user_data_script}
215+
--===============BOUNDARY==--
216+
"""
217+
218+
encoded_user_data = base64.b64encode(mime_data.encode()).decode()
204219
template_data["LaunchTemplateData"]["UserData"] = encoded_user_data
205220
else:
206221
logger.warning(f"User data file not found: {user_data_path}")

0 commit comments

Comments
 (0)