Aurélien Geron
73 lines
2.0 KiB
Python
Executable File
73 lines
2.0 KiB
Python
Executable File
#!/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)
|