From 8578bfb86ae661900093e5e4ce962930c399e5b6 Mon Sep 17 00:00:00 2001
From: Yngve Inntjore Levinsen <Yngve.Levinsen@esss.se>
Date: Mon, 7 Dec 2015 13:59:56 +0100
Subject: [PATCH] add force flag to savetohdf by default false. if true
overwrite existing group
---
ess/TraceWin.py | 12 +++++++++---
1 file changed, 9 insertions(+), 3 deletions(-)
diff --git a/ess/TraceWin.py b/ess/TraceWin.py
index 8d7bead..c4c9f0f 100644
--- a/ess/TraceWin.py
+++ b/ess/TraceWin.py
@@ -740,15 +740,21 @@ class density_file:
self.Np=self._sum_merge(o,'Np')
self.Nrun+=o.Nrun
- def savetohdf(self,filename='Density.h5',group='TraceWin'):
+ def savetohdf(self,filename='Density.h5',group='TraceWin', force=False):
'''
Saves data to HDF5
'''
- import h5py
+ import h5py,sys
fout = h5py.File(filename,'a')
if group in fout:
- del fout[group]
+ if force:
+ del fout[group]
+ else:
+ if sys.flags.debug:
+ print("Group {} already exist in {}".format(group,filename))
+ return
+
group = fout.create_group(group)
# header attributes..
--
GitLab