Skip to content

Commit 9e12f8a

Browse files
author
Benjamin Moody
committed
Record, MultiRecord: allow base_datetime as constructor argument.
For convenience and consistency, allow passing base_datetime to the BaseRecord, Record, or MultiRecord constructor. Since this attribute is an alias for base_date and base_time, it is an error for the caller to specify this argument while simultaneously specifying base_date or base_time.
1 parent 80ce12a commit 9e12f8a

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

wfdb/io/record.py

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,7 @@ def __init__(
208208
sig_len=None,
209209
base_time=None,
210210
base_date=None,
211+
base_datetime=None,
211212
comments=None,
212213
sig_name=None,
213214
):
@@ -217,8 +218,19 @@ def __init__(
217218
self.counter_freq = counter_freq
218219
self.base_counter = base_counter
219220
self.sig_len = sig_len
220-
self.base_time = base_time
221-
self.base_date = base_date
221+
if base_datetime is not None:
222+
if base_time is not None:
223+
raise TypeError(
224+
"cannot specify both base_time and base_datetime"
225+
)
226+
if base_date is not None:
227+
raise TypeError(
228+
"cannot specify both base_date and base_datetime"
229+
)
230+
self.base_datetime = base_datetime
231+
else:
232+
self.base_time = base_time
233+
self.base_date = base_date
222234
self.comments = comments
223235
self.sig_name = sig_name
224236

@@ -687,6 +699,7 @@ def __init__(
687699
sig_len=None,
688700
base_time=None,
689701
base_date=None,
702+
base_datetime=None,
690703
file_name=None,
691704
fmt=None,
692705
samps_per_frame=None,
@@ -716,6 +729,7 @@ def __init__(
716729
sig_len=sig_len,
717730
base_time=base_time,
718731
base_date=base_date,
732+
base_datetime=base_datetime,
719733
comments=comments,
720734
sig_name=sig_name,
721735
)
@@ -972,6 +986,7 @@ def __init__(
972986
sig_len=None,
973987
base_time=None,
974988
base_date=None,
989+
base_datetime=None,
975990
seg_name=None,
976991
seg_len=None,
977992
comments=None,
@@ -988,6 +1003,7 @@ def __init__(
9881003
sig_len=sig_len,
9891004
base_time=base_time,
9901005
base_date=base_date,
1006+
base_datetime=base_datetime,
9911007
comments=comments,
9921008
sig_name=sig_name,
9931009
)

0 commit comments

Comments
 (0)