IsaacShoebottom/daily-wallpaper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

clean up a bit

Browse Source
This commit is contained in:
Isaac Shoebottom
2024-11-20 18:56:19 -04:00
parent 90dc4b78f1
commit 303219dfb1
4 changed files with 24 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
providers/wikimedia.py
View File

@ -1,6 +1,7 @@
import datetime
import logging
import re
import time
from providers._provider import Provider
# https://api.wikimedia.org/wiki/Feed_API/Reference/Featured_content
@ -13,19 +14,24 @@ class Wikimedia(Provider):
super().__init__(settings, session)
def get_image_info(self):
today = datetime.datetime.now()
date = today.strftime('%Y/%m/%d')
logging.debug(f"Date: {date}")
url = 'https://api.wikimedia.org/feed/v1/wikipedia/en/featured/' + date
logging.debug(f"URL: {url}")
# Since wikipedia API seems to fail to provide image url, we will retry on key error until we get the image url
try:
today = datetime.datetime.now()
date = today.strftime('%Y/%m/%d')
logging.debug(f"Date: {date}")
url = 'https://api.wikimedia.org/feed/v1/wikipedia/en/featured/' + date
logging.debug(f"URL: {url}")
response = self.session.get(url).json()
# logging.debug(f"Response: {response}")
response = self.session.get(url).json()
# logging.debug(f"Response: {response}")
image = response['image']['image']['source']
logging.debug(f"Image: {image}")
image_url = image
image = response['image']['image']['source']
logging.debug(f"Image: {image}")
image_url = image
title = response['image']['description']['text']
return image_url, title
title = response['image']['description']['text']
return image_url, title
except KeyError:
logging.error("KeyError, retrying...")
time.sleep(10) # Wait 10 seconds
return self.get_image_info()