mirror of
https://github.com/triqs/dft_tools
synced 2024-06-23 05:32:17 +02:00
36 lines
1001 B
Python
36 lines
1001 B
Python
import h5py
|
|
import sys
|
|
import numpy
|
|
import subprocess
|
|
|
|
if len(sys.argv) < 2:
|
|
print "Usage: python update_archive.py old_archive"
|
|
sys.exit()
|
|
|
|
print """
|
|
This script is an attempt to update your archive to TRIQS 1.2.
|
|
Please keep a copy of your old archive as this script is
|
|
** not guaranteed ** to work for your archive.
|
|
If you encounter any problem please report it on github!
|
|
"""
|
|
|
|
filename = sys.argv[1]
|
|
A = h5py.File(filename)
|
|
|
|
old_to_new = {'SumK_LDA':'lda_input', 'SumK_LDA_ParProj':'lda_parproj_input',
|
|
'SymmCorr':'lda_symmcorr_input', 'SymmPar':'lda_symmpar_input', 'SumK_LDA_Bands':'lda_bands_input'}
|
|
|
|
for old, new in old_to_new.iteritems():
|
|
if old not in A.keys(): continue
|
|
print "Changing %s to %s ..."%(old, new)
|
|
A.copy(old,new)
|
|
del(A[old])
|
|
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])
|