minor bug fixes with cli

Author: LVivona

build/lib/identicon/__init__.py

@@ -75,12 +75,12 @@ def validate_and_convert_color(color):
 
 
 
-def save(icon: np.ndarray, file: str, dir: str, height: int, width: int):
+def save(icon: np.ndarray, filepath: str, height: int, width: int):
     try:
-        assert os.path.isdir(dir)
-        assert file.lower().endswith(('.png', '.jpg', '.jpeg'))
+        assert 100 <= height, f"{height} is far two small, the minimum height is 100 pixels"
+        assert 100 <= width, f"{width} is far two small, the minimum width is 100 pixels"
+        assert filepath.lower().endswith(('.png', '.jpg', '.jpeg'))
 
-        filepath = os.path.join(dir, file)
         i, j = icon.shape[:2]
         h, w = height // i, width // j
         large_identicon = np.repeat(icon, h, axis=0)
@@ -91,8 +91,9 @@ def save(icon: np.ndarray, file: str, dir: str, height: int, width: int):
         return temp
     except AssertionError as e:
         console.print("[bold red]Assertion Error:[/bold red]", e)
+        raise e
     except Exception as e:
         console.print("[bold red]Error:[/bold red]", e)
+        raise e
 
-# Rest of your functions...
 

build/lib/identicon/_generate/__init__.py

@@ -1,28 +1,63 @@
-__all__ = []
-
 import argparse
+
 from identicons._generate import generate, save
 
 def main():
 
     parser = argparse.ArgumentParser(description=f'CLI tool.')
-    parser.add_argument('-t', '--text', required=True, help="" )
-    parser.add_argument('-f', '--file', required=True, help="" )
-    parser.add_argument('-p', '--primary', default='0x00', type=str, required=False, help="" )
-    parser.add_argument('-s', '--secondary', default='0xffffff', type=str, required=False, help="" )
-    parser.add_argument('-w', '--width', default=500, type=int, required=False, help="Specify the width of the output image.")
-    parser.add_argument('-hi', '--height', default=500, type=int, required=False, help="Specify the height of the output image.")
-
-    parser.add_argument('-v', '--version', action='version', version='%(prog)s 0.1.3', help='byteTensor version ')
-
-    try:
-
-        args = parser.parse_args()
-        primary, secondary = int(args.primary, 16), int(args.secondary, 16)
-        icon = generate(args.text, primary, secondary)
-        save(icon, args.file, "./", args.height, args.width)
-    except Exception as e:
-        raise e
+    parser.add_argument(
+        '-t', '--text',
+        required=True,
+        help="Input text from which to generate the identicon."
+    )
+
+    parser.add_argument(
+        '-f', '--file',
+        required=True,
+        help="Path to the output file where the identicon will be saved."
+    )
+
+    parser.add_argument(
+        '-p', '--primary',
+        default='0xb4f8c8',
+        type=str,
+        required=False,
+        help="Primary color for the identicon in hexadecimal format. Default is a mint green color."
+    )
+
+    parser.add_argument(
+        '-s', '--secondary',
+        default='0xffffff',
+        type=str,
+        required=False,
+        help="Secondary color for the identicon in hexadecimal format. Default is white."
+    )
+
+    parser.add_argument(
+        '-w', '--width',
+        default=500,
+        type=int,
+        required=False,
+        help="Specify the width of the output image in pixels. Default is 500."
+    )
+
+    parser.add_argument(
+        '-hi', '--height',
+        default=500,
+        type=int,
+        required=False,
+        help="Specify the height of the output image in pixels. Default is 500."
+    )
+
+
+    parser.add_argument('-v', '--version', action='version', version='%(prog)s 0.1.0', help='byteTensor version ')
+
+
+    args = parser.parse_args()
+    primary, secondary = int(args.primary, 16), int(args.secondary, 16)
+    icon = generate(args.text, primary, secondary)
+
+    save(icon, args.file, args.height, args.width)
 
 
 if __name__ == "__name__":