Advent_of_code/Input_helper.py at main · mhytrek/Advent_of_code · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import requests
from bs4 import BeautifulSoup

#Here paste your data
COOKIES_VALUE = ""
USER_AGENT = "Mozilla"
YEAR = 2024

def download_input(day:int):
    global COOKIES_VALUE
    global USER_AGENT
    url = f"https://adventofcode.com/{YEAR}/day/{day}/input"
    r = requests.get(url, cookies={'session': COOKIES_VALUE}, headers={'User-Agent': USER_AGENT})
    open(f"input_day_{day}", 'wb').write(r.content)

def get_input(day:int):
    try:
        file = open(f"input_day_{day}")
    except FileNotFoundError:
        download_input(day)
        file = open(f"input_day_{day}")
    lines = file.readlines()
    return lines

def get_soup(day:int):
    global COOKIES_VALUE
    global USER_AGENT
    url = f"https://adventofcode.com/{YEAR}/day/{day}"
    page = requests.get(url, cookies={'session': COOKIES_VALUE}, headers={'User-Agent': USER_AGENT})
    soup = BeautifulSoup(page.text, features="html.parser")
    return soup

def get_example(day:int):
    soup = get_soup(day)
    soup.find_all("<p>For example</p>")
    code = soup.find_all("code")
    return str(code[0])[len("<code>"):-len("</code>")]

def check_example(day:int, task:int, answer:int):
    soup = get_soup(day)
    if task != 2 and task != 1:
        print("Wrong task number")
        return
    elif task == 2:
        part2 = soup.find_all('article')[1]
        code = part2.find_all("code")
        result = str(code[-1])[len("<code>" + "<em>"):-len("</em>"+"</code>")]
    else:
        part1 = soup.find_all('article')[0]
        code = part1.find_all("code")
        result = str(code[-1])[len("<code>" + "<em>"):-len("</em>" + "</code>")]
    if result == str(answer):
        return "It is working!!"
    else:
        return "Try again"