import os
import re
import sys
import click
import json
import logging
from random import choice
from datetime import datetime
from subprocess import check_call, CalledProcessError
from PIL import Image
try:
# for python3
from urllib.request import urlopen, urlretrieve, HTTPError, URLError
except ImportError:
# for python2
from urllib import urlretrieve
from urllib2 import urlopen, HTTPError, URLError
if sys.platform.startswith("win32"):
import win32api
import win32con
import win32gui
logging.basicConfig(level=logging.INFO, format="%(message)s")
log = logging.getLogger(__name__)
def set_wallpaper(image_path):
if image_path is None:
return
""" Given a path to an image, set it as the wallpaper """
if not os.path.exists(image_path):
log.error("Image does not exist.")
sys.exit(1)
log.info("Updating wallpaper..")
log.info(
"Name of wallpaper: {}".format(
re.sub("([a-z])([A-Z])", r"\1 \2", image_path.split("/")[-1].split("_")[0])
)
)
if sys.platform.startswith("win32"):
bmp_image = Image.open(image_path)
bmp_img_path = os.path.splitext(image_path)[0] + ".bmp"
bmp_image.save(bmp_img_path, "BMP")
key = win32api.RegOpenKeyEx(
win32con.HKEY_CURRENT_USER,
"Control Panel\\Desktop",
0,
win32con.KEY_SET_VALUE,
)
win32api.RegSetValueEx(key, "WallpaperStyle", 0, win32con.REG_SZ, "0")
win32api.RegSetValueEx(key, "TileWallpaper", 0, win32con.REG_SZ, "0")
win32gui.SystemParametersInfo(
win32con.SPI_SETDESKWALLPAPER, bmp_img_path, 1 + 2
)
os.remove(bmp_img_path)
elif sys.platform.startswith("darwin"):
try:
command = """
osascript -e 'tell application "System Events"
set desktopCount to count of desktops
repeat with desktopNumber from 1 to desktopCount
tell desktop desktopNumber
set picture to "{image_path}"
end tell
end repeat
end tell'
""".format(
image_path=image_path
)
check_call([command], shell=True)
except (CalledProcessError, FileNotFoundError):
log.error("Setting wallpaper failed.")
sys.exit(1)
elif sys.platform.startswith("linux"):
check_call(
[
"gsettings",
"set",
"org.gnome.desktop.background",
"picture-uri",
"file://{}".format(image_path),
]
)
log.info("Wallpaper successfully updated. :)")
def get_saved_wallpaper(wall_dir):
""" returns random saved wallpaper's path """
files = os.listdir(wall_dir)
if files:
log.info("\nRandomly picking from saved wallpapers.")
image_name = choice(files)
image_path = os.path.join(os.sep, wall_dir, image_name)
return image_path
else:
log.error("There is no wallpaper available offline.")
sys.exit(1)
def get_wallpaper_directory():
""" check if `default` wallpaper download directory exists or not, create if doesn't exist """
pictures_dir = ""
wall_dir_name = "freshpaper"
os.path.join(os.sep, os.path.expanduser("~"), "a", "freshpaper")
if sys.platform.startswith("win32"):
pictures_dir = "My Pictures"
elif sys.platform.startswith("darwin"):
pictures_dir = "Pictures"
elif sys.platform.startswith("linux"):
pictures_dir = "Pictures"
wall_dir = os.path.join(
os.sep, os.path.expanduser("~"), pictures_dir, wall_dir_name
)
if not os.path.isdir(wall_dir):
log.error("wallpaper directory does not exist.")
os.makedirs(wall_dir)
log.info("created wallpaper directory at: {}".format(wall_dir))
return wall_dir
def download_image_bing(download_dir, image_extension="jpg"):
"""
Download & save the image
:param download_dir: directory where to download the image
:param image_extension: directory where to download the image
:return: downloaded image path
"""
# mkt(s) HIN, EN-IN
url = "http://www.bing.com/HPImageArchive.aspx?format=js&idx=0&n=1&mkt=EN-IN"
try:
image_data = json.loads(urlopen(url).read().decode("utf-8"))
image_url = "http://www.bing.com" + image_data["images"][0]["url"]
image_name = re.search(r"OHR\.(.*?)_", image_url).group(1)
image_url_hd = "http://www.bing.com/hpwp/" + image_data["images"][0]["hsh"]
date_time = datetime.now().strftime("%d_%m_%Y")
image_file_name = "{image_name}_{date_stamp}.{extention}".format(
image_name=image_name, date_stamp=date_time, extention=image_extension
)
image_path = os.path.join(os.sep, download_dir, image_file_name)
log.debug("download_dir: {}".format(download_dir))
log.debug("image_file_name: {}".format(image_file_name))
log.debug("image_path: {}".format(image_path))
if os.path.isfile(image_path):
log.info("No new wallpaper yet..updating to latest one.\n")
return image_path
try:
log.info("Downloading..")
urlretrieve(image_url_hd, filename=image_path)
except HTTPError:
log.info("Downloading...")
urlretrieve(image_url, filename=image_path)
return image_path
except URLError:
log.error("Something went wrong..\nMaybe Internet is not working...")
raise ConnectionError
def download_image_nasa(download_dir, image_extension="jpg"):
"""
Download & save the image
:param download_dir: directory where to download the image
:param image_extension: directory where to download the image
:return: downloaded image path
"""
url = "https://api.nasa.gov/planetary/apod?api_key=DEMO_KEY"
try:
image_data = json.loads(urlopen(url).read().decode("utf-8"))
if image_data.get("media_type") != "image":
log.info("No NASA image of the day available. It can be a video.\n")
return None
image_url = image_data.get("url")
image_name = image_data.get("title").split(" ")[0]
image_url_hd = image_data.get("hdurl")
date_time = datetime.now().strftime("%d_%m_%Y")
image_file_name = "{image_name}_{date_stamp}.{extention}".format(
image_name=image_name, date_stamp=date_time, extention=image_extension
)
image_path = os.path.join(os.sep, download_dir, image_file_name)
log.debug("download_dir: {}".format(download_dir))
log.debug("image_file_name: {}".format(image_file_name))
log.debug("image_path: {}".format(image_path))
if os.path.isfile(image_path):
log.info("No new wallpaper yet..updating to latest one.\n")
return image_path
try:
log.info("Downloading..")
urlretrieve(image_url_hd, filename=image_path)
except HTTPError:
log.info("Downloading...")
urlretrieve(image_url, filename=image_path)
return image_path
except URLError:
log.error("Something went wrong..\nMaybe Internet is not working...")
raise ConnectionError
def download_image_unsplash_random(download_dir, image_extension="jpg"):
"""
Download & save the image
:param download_dir: directory where to download the image
:param image_extension: directory where to download the image
:return: downloaded image path
"""
url = "https://source.unsplash.com/random/1920x1080"
try:
image_url = url
image_name = "unsplash_random"
date_time = datetime.now().strftime("%d_%m_%Y_%H_%M_%S")
image_file_name = "{image_name}_{date_stamp}.{extention}".format(
image_name=image_name, date_stamp=date_time, extention=image_extension
)
image_path = os.path.join(os.sep, download_dir, image_file_name)
log.debug("download_dir: {}".format(download_dir))
log.debug("image_file_name: {}".format(image_file_name))
log.debug("image_path: {}".format(image_path))
if os.path.isfile(image_path):
log.info("No new wallpaper yet..updating to latest one.\n")
return image_path
log.info("Downloading...")
urlretrieve(image_url, filename=image_path)
return image_path
except URLError:
log.error("Something went wrong..\nMaybe Internet is not working...")
raise ConnectionError
def download_image_nat_geo(download_dir, image_extension="jpg"):
"""
Download & save the image
:param download_dir: directory where to download the image
:param image_extension: directory where to download the image
:return: downloaded image path
"""
url = "https://www.nationalgeographic.com/photography/photo-of-the-day/"
try:
request = urlopen(url)
except URLError:
log.error("Something went wrong..\nMaybe Internet is not working...")
raise ConnectionError
html = request.read().decode("utf-8")
url_regex = r"twitter:image:src\" content=\"(.*)\""
image_url = re.findall(url_regex, html)[0]
if not image_url:
log.info("No National Geographic image of the day available.\n")
return None
image_name_regex = r"json\":{\"title\":\"(.*)\""
image_name = re.findall(image_name_regex, html)[0]
try:
if not image_name:
image_name = "nat_geo"
date_time = datetime.now().strftime("%d_%m_%Y")
image_file_name = "{image_name}_{date_stamp}.{extention}".format(
image_name=image_name, date_stamp=date_time, extention=image_extension
)
image_path = os.path.join(os.sep, download_dir, image_file_name)
log.debug("download_dir: {}".format(download_dir))
log.debug("image_file_name: {}".format(image_file_name))
log.debug("image_path: {}".format(image_path))
if os.path.isfile(image_path):
log.info("No new wallpaper yet..updating to latest one.\n")
return image_path
log.info("Downloading...")
urlretrieve(image_url, filename=image_path)
return image_path
except URLError:
log.error("Something went wrong..\nMaybe Internet is not working...")
raise ConnectionError
freshpaper_sources = {
"bing": {"download": download_image_bing, "description": "Bing photo of the day"},
"nasa": {"download": download_image_nasa, "description": "NASA photo of the day"},
"unsplash_random": {
"download": download_image_unsplash_random,
"description": "Unsplash random photo",
},
"nat_geo": {
"download": download_image_nat_geo,
"description": "National Geographic photo of the day",
},
}
@click.group(invoke_without_command=True)
@click.pass_context
@click.option(
"--source",
default="bing",
type=click.Choice(freshpaper_sources.keys()),
help="Source for setting the wallpaper.",
)
def main(ctx, source):
if ctx.invoked_subcommand is None:
dir_name = get_wallpaper_directory() # Wallpaper directory name
try:
download_image = freshpaper_sources.get(source)["download"]
image_path = download_image(dir_name)
set_wallpaper(image_path)
except ConnectionError:
image_path = get_saved_wallpaper(dir_name)
set_wallpaper(image_path)
except Exception as e:
log.error(e)
if __name__ == "__main__":
main()
