day 2 and 3

Author: jung-chang

.gitignore

@@ -0,0 +1 @@
+__pycache__

2021/01/main.py

@@ -1,22 +1,15 @@
 # https://adventofcode.com/2021/day/1
 
-import os
-import pathlib
+from utils import read_input_gen
 
-
-def read_input():
-    with open(
-        os.path.join(pathlib.Path(__file__).parent.resolve(), "input.txt"), "r"
-    ) as file:
-        for line in file:
-            yield int(line.strip())
+file_path = "2021/01/input.txt"
 
 
 def part_one():
     increment_count = 0
     last_number = None
-    for line in read_input():
-        next_number = line
+    for line in read_input_gen(file_path):
+        next_number = int(line)
         if last_number is None:
             last_number = next_number
             continue
@@ -29,11 +22,10 @@ def part_one():
 def part_two():
     increment_count = 0
     num_list = []
-    for num in read_input():
+    for num in read_input_gen(file_path):
         if len(num_list) < 4:
-            num_list.append(num)
+            num_list.append(int(num))
         if len(num_list) == 4:
-            print(num_list)
             if sum(num_list[:3]) < sum(num_list[1:4]):
                 increment_count += 1
             num_list.pop(0)