From 52ce2afc40e599138d643fe04d42f9c149d91192 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gustav=20S=C3=B6rn=C3=A4s?= Date: Wed, 9 Dec 2020 08:14:27 +0100 Subject: common input --- 20/py/d08.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to '20/py/d08.py') diff --git a/20/py/d08.py b/20/py/d08.py index d435ea2..dd9535a 100644 --- a/20/py/d08.py +++ b/20/py/d08.py @@ -1,4 +1,5 @@ #!/usr/bin/env python3 +import aoc20 import sys @@ -46,9 +47,6 @@ def pt2(_in): if __name__ == "__main__": - if len(sys.argv) > 1: - input = open(sys.argv[1], "r").readlines() - else: - input = open("../input/08", "r").readlines() + input = aoc20.read_input(sys.argv[1:], 8) print(pt1(input)) print(pt2(input)) -- cgit v1.2.1