Skip to content

Commit 132a93d

Browse files
committed
Add function for setup of file version
1 parent e99bc00 commit 132a93d

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

can/io/trc.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -310,8 +310,15 @@ def __init__(
310310
self.header_written = False
311311
self.msgnr = 0
312312
self.first_timestamp = None
313-
self.file_version = TRCFileVersion.V2_1
314-
self._msg_fmt_string = self.MESSAGE_FORMAT_MAP[self.file_version]
313+
self._setup_file_version(TRCFileVersion.V2_1)
314+
315+
def _setup_file_version(self, file_version: Union[int, TRCFileVersion]):
316+
try:
317+
self.file_version = TRCFileVersion(file_version)
318+
self._msg_fmt_string = self.MESSAGE_FORMAT_MAP[self.file_version]
319+
except (KeyError, ValueError) as exc:
320+
err_msg = f"File version is not supported: {file_version}"
321+
raise NotImplementedError(err_msg) from exc
315322

316323
def _write_header_v1_0(self, start_time: datetime) -> None:
317324
lines = [

0 commit comments

Comments
 (0)