Skip to content

Commit eadf747

Browse files
authored
Merge pull request #611 from USACE/fix/vulnerabilites_05-29-2026
packager fix
2 parents c7515af + 6f51195 commit eadf747

1 file changed

Lines changed: 4 additions & 7 deletions

File tree

async_packager/packager.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -33,17 +33,14 @@
3333
this = os.path.basename(__file__)
3434

3535

36-
def handle_message(message):
36+
def handle_message(message_body):
3737
package_file = None
3838
dst = None
3939
try:
4040
logger.info("%(spacer)s new message %(spacer)s" % {"spacer": "*" * 20})
41-
42-
# parse message to payload as json object and get the download id
43-
message_body = message.body
4441
logger.debug(f"{message_body=}")
4542

46-
download_id = json.loads(message.body)["id"]
43+
download_id = json.loads(message_body)["id"]
4744
logger.debug(f"Download ID: {download_id}")
4845

4946
# get the payload from the download endpoint with the download_id
@@ -147,7 +144,6 @@ def handle_message(message):
147144
if dst is not None and os.path.exists(dst.name):
148145
shutil.rmtree(dst.name, ignore_errors=True)
149146
dst = None
150-
message.delete()
151147

152148
return 0
153149

@@ -176,6 +172,7 @@ def handle_message(message):
176172
MaxNumberOfMessages=MAX_Q_MESSAGES, WaitTimeSeconds=WAIT_TIME_SECONDS
177173
)
178174
for message in messages:
179-
p = multiprocessing.Process(target=handle_message, args=(message,))
175+
p = multiprocessing.Process(target=handle_message, args=(message.body,))
180176
p.start()
181177
p.join()
178+
message.delete()

0 commit comments

Comments
 (0)