Fixed UUID keys

This commit is contained in:
Joel Collins 2020-01-14 15:05:42 +00:00
parent 26e719feff
commit 1576bb7553

View file

@ -77,15 +77,16 @@ class ZipManager:
update_task_progress(int((index / n_files) * 100)) update_task_progress(int((index / n_files) * 100))
session_id = uuid.uuid4() session_id = uuid.uuid4()
session_key = str(session_id)
# self.session_zips[session_id] = fp # self.session_zips[session_id] = fp
self.session_zips[session_id] = { self.session_zips[session_key] = {
"id": session_id, "id": session_id,
"fp": fp, "fp": fp,
"data_size": data_size_megabytes, "data_size": data_size_megabytes,
"zip_size": os.path.getsize(fp.name) * 1e-6, "zip_size": os.path.getsize(fp.name) * 1e-6,
} }
return self.session_zips[session_id] return self.session_zips[session_key]
def zip_from_id(self, session_id): def zip_from_id(self, session_id):
return self.session_zips[session_id]["fp"] return self.session_zips[session_id]["fp"]