diff --git a/twoot.py b/twoot.py index c2157f4..40b5420 100755 --- a/twoot.py +++ b/twoot.py @@ -237,20 +237,22 @@ def update_profile(nitter_url, soup, sql, mast_password): # Download images new_avatar = requests.get(nitter_url + new_avatar_url, timeout=HTTPS_REQ_TIMEOUT) if new_avatar_url is not None else None - new_avatar_img = new_avatar.content if new_avatar.status_code == 200 else None - new_avatar_mime = new_avatar.headers['content-type'] if new_avatar.status_code == 200 else None - if new_avatar.status_code !=200: - logging.error("Could not download avatar image from " + nitter_url + new_avatar_url) - else: - logging.debug("Avatar image downloaded") + if new_avatar is not None: + new_avatar_img = new_avatar.content if new_avatar.status_code == 200 else None + new_avatar_mime = new_avatar.headers['content-type'] if new_avatar.status_code == 200 else None + if new_avatar.status_code !=200: + logging.error("Could not download avatar image from " + nitter_url + new_avatar_url) + else: + logging.debug("Avatar image downloaded") new_banner = requests.get(nitter_url + new_banner_url, timeout=HTTPS_REQ_TIMEOUT) if new_banner_url is not None else None - new_banner_img = new_banner.content if new_banner.status_code == 200 else None - new_banner_mime = new_banner.headers['content-type'] if new_banner.status_code == 200 else None - if new_banner.status_code !=200: - logging.error("Could not download banner image from " + nitter_url + new_banner_url) - else: - logging.debug("Banner image downloaded") + if new_banner is not None: + new_banner_img = new_banner.content if new_banner.status_code == 200 else None + new_banner_mime = new_banner.headers['content-type'] if new_banner.status_code == 200 else None + if new_banner.status_code !=200: + logging.error("Could not download banner image from " + nitter_url + new_banner_url) + else: + logging.debug("Banner image downloaded") mastodon = login(mast_password) # Update profile on Mastodon