import pystray
from PIL import Image
import os

walls = os.listdir('/home/ritu/wallpapers')
walls.sort()
wallsMenu = map(lambda wall:pystray.MenuItem(wall, on_wall_change), walls)

image = Image.open('/home/ritu/scripts/logo.png')

def on_wall_change(tray, item):
    os.system('feh --bg-fill ~/wallpapers/'+str(item))

def on_click(tray, item):
    print(item)

def on_exit(tray, item):
    tray.stop()


tray = pystray.Icon("Ritu",
                    image,
                    title="rituparna-ui",
                    menu=pystray.Menu(
                        pystray.MenuItem('Wallpapers', pystray.Menu(
                            *wallsMenu
                        ))
                    ))


tray.run()