decibyte/frokostbot
decibyte/frokostbot
1
0
Fork 0
Code Pull requests Activity

Wrap translation call in try-except

Browse source
This commit is contained in:
Mikkel Munch Mortensen 2023-06-06 13:54:39 +02:00
parent a83382ded8
commit 17c56f957e
1 changed files with 23 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
frokostbot.py
View file

@ -31,10 +31,15 @@ except KeyError as e:
sys.exit(1)
sys.stdout.write("Configuration loaded.\n")
def get_photo_url(menu: str):
"""Retrieve a photo of the menu."""
# Translate menu from Danish to English.
translation = Translator().translate(menu, src="da", dest="en").text
try:
translation = Translator().translate(menu, src="da", dest="en").text
except TypeError:
sys.stderr.write("Unable to translate menu :(\n")
return None
sys.stdout.write(f"English translation of the menu is: {translation}\n")
# Ask Crayion to be creative.
@ -97,18 +102,14 @@ def get_photo_url(menu: str):
# Set up and execute request to SharePoint.
cookies = browser_cookie3.firefox(
domain_name="sharepoint.com"
)
cookies = browser_cookie3.firefox(domain_name="sharepoint.com")
headers = {
"Accept": "application/json;odata=verbose",
"Content-Type": "application/json;odata=verbose",
}
request_data = {
"parameters": {
"__metadata": {
"type": "SP.RenderListDataParameters"
},
"parameters": {
"__metadata": {"type": "SP.RenderListDataParameters"},
"AddRequiredFields": True,
"AllowMultipleValueFilterForTaxonomyFields": True,
"FilterOutChannelFoldersInDefaultDocLib": True,
@ -126,9 +127,9 @@ sys.stdout.write("Data retrieved from SharePoint.\n")
if response.status_code != 200:
hook_url = SLACK_ERROR_HOOK
photo = None
message = plain_message = (
f"SharePoint responded with status {response.status_code} :("
)
message = (
plain_message
) = f"SharePoint responded with status {response.status_code} :("
else:
# Extract today's menu from the SharePoint data.
today = datetime.date.today().strftime("%d-%m-%Y")
@ -190,6 +191,7 @@ else:
"cloud": ["sky"],
"potato": ["kartoffel", "kartofler"],
"apple": ["æble"],
"peanuts": ["peanut"],
"baguette_bread": ["flute"],
"flag-dk": ["tillykke", "fødselsdag"],
"wave": ["farvel"],
@ -207,13 +209,15 @@ else:
sys.stdout.write("No emojis determined.\n")
# Pick an introduction for the menu.
introduction = random.choice([
"Dagens menu er",
"I dag forkæler kantinen os med",
"Du kan godt glæde dig til senere! For vi skal have",
"Der bliver knoklet i køkkenet for at blive klar til at servere",
"Klokken 11:30 har kantinen fremtryllet en lækker omgang",
])
introduction = random.choice(
[
"Dagens menu er",
"I dag forkæler kantinen os med",
"Du kan godt glæde dig til senere! For vi skal have",
"Der bliver knoklet i køkkenet for at blive klar til at servere",
"Klokken 11:30 har kantinen fremtryllet en lækker omgang",
]
)
sys.stdout.write(f"Introduction picked: {introduction}\n")
# Retrieve a photo of the menu from Craiyon.
@ -249,7 +253,7 @@ if photo is not None:
"type": "plain_text",
"text": photo_text,
},
}
},
)
sys.stdout.write("Posting menu to Slack...\n")