Initial release

get_data.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+import sys
+import requests
+from datetime import datetime
+from pytz import timezone
+from time import sleep
+from pathlib import Path
+
+# Eric Wastl, the author of Advent-of-Code, asked that if you're automatically
+# querying adventofcode.com (as this script does), then you should include
+# contact details in the User-Agent field. So please replace my name with yours:
+NAME = "Aurélien Geron"
+
+def usage():
+    print("Usage:")
+    print(f"{sys.argv[0]} {{year}} {{day}}")
+    sys.exit(1)
+
+def session_error():
+    print("""Please open your browser, login to adventofcode.com, lookup 
+the session cookie, and save its value to the .session file.
+Here's how to find this cookie in Chrome: right-click > Inspect, select the
+Application tab in the inspector, then in the left menu select
+Storage > Cookies > https://adventofcode.com, and click on session in the list.
+Copy the cookie value: it's a long hexadecimal .""")
+    sys.exit(2)
+
+if len(sys.argv) != 3:
+    usage()
+
+try:
+    year, day = map(int, sys.argv[1:3])
+except ValueError:
+    usage()
+
+session_path = Path(".session")
+if not session_path.is_file():
+    session_error()
+
+eastern = timezone('US/Eastern')
+requested_day = eastern.localize(datetime(year, 12, day))
+
+while True:
+    now = datetime.now(eastern)
+    seconds_left = (requested_day - now).total_seconds()
+    if seconds_left > 0:
+        wait_time = seconds_left % 1
+        if wait_time < 0.5:
+            wait_time += 1
+        sleep(wait_time)
+        print(f"⏰ {int(seconds_left)} ", end="\r");
+    else:
+        break
+
+sleep(1) # just to be safe
+
+url = f"https://adventofcode.com/{year}/day/{day}/input"
+cookies = {"session": open(".session").read().strip()}
+headers = {"User-Agent": NAME}
+request = requests.get(url, cookies=cookies, headers=headers)
+if request.status_code != 200:
+    print(f"HTTP Error {request.status_code}")
+    print(request.text)
+    session_error()
+
+text = request.text
+
+if year == 2023:
+    with open(f"data/day{day:02}.txt", "w") as f:
+        f.write(text)
+else:
+    print(text)