Python Script to update the XML for Linux Mint Background Slideshow

mint_update_background.py

#!/usr/bin/env python
import os,sys,re
from lxml import etree as ET

folders = sys.argv[1:]

if len(folders) == 0:
    sys.stderr.write("No folders given\n")
    sys.exit()

files = []

for folder in folders:
    for basepath,_,newfiles in os.walk(folder):
        files.extend(os.path.abspath(os.path.join(basepath, f)) for f in newfiles)

root = ET.Element('wallpapers')
for f in files:
    e_e = ET.SubElement(root,'wallpaper',{'deleted': 'false'})
    ET.SubElement(e_e,'filename').text = f
    ET.SubElement(e_e,'title').text = re.match(".+/(.+)/.+$",f).group(1)
    ET.SubElement(e_e,'author').text = os.environ['USER']

sys.stdout.write(ET.tostring(root, encoding="UTF-8", pretty_print=False, xml_declaration=True, doctype='<!DOCTYPE wallpapers SYSTEM "cinnamon-wp-list.dtd">', with_tail=False))

update_file.sh

python mint_update_background.py ~/Documents/Pictures/Screenshots/ | sudo tee /usr/share/cinnamon-background-properties/screenshots.xml | xmllint --format -