Filip/analyze-sport-data
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Added heart rate offset parameters to TCX and FIT file extraction functions.

Browse source
This commit is contained in:
Filip 2025-03-16 14:53:47 +01:00
parent 0206867af5
commit e14d661571
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
analyze_sport_data/file_records.py
View file

@ -18,8 +18,11 @@ class FileRecords:
self.heart_rate_time_offset = heart_rate_time_offset
def from_fit_file(file_path: str,
friendly_name: str | None = None) -> FileRecords:
def from_fit_file(
file_path: str,
friendly_name: str | None = None,
heart_rate_time_offset: timedelta = timedelta()
) -> FileRecords:
stream = Stream.from_file(file_path)
decoder = Decoder(stream)
messages, _errors = decoder.read()
@ -30,11 +33,14 @@ def from_fit_file(file_path: str,
if "heart_rate" in record.keys() and record["heart_rate"] != None
]
return FileRecords(friendly_name if friendly_name != None else file_path,
heart_rate_records)
heart_rate_records, heart_rate_time_offset)
def from_tcx_file(file_path: str,
friendly_name: str | None = None) -> FileRecords:
def from_tcx_file(
file_path: str,
friendly_name: str | None = None,
heart_rate_time_offset: timedelta = timedelta()
) -> FileRecords:
empty_file_records = FileRecords(file_path, [])
heart_rate_records: list[HeartRateRecord] = []
@ -76,4 +82,4 @@ def from_tcx_file(file_path: str,
float(heart_rate_value.text)))
return FileRecords(friendly_name if friendly_name != None else file_path,
heart_rate_records)
heart_rate_records, heart_rate_time_offset)