mirror of
https://github.com/triqs/dft_tools
synced 2024-11-07 06:33:48 +01:00
26 lines
626 B
Python
26 lines
626 B
Python
import h5py
|
|
import sys
|
|
import subprocess
|
|
|
|
if len(sys.argv) < 2:
|
|
print "Usage: python clear_h5_output.py archive"
|
|
sys.exit()
|
|
|
|
print """
|
|
This script is to remove any SumkDFT generated output from the h5 archive
|
|
and to restore it to the original post-converter state.
|
|
"""
|
|
|
|
filename = sys.argv[1]
|
|
A = h5py.File(filename)
|
|
for group in ['dmft_output','user_data']:
|
|
if group in A: del(A[group])
|
|
A.close()
|
|
|
|
# Repack to reclaim disk space
|
|
retcode = subprocess.call(["h5repack","-i%s"%filename, "-otemphgfrt.h5"])
|
|
if retcode != 0:
|
|
print "h5repack failed!"
|
|
else:
|
|
subprocess.call(["mv","-f","temphgfrt.h5","%s"%filename])
|