Skip to content

Commit 35d5744

Browse files
committed
fix get LogEntry
1 parent 5a5197e commit 35d5744

2 files changed

Lines changed: 30 additions & 5 deletions

File tree

native_fisher_py/python/native_fisher_py/data/classes.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -702,6 +702,22 @@ def length(self): return len(self._values)
702702
@property
703703
def values(self): return self._values
704704

705+
def keys(self):
706+
return [l.strip().rstrip(':') for l in self._labels]
707+
708+
def __getitem__(self, key):
709+
clean_key = key.strip().rstrip(':')
710+
for i, label in enumerate(self._labels):
711+
if label.strip().rstrip(':') == clean_key:
712+
return self._values[i]
713+
raise KeyError(key)
714+
715+
def get(self, key, default=None):
716+
try:
717+
return self[key]
718+
except KeyError:
719+
return default
720+
705721
class HeaderItem(CommonCoreDataObject):
706722
def __init__(self, data):
707723
if "###TYPE###" in data:

native_fisher_py/python/native_fisher_py/raw_file.py

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -246,23 +246,32 @@ def get_instrument_count_of_type(self, device_type):
246246

247247
def get_trailer_extra_information(self, scan_number):
248248
from .data.classes import LogEntry
249-
return LogEntry(get_trailer_extra_values(scan_number))
249+
labels = [h.label for h in self.get_trailer_extra_header_information()]
250+
return LogEntry(get_trailer_extra_values(scan_number), labels)
251+
250252
def get_trailer_extra_header_information(self):
251253
from .data.classes import HeaderItem
252254
return [HeaderItem(h) for h in get_trailer_extra_header()]
253-
def get_trailer_extra_values(self, scan_number, formatted):
255+
256+
def get_trailer_extra_values(self, scan_number, formatted=False):
254257
return get_trailer_extra_values(scan_number)
258+
255259
def get_status_log_header_information(self):
256260
from .data.classes import HeaderItem
257261
return [HeaderItem(h) for h in get_status_log_header()]
258-
def get_status_log_values(self, scan_number, formatted):
262+
263+
def get_status_log_values(self, scan_number, formatted=False):
259264
from .data.classes import LogEntry
260-
return LogEntry(get_status_log_values(scan_number))
265+
labels = [h.label for h in self.get_status_log_header_information()]
266+
return LogEntry(get_status_log_values(scan_number), labels)
267+
261268
def get_status_log_entries_count(self):
262269
return get_status_log_count()
270+
263271
def get_status_log_for_retention_time(self, rt):
264272
from .data.classes import LogEntry
265-
return LogEntry(get_status_log_values_for_rt(rt))
273+
labels = [h.label for h in self.get_status_log_header_information()]
274+
return LogEntry(get_status_log_values_for_rt(rt), labels)
266275
def get_tune_data_count(self):
267276
return get_tune_data_count()
268277
def get_tune_data(self, index): return None

0 commit comments

Comments
 (0)