diff options
author | Franck Cuny <franck@fcuny.net> | 2024-05-04 13:27:23 -0700 |
---|---|---|
committer | Franck Cuny <franck@fcuny.net> | 2024-05-04 13:28:46 -0700 |
commit | 4a86b5122c3c1bfcf442a59287da809248406526 (patch) | |
tree | 4ab5ec2af31159284c36765a54981fd2ceb28991 /src | |
parent | add a comment to set environment variables (diff) | |
download | world-4a86b5122c3c1bfcf442a59287da809248406526.tar.gz |
start the conversion to python
Add support to move all the tools to python and organize them in a way that should be simple. I'm using https://rye-up.com for this.
Diffstat (limited to 'src')
-rwxr-xr-x | src/cli/pizza.py | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/src/cli/pizza.py b/src/cli/pizza.py new file mode 100755 index 0000000..d07abc8 --- /dev/null +++ b/src/cli/pizza.py @@ -0,0 +1,54 @@ +#!/usr/bin/env python3 + +from textwrap import fill + +import requests +from bs4 import BeautifulSoup +from termcolor import colored +from prettytable import PrettyTable + +url = "https://cheeseboardcollective.coop/pizza/" + + +def main(): + content = requests.get(url) + soup = BeautifulSoup(content.text, "html.parser") + + magic_ingredients = ["corn"] + + table = PrettyTable() + table.field_names = ["date", "pizza"] + table.align = "l" + + pizzas = soup.select(".pizza-list > article") + for pizza in pizzas: + date_color = "white" + menu_color = "white" + + date = pizza.find(class_="date").text + + # the pizza is the 1st element, the 2nd is the salad, and i don't + # care about the salad :) + menu = pizza.select(".menu > p:nth-of-type(1)")[0].get_text(strip=True, separator="\n").split("\n") + + if "closed" in menu[0]: + date_color = "red" + + if "Parbake pizza is available" in menu[0]: + menu.pop(0) + + if "Lunch from" in menu[0]: + menu.pop(0) + + if "No hot pizza today" in menu[0]: + menu.pop(0) + + final_menu = "".join(menu) + + for ingredient in magic_ingredients: + if ingredient in final_menu: + menu_color = "yellow" + + table.add_row([colored(date, date_color), fill(colored(final_menu, menu_color), width=80)]) + + print(table) |