Cleanup for stow ---STOW

[dotfiles/.git] / .local / lib / python3.9 / site-packages / pywal / util.py

diff --git a/.local/lib/python3.9/site-packages/pywal/util.py b/.local/lib/python3.9/site-packages/pywal/util.py
new file mode 100644 (file)
index 0000000..de8eb09
--- /dev/null
+++ b/.local/lib/python3.9/site-packages/pywal/util.py
@@ -0,0 +1,186 @@
+"""
+Misc helper functions.
+"""
+import colorsys
+import json
+import logging
+import os
+import subprocess
+import sys
+
+
+class Color:
+    """Color formats."""
+    alpha_num = "100"
+
+    def __init__(self, hex_color):
+        self.hex_color = hex_color
+
+    def __str__(self):
+        return self.hex_color
+
+    @property
+    def rgb(self):
+        """Convert a hex color to rgb."""
+        return "%s,%s,%s" % (*hex_to_rgb(self.hex_color),)
+
+    @property
+    def xrgba(self):
+        """Convert a hex color to xrdb rgba."""
+        return hex_to_xrgba(self.hex_color)
+
+    @property
+    def rgba(self):
+        """Convert a hex color to rgba."""
+        return "rgba(%s,%s,%s,%s)" % (*hex_to_rgb(self.hex_color),
+                                      int(self.alpha_num)/100)
+
+    @property
+    def alpha(self):
+        """Add URxvt alpha value to color."""
+        return "[%s]%s" % (self.alpha_num, self.hex_color)
+
+    @property
+    def octal(self):
+        """Export color in octal"""
+        return "%s%s" % ("#", oct(int(self.hex_color[1:], 16))[2:])
+
+    @property
+    def octal_strip(self):
+        """Strip '#' from octal color."""
+        return oct(int(self.hex_color[1:], 16))[2:]
+
+    @property
+    def strip(self):
+        """Strip '#' from color."""
+        return self.hex_color[1:]
+
+
+def read_file(input_file):
+    """Read data from a file and trim newlines."""
+    with open(input_file, "r") as file:
+        return file.read().splitlines()
+
+
+def read_file_json(input_file):
+    """Read data from a json file."""
+    with open(input_file, "r") as json_file:
+        return json.load(json_file)
+
+
+def read_file_raw(input_file):
+    """Read data from a file as is, don't strip
+       newlines or other special characters.."""
+    with open(input_file, "r") as file:
+        return file.readlines()
+
+
+def save_file(data, export_file):
+    """Write data to a file."""
+    create_dir(os.path.dirname(export_file))
+
+    try:
+        with open(export_file, "w") as file:
+            file.write(data)
+    except PermissionError:
+        logging.warning("Couldn't write to %s.", export_file)
+
+
+def save_file_json(data, export_file):
+    """Write data to a json file."""
+    create_dir(os.path.dirname(export_file))
+
+    with open(export_file, "w") as file:
+        json.dump(data, file, indent=4)
+
+
+def create_dir(directory):
+    """Alias to create the cache dir."""
+    os.makedirs(directory, exist_ok=True)
+
+
+def setup_logging():
+    """Logging config."""
+    logging.basicConfig(format=("[%(levelname)s\033[0m] "
+                                "\033[1;31m%(module)s\033[0m: "
+                                "%(message)s"),
+                        level=logging.INFO,
+                        stream=sys.stdout)
+    logging.addLevelName(logging.ERROR, '\033[1;31mE')
+    logging.addLevelName(logging.INFO, '\033[1;32mI')
+    logging.addLevelName(logging.WARNING, '\033[1;33mW')
+
+
+def hex_to_rgb(color):
+    """Convert a hex color to rgb."""
+    return tuple(bytes.fromhex(color.strip("#")))
+
+
+def hex_to_xrgba(color):
+    """Convert a hex color to xrdb rgba."""
+    col = color.lower().strip("#")
+    return "%s%s/%s%s/%s%s/ff" % (*col,)
+
+
+def rgb_to_hex(color):
+    """Convert an rgb color to hex."""
+    return "#%02x%02x%02x" % (*color,)
+
+
+def darken_color(color, amount):
+    """Darken a hex color."""
+    color = [int(col * (1 - amount)) for col in hex_to_rgb(color)]
+    return rgb_to_hex(color)
+
+
+def lighten_color(color, amount):
+    """Lighten a hex color."""
+    color = [int(col + (255 - col) * amount) for col in hex_to_rgb(color)]
+    return rgb_to_hex(color)
+
+
+def blend_color(color, color2):
+    """Blend two colors together."""
+    r1, g1, b1 = hex_to_rgb(color)
+    r2, g2, b2 = hex_to_rgb(color2)
+
+    r3 = int(0.5 * r1 + 0.5 * r2)
+    g3 = int(0.5 * g1 + 0.5 * g2)
+    b3 = int(0.5 * b1 + 0.5 * b2)
+
+    return rgb_to_hex((r3, g3, b3))
+
+
+def saturate_color(color, amount):
+    """Saturate a hex color."""
+    r, g, b = hex_to_rgb(color)
+    r, g, b = [x/255.0 for x in (r, g, b)]
+    h, l, s = colorsys.rgb_to_hls(r, g, b)
+    s = amount
+    r, g, b = colorsys.hls_to_rgb(h, l, s)
+    r, g, b = [x*255.0 for x in (r, g, b)]
+
+    return rgb_to_hex((int(r), int(g), int(b)))
+
+
+def rgb_to_yiq(color):
+    """Sort a list of colors."""
+    return colorsys.rgb_to_yiq(*hex_to_rgb(color))
+
+
+def disown(cmd):
+    """Call a system command in the background,
+       disown it and hide it's output."""
+    subprocess.Popen(cmd,
+                     stdout=subprocess.DEVNULL,
+                     stderr=subprocess.DEVNULL)
+
+
+def get_pid(name):
+    """Check if process is running by name."""
+    try:
+        subprocess.check_output(["pidof", "-s", name])
+    except subprocess.CalledProcessError:
+        return False
+
+    return True