Add puzzle text for day 6 and day 7
This commit is contained in:
@@ -0,0 +1,50 @@
|
||||
import os
|
||||
import requests
|
||||
|
||||
def get_puzzle_text(year, day):
|
||||
url = f"https://adventofcode.com/{year}/day/{day}"
|
||||
response = requests.get(url)
|
||||
response.raise_for_status()
|
||||
return response.text
|
||||
|
||||
def get_puzzle_input(year, day):
|
||||
session_cookie = ''
|
||||
url = f"https://adventofcode.com/{year}/day/{day}/input"
|
||||
cookies = {'session': session_cookie}
|
||||
response = requests.get(url, cookies=cookies)
|
||||
response.raise_for_status()
|
||||
return response.text
|
||||
|
||||
def save_puzzle_text(year, day):
|
||||
folder = rf"advent_of_code\{year}\day_{day}"
|
||||
os.makedirs(folder, exist_ok=True)
|
||||
input_file = os.path.join(folder, "puzzle_text.md")
|
||||
puzzle_text = get_puzzle_text(year, day)
|
||||
# Remove everything before the puzzle text which is all the things inside the <main> tag
|
||||
puzzle_text = puzzle_text.split('<main>')[1]
|
||||
# Remove everything after the puzzle text which is all the things after the </article> tag
|
||||
puzzle_text = puzzle_text.split('</article>')[0]
|
||||
print(f'Saving puzzle text to {input_file}')
|
||||
with open(input_file, "w") as file:
|
||||
file.write(puzzle_text)
|
||||
|
||||
def save_puzzle_input(year, day):
|
||||
folder = rf"advent_of_code\{year}\day_{day}"
|
||||
os.makedirs(folder, exist_ok=True)
|
||||
input_file = os.path.join(folder, "input.txt")
|
||||
puzzle_input = get_puzzle_input(year, day)
|
||||
print(f'Saving puzzle input to {input_file}')
|
||||
with open(input_file, "w") as file:
|
||||
file.write(puzzle_input)
|
||||
|
||||
|
||||
# Usage example
|
||||
year = 2022
|
||||
day = 7
|
||||
save_puzzle_text(year, day)
|
||||
save_puzzle_input(year, day)
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user