working implementationgit add .!

5 files changed, 126 insertions, 71 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..c18dd8d
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+__pycache__/
diff --git a/README.md b/README.md
index 4625f81..b21d2ef 100644
--- a/README.md
+++ b/README.md
@@ -4,9 +4,20 @@ This program will take in an image and generate a color palette based on the ima
 
 ## Usage
 
-`newTheme.py [image]`
+`python3 newTheme.py [image]`
 
 ## Requirements
 
  - Need Python3 version `> 3.6`
  - `numpy`, `cv2`, and `scipy_learn` module
+
+ ## Installation
+
+ - Clone this repo to a safe place
+ - Modify this script to adapt to your configuration
+    - With polybar, it is hard to uniquely identify the lines for the colors without knowing the line number, for this reason, the script will look for a specific line number to modify, and depending on your configuration you might have to change these line numbers. I use modified version of the stock config files for `i3` and `polybar`, so if your config is based on the stock config file, no changes *should* be necessary.
+ - Update `paths.py` with the appropiate paths to your config files
+ - In your `.bashrc`, create an alias to the python script
+    - IE, `alias newtheme='python3 /home/chandler/Documents/newColors/newTheme.py'`
+ - reload your bash config/open & close your terminal
+ - apply a new theme to your system!
diff --git a/newTheme.py b/newTheme.py
new file mode 100644
index 0000000..0511e47
--- /dev/null
+++ b/newTheme.py
@@ -0,0 +1,109 @@
+"""
+Basically what I want this program to dynamically update the colors based on the desktop background
+so it will grab colors from the desktop background by finding the most promident colors and then set
+the system colors to these colors. I believe there are two major components involved in making this work
+
+i. The part of the program that analyzes an image and determines the most prominent colors
+ii. The part of the program that goes into all the configuration files and updates their color schemes
+"""
+import sys
+import numpy as np
+import PIL
+import cv2 as cv
+from sklearn.cluster import KMeans
+from paths import Paths
+def grabColors(img_path: str, num_colors: int) -> list():
+    """
+    Takes in an image, and Number of colors, then returns a list of those colors.
+    The list of colors will contain the most prominent colors present in the image.
+    img_path - the path where your image lives (IE, /home/chandler/Pictures/moss.png)
+    num_colors - the number of colors you need back from the image
+    """
+    img = cv.imread(img_path)
+    img = cv.cvtColor(img, cv.COLOR_BGR2RGB)
+    
+    # scale image down by factor of 10 to decrease computation time
+    dim = (int(len(img[0])/10), int(len(img)/10))
+    img = cv.resize(img, dim, interpolation= cv.INTER_AREA)
+    clt = KMeans(n_clusters=num_colors, n_init='auto')
+    clt.fit(img.reshape(-1, 3))
+    return clt.cluster_centers_
+
+def rgbToHex(input_values):
+    """
+    Takes in a list of RBG color values and returns a list of those same colors as hex values
+    """
+    hex_list=[]
+    for color in input_values:
+        red = int(color[0])
+        green = int(color[1])
+        blue = int(color[2])
+        hex_list.append('#{:02x}{:02x}{:02x}'.format(red, green, blue))
+    return hex_list
+
+def compColors(color_list):
+    """
+    given a list of colors, generate complimentary colors to contrast the prominent colors.
+    return a list of these colors.
+    """
+    compliments = []
+    for color in color_list:
+        curr_hex = color[1:] # slice off the # from the hex code
+        rgb = (curr_hex[0:2], curr_hex[2:4], curr_hex[4:6])
+        comp = ['%02X' % (255 - int(a, 16)) for a in rgb]
+        compliments.append('#' + ''.join(comp))
+    return compliments
+def updatei3Theme(config_path: str, img_path: str, colors: list, compliments: list):
+    # read in the config file
+    data = ''
+    with open(config_path, 'r') as file:
+        data = file.readlines()
+    for i, line in enumerate(data):
+        # update colors
+        if "set $bgcolor" in line:
+            data[i] = 'set $bgcolor ' + colors[0] + '\n'
+        if "set $in-bgcolor" in line:
+            data[i] = 'set $in-bgcolor ' + colors[1] + '\n'
+        if "set $text" in line:
+            data[i] = 'set $text ' + compliments[0] + '\n'
+        if "set $indicator" in line:
+            data[i] = 'set $indicator ' + colors[2] + '\n'
+        if "set $in-text" in line:
+            data[i] = 'set $in-text ' + compliments[1] + '\n'
+        #update background image
+        if "exec_always feh" in line:
+            data[i] = 'exec_always feh --bg-fil ' + img_path + '\n'
+    with open(config_path, 'w') as file:
+        file.writelines(data)
+def updatePolybarTheme(config_path: str, colors: list, compliments: list):
+    data = ''
+    with open(config_path, 'r') as file:
+        data = file.readlines()
+    for i,line in enumerate(data):
+        #update colors
+        if "background =" in line and i == 19:
+            data[i] = 'background = ' + colors[0] + '\n'
+        if "background-alt =" in line and i == 20:
+            data[i] = 'background-alt = ' + colors[1] + '\n'
+        if "foreground =" in line and i == 21:
+            data[i] = 'foreground = ' + compliments[1] + '\n'
+        if "primary =" in line and i == 22:
+            data[i] = 'primary = ' + compliments[2] + '\n'
+        if "secondary =" in line and i == 23:
+            data[i] = 'secondary = ' + compliments[3] + '\n'
+        if "disabled =" in line and i == 25:
+            data[i] = 'disabled = ' + colors[2] + '\n'
+    with open(config_path, 'w') as file:
+        file.writelines(data)
+
+def main():
+    img_path = sys.argv[1]
+    popularColors = grabColors(img_path, 5)
+    hex_colors = rgbToHex(popularColors)
+    hex_compliments = compColors(hex_colors)
+    updatei3Theme(Paths['i3'], img_path, hex_colors, hex_compliments)
+    updatePolybarTheme(Paths['polybar'], hex_colors, hex_compliments)
+    print("Theme changed successfully, please reload i3")
+
+
+main()
diff --git a/paths.py b/paths.py
new file mode 100644
index 0000000..034a025
--- /dev/null
+++ b/paths.py
@@ -0,0 +1,4 @@
+Paths = {
+        'i3': '/home/chandler/.config/i3/config',
+        'polybar': '/home/chandler/.config/polybar/config.ini'
+        }
diff --git a/updateColors.py b/updateColors.py
deleted file mode 100644
index 266dc5f..0000000
--- a/updateColors.py
+++ /dev/null
@@ -1,70 +0,0 @@
-"""
-Basically what I want this program to dynamically update the colors based on the desktop background
-so it will grab colors from the desktop background by finding the most promident colors and then set
-the system colors to these colors. I believe there are two major components involved in making this work
-
-i. The part of the program that analyzes an image and determines the most prominent colors
-ii. The part of the program that goes into all the configuration files and updates their color schemes
-"""
-import sys
-import numpy as np
-import PIL
-import cv2 as cv
-from sklearn.cluster import KMeans
-def grabColors(img_path: str, num_colors: int) -> list():
-    """
-    Takes in an image, and Number of colors, then returns a list of those colors.
-    The list of colors will contain the most prominent colors present in the image.
-    img_path - the path where your image lives (IE, /home/chandler/Pictures/moss.png)
-    num_colors - the number of colors you need back from the image
-    """
-    img = cv.imread(img_path)
-    img = cv.cvtColor(img, cv.COLOR_BGR2RGB)
-    
-    # scale image down by factor of 10 to decrease computation time
-    dim = (int(len(img[0])/10), int(len(img)/10))
-    img = cv.resize(img, dim, interpolation= cv.INTER_AREA)
-    clt = KMeans(n_clusters=num_colors, n_init='auto')
-    clt.fit(img.reshape(-1, 3))
-    return clt.cluster_centers_
-
-def rgbToHex(input_values):
-    """
-    Takes in a list of RBG color values and returns a list of those same colors as hex values
-    """
-    hex_list=[]
-    for color in input_values:
-        red = int(color[0])
-        green = int(color[1])
-        blue = int(color[2])
-        hex_list.append('#{:02x}{:02x}{:02x}'.format(red, green, blue))
-    return hex_list
-
-def compColors(color_list):
-    """
-    given a list of colors, generate complimentary colors to contrast the prominent colors.
-    return a list of these colors.
-    """
-    compliments = []
-    for color in color_list:
-        curr_hex = color[1:] # slice off the # from the hex code
-        rgb = (curr_hex[0:2], curr_hex[2:4], curr_hex[4:6])
-        comp = ['%02X' % (255 - int(a, 16)) for a in rgb]
-        compliments.append('#' + ''.join(comp))
-    return compliments
-def updatei3Colors(config_path: str):
-    pass
-    """
-    Given a list of colors, update the color scheme using the most popular colors
-    
-    """
-popularColors = grabColors('/home/chandler/Pictures/papes/blueyellow.jpg', 5)
-hex_colors = rgbToHex(popularColors)
-hex_compliments = compColors(hex_colors)
-print(hex_colors)
-print(hex_compliments)
-
-def main():
-    print(sys.argv[0])
-
-main()