From 7ee24173133cb49e12489df6c5d383011151ab10 Mon Sep 17 00:00:00 2001
From: Jake Pullen <Jake.Pullen@theaccessgroup.com>
Date: Wed, 6 Dec 2023 08:02:02 +0000
Subject: [PATCH] Add function to populate data for Advent of Code puzzle

---
 get_puzzle_text.py | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/get_puzzle_text.py b/get_puzzle_text.py
index 657b9b7..9c5c210 100644
--- a/get_puzzle_text.py
+++ b/get_puzzle_text.py
@@ -1,5 +1,6 @@
 import os
 import requests
+from datetime import datetime
 
 def get_puzzle_text(year, day):
     url = f"https://adventofcode.com/{year}/day/{day}"
@@ -8,7 +9,8 @@ def get_puzzle_text(year, day):
     return response.text
 
 def get_puzzle_input(year, day):
-    session_cookie = ''
+    with open(r'advent_of_code/session_cookie.txt', 'r') as file:
+        session_cookie = file.read().strip()
     url = f"https://adventofcode.com/{year}/day/{day}/input"
     cookies = {'session': session_cookie}
     response = requests.get(url, cookies=cookies)
@@ -39,10 +41,25 @@ def save_puzzle_input(year, day):
 
 
 # Usage example
-year = 2022
-day = 7
-save_puzzle_text(year, day)
-save_puzzle_input(year, day)
+current_year = datetime.now().year
+current_day = datetime.now().day
+
+def populate_data(year = current_year, day=current_day):
+    """
+    Populates data for the Advent of Code puzzle.
+    requires a session cookie to be saved in advent_of_code/session_cookie.txt
+
+    Args:
+        year (int): The year of the Advent of Code puzzle. Defaults to the current year.
+        day (int): The day of the Advent of Code puzzle. Defaults to the current day.
+
+    Returns:
+        None
+    """
+    save_puzzle_text(year, day)
+    save_puzzle_input(year, day)
+
+populate_data()