Update cli.py: try to make work replace_color

Schroedinger-Hat · Apr 7, 2021 · 5b297e2 · 5b297e2
1 parent c733af1
commit 5b297e2
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 20 deletions.
diff --git a/src/cli.py b/src/cli.py
@@ -47,10 +47,8 @@
 import re
 from os import path
 
-from PIL import Image
-
 import configs.arguments as confarg
-from utility.quantize import quantize_to_palette
+from utility.quantize import quantize_to_palette, replace_color
 import utility.palette_loader as pl
 
 VERSION = open(path.dirname(path.realpath(__file__)) +
@@ -60,8 +58,9 @@
 QUIET_MODE = False
 IS_NO_AVERAGE = False
 PIXELS_AREA = 10
-OUTPUT_IMAGE_NAME = ""
+OUTPUT_IMAGE_NAME = "nord" + DEAFAULT_EXTENSION
 BLUR = False
+PALETTE_DATA = []
 
 
 def is_colors_selected(selection, color_name):
@@ -130,6 +129,17 @@ def get_version():
     return file_version.readline()
 
 
+def load_all_palette(palette_path):
+    palette_set = pl.load_palette_set(palette_path)
+    palette_data = []
+    for colors_name in palette_set:
+        colors_palette = pl.import_palette_from_file(
+            palette_path + colors_name + ".txt")
+        colors_set = pl.create_data_colors(colors_palette)
+        palette_data.extend(colors_set)
+    return palette_data
+
+
 if __name__ == '__main__':
     args = sys.argv[1:]
 
@@ -237,22 +247,19 @@ def get_version():
             continue
         del condition_argument
 
-        palettedata = []
-
         for palette in palettes:
             palette_path = src_path + "/palettes/" + palette.capitalize() + "/"
 
             if "--{}".format(palette) in key:
 
                 if len(key_value) == 1:
+                    palette_set = pl.load_palette_set(palette_path)
                     to_console(confarg.logs["pals"][0].format(
                         palette.capitalize()))
-                    palette_set = pl.load_palette_set(palette_path)
-                    for colors_name in palette_set:
-                        colors_palette = pl.import_palette_from_file(
-                            palette_path + colors_name + ".txt")
-                        colors_set = pl.create_data_colors(colors_palette)
-                        palettedata.extend(colors_set)
+                    PALETTE_DATA = load_all_palette(palette_path)
+                    for selected_color in palette_set:
+                        to_console(confarg.logs["pals"]
+                                   [2].format(selected_color))
                 else:
                     to_console(confarg.logs["pals"]
                                [1].format(palette.capitalize()))
@@ -266,25 +273,37 @@ def get_version():
                                     confarg.logs["pals"][2].format(colors_name))
                                 colors_palette = pl.import_palette_from_file(
                                     palette_path + colors_name + ".txt")
-                                colors_set = pl.create_data_colors(colors_palette)
-                                palettedata.extend(colors_set)
+                                colors_set = pl.create_data_colors(
+                                    colors_palette)
+                                PALETTE_DATA.extend(colors_set)
                                 palette_set.remove(colors_name)
                                 FOUND = True
                         if not FOUND:
-                            to_console(confarg.logs["pals"][4].format(
+                            to_console(confarg.logs["pals"][-3].format(
                                 selected_color_set))
                         del FOUND
                     for unselected_color_set in palette_set:
                         to_console(confarg.logs["pals"][3].format(
                             unselected_color_set))
-                    if len(palettedata) == 0:
+                    if len(PALETTE_DATA) == 0:
                         to_console(confarg.logs["pals"][-2].format(arg))
                         to_console(confarg.logs["pals"][-1])
                         sys.exit(1)
 
+    if len(PALETTE_DATA) == 0:
+        palette_path = src_path + "/palettes/Nord/"
+        palette_set = pl.load_palette_set(palette_path)
+        PALETTE_DATA = load_all_palette(palette_path)
+        to_console(confarg.logs["pals"][4])
+        for selected_color in palette_set:
+            to_console(confarg.logs["pals"][2].format(selected_color))
+
     # padding with black color | nordtheme palette is only 48
-    while len(palettedata) < 768:
-        palettedata.extend(pl.export_tripletes_from_color(BLACK_REPLACE))
+    while len(PALETTE_DATA) < 768:
+        PALETTE_DATA.extend(pl.export_tripletes_from_color(BLACK_REPLACE))
+
+    quantize_to_palette(
+        INPUT_IMAGE_NAME, OUTPUT_IMAGE_NAME, PALETTE_DATA, BLUR)
 
-    quantize_to_palette(Image.open(INPUT_IMAGE_NAME),
-                        Image.new('P', (1, 1)).putpalette(palettedata))
+    # NON FUNZIONA
+    # replace_color(INPUT_IMAGE_NAME, OUTPUT_IMAGE_NAME, PALETTE_DATA, BLUR)
diff --git a/src/configs/arguments.py b/src/configs/arguments.py
@@ -36,6 +36,7 @@
         "[INFO] Use palette set: {}",
         "\t {} \u2713",
         "\t {} \u2718",
+        "[WARNING] No theme specified, Nord theme selected with color set:",
         "[WARNING] No set found for: {}",
         "[ERROR] On '{}': no colors correctly given!",
         "\te.g. --nord=aurora,polarnight"