CullaX.py

#!/usr/bin/python3
"""Culla generates a desktop theme using colours
from the current wallpaper"""

import sys
import os
import subprocess
import time
import colorsys
from PIL import Image
import colorgram



#Template for our Plasma theme
plasma_colors = """[Colors:Window]
ForegroundNormal=bbb
BackgroundNormal=aaa
ForegroundNegative=212,84,0

[Colors:Selection]
BackgroundNormal=ddd

[Colors:Complimentary]
BackgroundNormal=255,0,0
ForegroundNormal=0,255,0

[Colors:Button]
ForegroundNormal=eee
BackgroundNormal=fff
DecorationFocus=ddd
DecorationHover=0,255,0

[Colors:View]
ForegroundNormal=bbb
BackgroundNormal=ccc
DecorationHover=ddd"""


def notify_user(message):
    """ Simple notification to show something's happening """
    icon_path = os.path.expanduser('~/.local/share/pixmaps/cullax.png')
    icon = '--icon={}'.format(icon_path)

    try:
        subprocess.run(['notify-send',
                        icon,
                        '--expire-time=3000',
                        message])
    except:
        pass

def color_triplet(h, l, s):
    r, g, b = colorsys.hls_to_rgb(h, l, s)

    if r > 1.0:
        r = 1.0
    r = int(r * 255)

    if g > 1.0:
        g = 1.0
    g = int(g * 255)

    if b > 1.0:
        r = 1.0
    b = int(b * 255)

    return ','.join([str(r), str(g), str(b)])




# ----  CullaX  ----


# Try sending a notification to show we're working
notify_user("CullaX - Beautifying Desktop ;) ")

# Raise flag when finding correct session in plasmarc
flag = False
# Holder for current activity ID
activity = ""

try:
    with open(os.path.expanduser(
            '~/.config/plasma-org.kde.plasma.desktop-appletsrc')) as f:
        plasmaconfig = f.readlines()
except:
    sys.exit('Unable to find plasma config.')


try:
    with open(os.path.expanduser('~/.config/kactivitymanagerdrc')) as f:
        activityrc = f.readlines()
except:
    print('Unable to find kactivity manager rc. Presuming only default activity.')
    activityrc = None
    flag = True   # There is only default activity

# Retrieve current activity
if activityrc is not None:
    a = [a for a in activityrc if 'current' in a]
    a = a[0].split('=')
    activity = a[1].rstrip()

# Flag if wallpaper is found
found = False

# Find current activity then grab next Image= key
for line in plasmaconfig:
    if activity in line:
        flag = True
    if 'Image=' in line and flag:
        found = True
        break

if not found:
    notify_user(
"""You need to set your own wallpaper\n\
before I can work."""
                )
    sys.exit("I didn't find your wallpaper. Have you set one yet?")

#tmp, wallpaper = line.split('//')
line = line[6:]
if line[0] == 'f':
    line = line[7:]

wallpaper = line.strip()

if not os.path.exists(wallpaper):
    notify_user("I can't your wallpaper."
             .format(wallpaper))
    sys.exit("I think the wallpaper is {0} but I can't find it."
             .format(wallpaper))

# Resize and write to temp file - massive speedup
tmp_img = Image.open(wallpaper.rstrip())
tmp_img = tmp_img.resize((512, 512))
tmp_img_path = os.path.expanduser('~/.cullax.png')
tmp_img.save(tmp_img_path)
colors = colorgram.extract(tmp_img_path, 3)

#background_hue = colors[0].hsl[0]
#higlight_hsl = colors[0].hsl
#diff = 0


#for i in colors:
    #print(i.hsl, i)
    ##tmp_hue = i.hsl[0]
    
    #if abs(background_hue - i.hsl[0]) > diff:
        #contrast_color = i.hsl
        #diff = abs(background_hue - i.hsl[0])

#print(contrast_color)
#sys.exit(0)



h,s,l = [ c/255.0 for c in colors[1].hsl ]
hb,sb,lb = [ c/255.0 for c in colors[0].hsl ]

# Cleanup temp file
os.remove(tmp_img_path)


#print(h,s,l)
#sys.exit(0)

# Remove hue for b/w images
if s < 0.011:
    s = 0.0
    h = 0.0

#l_midlight = (1.0 - l) / 4.0 + l
panel_background = color_triplet(hb, 0.15, 0.05)
panel_header = color_triplet(hb, 0.25, 0.4)
button_foreground = color_triplet(h, 0.85, s/3.0)
foreground = color_triplet(h, 0.99, 0.9)
highlight_color = color_triplet(h, 0.88, 1)
midlight = color_triplet(h, 0.5, 0.6)

print(panel_background)

# Write the colorscheme
plasma_colors = plasma_colors.replace('aaa', panel_background)
plasma_colors = plasma_colors.replace('bbb', foreground)
plasma_colors = plasma_colors.replace('ddd', highlight_color)
plasma_colors = plasma_colors.replace('eee', button_foreground)
plasma_colors = plasma_colors.replace('fff', midlight)
plasma_colors = plasma_colors.replace('ccc', panel_header)

try:
    with open(os.path.expanduser(
            '~/.local/share/plasma/desktoptheme/CullaX/colors'), 'w') as f:
        f.write(plasma_colors)
except:
    sys.exit("Unable to open Culla Plasma colors. Is it installed?")

# Call kwriteconfig5 to change Plasma theme
try:
    subprocess.run(['kwriteconfig5', '--file=plasmarc',
                    '--group=Theme', '--key=name', 'Default'])

    # Do this too quickly and Plasma won't reload
    time.sleep(0.6)

    subprocess.run(['kwriteconfig5', '--file=plasmarc',
                    '--group=Theme', '--key=name', 'CullaX'])
except IOError as e:
    sys.exit(e)