wallpaper_change.py

#!/usr/bin/env python

import os
import struct
import ctypes
import random
import logging
import platform
import blackram_utils
import sqlite3

# TODO: check previously set wallpaper is not selected again to set

logging.basicConfig()
logger = logging.getLogger()
logger.setLevel(logging.INFO)

# read all wallpapers stored location
wallpapers_location = blackram_utils.get_param_value("blackscripts.settings.wallpaper")
logger.info("wallpapers location is "+wallpapers_location)
allowed_files = ['jpg', 'jpeg', 'png', 'bmp']
wallpapers_list = []

# getting list of all files in the folder
list = os.listdir(wallpapers_location)
logger.info(list)

#Checking the OS
ops = blackram_utils.checkOS()

# iterate through the list of wallpapers list
for file in list:
    # for checking whether it is a file
    # os.path.isfile needs complete file path instead of file name alone
    if os.path.isfile(wallpapers_location+"/"+file) and file.split('.')[1] in allowed_files:
        logger.info(wallpapers_location+"/"+file)
        wallpapers_list.append(file)

# find total wallpaper files present in the directory
max_len = len(wallpapers_list)
logger.info("max length is "+str(max_len))

# generate a random number to get the wallpaper number to be set
wallpaper_num = random.randint(0, (max_len-1))
logger.info("Random number is "+str(wallpaper_num))
logger.info("wallpaper selected is "+wallpapers_list[wallpaper_num])

logger.info("Performing Wallpaper change for "+ ops)

# for running Ubuntu 18.04 gsettings has to be changed to set wallpaper
if ops == 'LIN':
    os.system("/usr/bin/gsettings set org.gnome.desktop.background picture-uri "+wallpapers_location+"/"+wallpapers_list[wallpaper_num])
    logger.info("Command Executed is : /usr/bin/gsettings set org.gnome.desktop.background picture-uri "+wallpapers_location+"/"+wallpapers_list[wallpaper_num])

# for windows to set wallpaper
elif ops == 'WIN':
    TOTAL=wallpapers_location+"/"+wallpapers_list[wallpaper_num]
    PATH =bytes(TOTAL,'utf-8')
    SPI_SETDESKWALLPAPER = 20
    ctypes.windll.user32.SystemParametersInfoA(SPI_SETDESKWALLPAPER, 0, PATH, 3)

elif ops == 'MAC':
    logger.info("Work in Progress for MAC")