From 39c9426f71129a285ba04ddfca204d2edb903aa8 Mon Sep 17 00:00:00 2001 From: Chris Smith Date: Thu, 2 Jul 2026 23:07:35 -0400 Subject: [PATCH] Add parse_datafile function --- util/reader.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/util/reader.py b/util/reader.py index 740de5f..82f0e2c 100644 --- a/util/reader.py +++ b/util/reader.py @@ -1,5 +1,17 @@ from pathlib import Path +def parse_datafile(data_file: Path, output_format: str) -> str | list: + with open(data_file, 'r', encoding="utf-8") as data_file: + match output_format: + case 'string': + return data_file.read().strip() + case 'list': + return data_file.read().strip().splitlines() + case 'int_list': + return [int(line) for line in data_file.read().strip().splitlines()] + return [] + + def data_as_string(data_file: Path) -> str: with open(data_file) as file: return file.read().strip()