Improve logging

This commit is contained in:
Noe Brucy
2022-09-02 16:16:17 +02:00
parent f1eb57a2b0
commit 6e0f5b4297
3 changed files with 40 additions and 45 deletions
+18 -15
View File
@@ -313,7 +313,7 @@ class SnapshotProcessor(HDF5Container):
self.path = path
self.run = os.path.basename(path)
self.num = num
self.log_id = "{}, {}".format(self.run, self.num)
self.log_id = "snap({}, {})".format(self.run, self.num)
super(SnapshotProcessor, self).__init__(path, path_out, params, tag)
@@ -400,8 +400,9 @@ class SnapshotProcessor(HDF5Container):
try:
self.init_pymses()
except FileNotFoundError:
self._log("Pymses not initialized", "WARNING")
except:
self.logger.error("Pymses not initialized", exc_info=1)
self.def_rules()
@@ -483,19 +484,21 @@ class SnapshotProcessor(HDF5Container):
far_cut_depth=distance,
up_vector=ax_v,
map_max_size=self.params.pymses.map_size,
)
)
self.open()
if "/maps" not in self.save:
self.save.create_group("/", "maps", "2D maps")
if "/hist" not in self.save:
self.save.create_group("/", "hist", "Histograms")
if "/datasets" not in self.save:
self.save.create_group("/", "datasets", "Complex datasets")
self.save.root.maps._v_attrs.center = center
self.save.root.maps._v_attrs.radius = self._radius
self.save.root.maps._v_attrs.im_extent = im_extent
self.close()
# Initialize HDF5 group
try:
self.open()
if "/maps" not in self.save:
self.save.create_group("/", "maps", "2D maps")
self.save.root.maps._v_attrs.center = center
self.save.root.maps._v_attrs.radius = self._radius
self.save.root.maps._v_attrs.im_extent = im_extent
except:
self.logger.error("Error in HDF5", exc_info=1)
raise
finally:
self.close()
def load_data(self, points_src, filename, save, keys=None):
"""