Add cli.py
This commit is contained in:
parent
8be9505510
commit
9f2ef7a920
109
utils/cli.py
Normal file
109
utils/cli.py
Normal file
@ -0,0 +1,109 @@
|
|||||||
|
#!/usr/bin/python3
|
||||||
|
import requests
|
||||||
|
from bs4 import BeautifulSoup
|
||||||
|
from optparse import OptionParser
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
with open("utils/cookie.txt", "r") as f:
|
||||||
|
cookie = f.read()
|
||||||
|
|
||||||
|
session = requests.Session()
|
||||||
|
session.cookies["session"] = cookie
|
||||||
|
|
||||||
|
def download_input(year, day):
|
||||||
|
r = session.get(f"https://adventofcode.com/{year}/day/{int(day)}/input")
|
||||||
|
with open(f"{year}/inputs/day{str(day).zfill(2)}.txt", "w") as f:
|
||||||
|
f.write(r.content.decode())
|
||||||
|
print("ok")
|
||||||
|
|
||||||
|
def show_puzzle(year, day, part=None):
|
||||||
|
r = session.get(f"https://adventofcode.com/{year}/day/{int(day)}")
|
||||||
|
soup = BeautifulSoup(r.content, "html.parser")
|
||||||
|
day_desc = soup.find_all("article", {"class": "day-desc"})
|
||||||
|
if part is None:
|
||||||
|
for i in range(len(day_desc)):
|
||||||
|
print(day_desc[i].text)
|
||||||
|
return
|
||||||
|
if part > len(day_desc) or part <= 0:
|
||||||
|
print(f"Partie {part} non disponible")
|
||||||
|
return
|
||||||
|
print(day_desc[part-1].text)
|
||||||
|
|
||||||
|
def submit(year, day, answer, level=1):
|
||||||
|
r = session.post(f"https://adventofcode.com/{year}/day/{int(day)}/answer", data={"answer": answer, "level": level})
|
||||||
|
known_msgs = [
|
||||||
|
"That's the right answer!",
|
||||||
|
"That's not the right answer",
|
||||||
|
"You gave an answer too recently",
|
||||||
|
"You don't seem to be solving the right level. Did you already complete it?"
|
||||||
|
]
|
||||||
|
for msg in known_msgs:
|
||||||
|
if msg in r.text:
|
||||||
|
print(msg)
|
||||||
|
return
|
||||||
|
with open(f"answer-test-{year}-{int(day)}-{level}", "w") as f:
|
||||||
|
f.write(r.content.decode())
|
||||||
|
|
||||||
|
print("ok ?")
|
||||||
|
|
||||||
|
|
||||||
|
def __main__(options, args):
|
||||||
|
if len(args) == 0:
|
||||||
|
print("Une action doit être spécifiée")
|
||||||
|
return
|
||||||
|
if args[0] == "download" or args[0] == "dl":
|
||||||
|
download_input(options.year, options.day)
|
||||||
|
elif args[0] == "show" or args[0] == "sh":
|
||||||
|
if options.level == "":
|
||||||
|
options.level = None
|
||||||
|
else:
|
||||||
|
options.level = int(options.level)
|
||||||
|
show_puzzle(options.year, options.day, options.level)
|
||||||
|
elif args[0] == "submit" or args[0] == "sb":
|
||||||
|
if len(args) < 2:
|
||||||
|
answer = input("Answer ?\n>> ")
|
||||||
|
else:
|
||||||
|
answer = args[1]
|
||||||
|
if options.level == "":
|
||||||
|
options.level = 1
|
||||||
|
submit(options.year, options.day, answer, options.level)
|
||||||
|
else:
|
||||||
|
print(f"Invalid action {args[0]}")
|
||||||
|
result = 0
|
||||||
|
|
||||||
|
|
||||||
|
today = datetime.datetime.now()
|
||||||
|
|
||||||
|
USAGE = "usage: %prog (download|dl|show|sh|submit|sb) [options]"
|
||||||
|
parser = OptionParser(usage=USAGE)
|
||||||
|
parser.add_option(
|
||||||
|
"-y",
|
||||||
|
"--year",
|
||||||
|
dest="year",
|
||||||
|
help="Année",
|
||||||
|
action="store",
|
||||||
|
default=today.year
|
||||||
|
)
|
||||||
|
|
||||||
|
parser.add_option(
|
||||||
|
"-d",
|
||||||
|
"--day",
|
||||||
|
dest="day",
|
||||||
|
help="Jour",
|
||||||
|
action="store",
|
||||||
|
default=today.day
|
||||||
|
)
|
||||||
|
|
||||||
|
parser.add_option(
|
||||||
|
"-l",
|
||||||
|
"--level",
|
||||||
|
dest="level",
|
||||||
|
help="Niveau de la difficulté",
|
||||||
|
action="store",
|
||||||
|
default=""
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__" :
|
||||||
|
(OPTIONS, args) = parser.parse_args()
|
||||||
|
__main__(OPTIONS, args)
|
@ -10,6 +10,9 @@ DAY="$1"
|
|||||||
YEAR="$2"
|
YEAR="$2"
|
||||||
FILENAME="${YEAR}/day${DAY}.py"
|
FILENAME="${YEAR}/day${DAY}.py"
|
||||||
|
|
||||||
|
python utils/cli.py download -y $YEAR -d $DAY
|
||||||
|
python utils/cli.py show -y $YEAR -d $DAY
|
||||||
|
|
||||||
cp utils/template.py "$FILENAME"
|
cp utils/template.py "$FILENAME"
|
||||||
sed -i "s/DAY/${DAY}/" "$FILENAME"
|
sed -i "s/DAY/${DAY}/" "$FILENAME"
|
||||||
sed -i "s/YEAR/${YEAR}/" "$FILENAME"
|
sed -i "s/YEAR/${YEAR}/" "$FILENAME"
|
||||||
|
@ -24,3 +24,6 @@ def main():
|
|||||||
sample = read_sample()
|
sample = read_sample()
|
||||||
print(f"part1: {part1(sample)}")
|
print(f"part1: {part1(sample)}")
|
||||||
print(f"part2: {part2(sample)}")
|
print(f"part2: {part2(sample)}")
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
Loading…
Reference in New Issue
Block a user