Add func to retrieve input file

Author: derailed-dash

.env

@@ -1,2 +1,2 @@
-input.txt filter=git-crypt diff=git-crypt
+input*.txt filter=git-crypt diff=git-crypt
 .env filter=git-crypt diff=git-crypt

.gitattributes

@@ -32,9 +32,13 @@
 from itertools import combinations
 import common.aoc_commons as td
 
+YEAR = 2015
+DAY = 24
+
 locations = td.get_locations(__file__)
 logger = td.retrieve_console_logger(locations.script_name)
 logger.setLevel(logging.INFO)
+td.write_puzzle_input_file(YEAR, DAY, locations.input_file)
 
 def main():
     # with open(locations.sample_input_file, mode="rt") as f:

src/AoC_2015/d24_sleigh_balance_subset_sum/input/input.txt

@@ -17,6 +17,8 @@
 import logging
 import os
 from pathlib import Path
+from dotenv import load_dotenv
+import requests
 from colorama import Fore
 
 #################################################################
@@ -116,7 +118,40 @@ def get_locations(script_file):
     input_file = Path(input_dir, "input.txt")
     sample_input_file = Path(script_dir, "input/sample_input.txt")
 
-    return Locations(script_name, script_dir, input_dir, output_dir, sample_input_file, input_file)
+    return Locations(script_name, script_dir, 
+                     input_dir, 
+                     output_dir, 
+                     sample_input_file, input_file)
+
+##################################################################
+# Retrieving input data
+##################################################################
+
+def write_puzzle_input_file(year: int, day: int, input_file):
+    if os.path.exists(input_file):
+        logger.debug("%s already exists", os.path.basename(input_file))
+        return
+
+    load_dotenv(os.path.join(os.path.dirname(__file__), '../../.env'))
+    SESSION_COOKIE = os.getenv('AOC_SESSION_COOKIE')
+    if SESSION_COOKIE:
+        logger.info('Session cookie retrieved.')
+    else:
+        logger.error('Failed to retrieve session cookie')
+        
+    url = f"https://adventofcode.com/{year}/day/{day}/input"
+    cookies = {"session": SESSION_COOKIE}
+    response = requests.get(url, cookies=cookies)
+    data = ""
+    
+    if response.status_code == 200:
+        data = response.text
+    else:
+        data = f"Failed to retrieve puzzle input: {response.status_code}"
+    
+    logger.debug("Writing %s", os.path.basename(input_file))
+    with open(input_file, 'w') as file:
+        file.write(data)
 
 #################################################################
 # POINTS, VECTORS AND GRIDS

src/AoC_2015/d24_sleigh_balance_subset_sum/sleigh_balance.py

@@ -13,10 +13,14 @@
 import time
 import common.aoc_commons as td
 
+YEAR = 2017
+DAY = 1
+
 locations = td.get_locations(__file__)
 logger = td.retrieve_console_logger(locations.script_name)
 logger.setLevel(logging.DEBUG)
 # td.setup_file_logging(logger, locations.output_dir)
+td.write_puzzle_input_file(YEAR, DAY, locations.input_file)
 
 def main():
     with open(locations.sample_input_file, mode="rt") as f: