Skip to content

Commit 07fbe83

Browse files
authored
Merge pull request #104 from kif/99_export_HDF5_to_ZIP
use sample basename for zipped files as well.
2 parents 0b003fd + 9e6b366 commit 07fbe83

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/freesas/app/extract_ascii.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
__author__ = "Jérôme Kieffer"
2828
__license__ = "MIT"
2929
__copyright__ = "2020-2024, ESRF"
30-
__date__ = "16/04/2025"
30+
__date__ = "22/04/2025"
3131

3232
import io
3333
import os
@@ -362,16 +362,18 @@ def main():
362362
for src in files:
363363
print(f"{src} \t --> ", end="")
364364
if args.all:
365-
dest = os.path.splitext(src)[0] + "%04i.dat"
365+
dest = os.path.splitext(src)[0] + "_%04i.dat"
366366
for idx, frame in enumerate(extract_all(src)):
367367
print(src, " --> ", dest % idx)
368368
write_ascii(frame, dest % idx)
369369
print(dest)
370370
elif args.zip:
371-
dest = os.path.splitext(src)[0] + ".zip"
371+
base = os.path.splitext(src)[0]
372+
dest = base + ".zip"
373+
destz = base + "_%04i.dat"
372374
with zipfile.ZipFile(dest, "w") as z:
373375
for idx, frame in enumerate(extract_all(src)):
374-
z.writestr(f'frame_{idx:04d}.dat', write_ascii(frame))
376+
z.writestr(destz % idx, write_ascii(frame))
375377
print(dest)
376378
else:
377379
dest = os.path.splitext(src)[0] + ".dat"

0 commit comments

Comments
 (0)