From d75a9d510ced0a29ad2fe7ebc247dc7b0766320c Mon Sep 17 00:00:00 2001
From: Tomasz Kramkowski <tk@the-tk.com>
Date: Fri, 3 Dec 2021 16:56:16 +0000
Subject: Implement open_day utility function

---
 utils.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'utils.py')

diff --git a/utils.py b/utils.py
index d322ed0..91b90ff 100644
--- a/utils.py
+++ b/utils.py
@@ -1,6 +1,7 @@
 from collections import deque
 from collections.abc import Iterable, Iterator, Generator
 from itertools import islice
+from sys import argv
 from typing import TypeVar
 
 T = TypeVar('T')
@@ -13,3 +14,8 @@ def sliding_window(iterable: Iterable[T], n: int) -> Generator[tuple[T, ...], No
     for x in it:
         window.append(x)
         yield tuple(window)
+
+def open_day(n: int):
+    if len(argv) == 2:
+        return open(argv[1])
+    return open(f'{n}.in')
-- 
cgit v1.2.3-54-g00ecf