dece/AdventOfCode
1
0
Fork 0
Code Issues Pull requests Releases Activity

Make fetch script usable with different years

Browse source
This commit is contained in:
Dece 2019-12-22 01:28:04 +01:00
parent dc85aadc3a
commit e497884c30
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
2019/fetch.py → fetch.py
View file

@ -1,16 +1,18 @@
from datetime import datetime
import os import os
import requests import requests
import sys import sys
if len(sys.argv) != 2: if len(sys.argv) < 2:
print("Usage: <fetch> day") print("Usage: <fetch> day")
sys.exit() sys.exit()
day = sys.argv[1] day = sys.argv[1]
year = sys.argv[2] if len(sys.argv) > 2 else str(datetime.now().year)
URL = "https://adventofcode.com/2019/day/{}/input" URL = "https://adventofcode.com/{}/day/{}/input"
SESSION_ID = os.environ["AOC_SESSION"] SESSION_ID = os.environ["AOC_SESSION"]
response = requests.get(URL.format(day), cookies={"session": SESSION_ID}) response = requests.get(URL.format(year, day), cookies={"session": SESSION_ID})
response.raise_for_status() response.raise_for_status()
with open("day{}.txt".format(day), "wt") as output_file: with open("day{}.txt".format(day), "wt") as output_file: