mirror of
https://github.com/alanorth/safari-bingo-cards.git
synced 2024-11-22 14:55:07 +01:00
Compare commits
2 Commits
b27380e81e
...
f3224bee2e
Author | SHA1 | Date | |
---|---|---|---|
f3224bee2e | |||
9a0acd3717 |
22
generate.py
22
generate.py
@ -25,7 +25,7 @@ def create_thumbnail(animal):
|
||||
# check if the file has been downloaded
|
||||
if not os.path.isfile(image_path):
|
||||
logger.error(
|
||||
Fore.YELLOW + f"> Missing image for {animal['common_name']}." + Fore.RESET
|
||||
Fore.RED + f"> Missing image for {animal['common_name']}." + Fore.RESET
|
||||
)
|
||||
# check if we already have a thumbnail
|
||||
elif os.path.isfile(thumbnail_path):
|
||||
@ -36,11 +36,6 @@ def create_thumbnail(animal):
|
||||
+ Fore.RESET
|
||||
)
|
||||
else:
|
||||
logger.info(
|
||||
Fore.GREEN
|
||||
+ f"> Creating thumbnail for {animal['common_name']}..."
|
||||
+ Fore.RESET
|
||||
)
|
||||
vips_image = pyvips.Image.new_from_file(image_path)
|
||||
# Crop to a 600x600 square using smartcrop to focus attention
|
||||
# See: https://stackoverflow.com/questions/47852390/making-a-huge-image-mosaic-with-pyvips
|
||||
@ -71,6 +66,13 @@ def create_thumbnail(animal):
|
||||
# conversion twice.
|
||||
vips_thumbnail.vipssave(thumbnail_path, strip=True)
|
||||
|
||||
logger.info(
|
||||
Fore.GREEN
|
||||
+ f"> Created thumbnail for {animal['common_name']}..."
|
||||
+ Fore.RESET
|
||||
)
|
||||
|
||||
|
||||
return
|
||||
|
||||
|
||||
@ -87,10 +89,6 @@ def download_image(animal):
|
||||
+ Fore.RESET
|
||||
)
|
||||
else:
|
||||
logger.info(
|
||||
Fore.GREEN + f"> Downloading {animal['common_name']}..." + Fore.RESET
|
||||
)
|
||||
|
||||
headers = {
|
||||
"User-Agent": "safari-bingo-cards-bot/0.1 (https://git.mjanja.ch/alanorth/safari-bingo-cards)"
|
||||
}
|
||||
@ -99,6 +97,10 @@ def download_image(animal):
|
||||
with open(image_path, "wb") as fd:
|
||||
for chunk in response:
|
||||
fd.write(chunk)
|
||||
|
||||
logger.info(
|
||||
Fore.GREEN + f"> Downloaded {animal['common_name']}..." + Fore.RESET
|
||||
)
|
||||
else:
|
||||
logger.error(
|
||||
Fore.RED
|
||||
|
Loading…
Reference in New Issue
Block a user