Skip to content

Commit e8af515

Browse files
committed
improve expand_zip
1 parent 63094c4 commit e8af515

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

util/utils.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,6 @@ def expand_zip(zipped, fof_type=None, member_ids=None, member_type=None):
301301
expanded = []
302302

303303
for items in zipped:
304-
305304
try:
306305
file_info = FileInfo(items[0])
307306
file_type = getattr(file_info, "type", None)
@@ -319,18 +318,24 @@ def expand_zip(zipped, fof_type=None, member_ids=None, member_type=None):
319318
member_values_expanded = []
320319

321320
if file_type is FileType.STATS and member_type_list:
322-
for m_id in member_values:
323-
for m_type in member_type_list:
324-
member_values_expanded.append(f"{m_type}_{m_id}")
321+
if member_values:
322+
323+
for m_id in member_values:
324+
for m_type in member_type_list:
325+
member_values_expanded.append(f"{m_type}_{m_id}")
326+
else:
327+
328+
member_values_expanded = member_type_list.copy()
325329
else:
326-
member_values_expanded = member_values.copy()
330+
member_values_expanded = member_values.copy() or member_type_list.copy()
327331

328332
for fof_val in fof_values:
329333
for member_val in member_values_expanded:
330334
formatted = [
331335
item.format(
332336
fof_type=fof_val or "{fof_type}",
333337
member_id=member_val or "{member_id}",
338+
member_type="{member_type}",
334339
)
335340
for item in items
336341
]

0 commit comments

Comments
 (0)