[IDEV-282] bugbash: add short-hand for source file param; add max limit of 100 to list of domains in a source file

Author: briluza

domaintools/cli/api.py

@@ -6,14 +6,11 @@
 
 from typing import Optional, Dict, Tuple
 from rich.progress import Progress, SpinnerColumn, TextColumn
-from rich import console
 
 from domaintools.api import API
 from domaintools.exceptions import ServiceException
 from domaintools.cli.utils import get_file_extension
 
-console = console.Console()
-
 
 class DTCLICommand:
     API_SUCCESS_STATUS = 200
@@ -129,6 +126,13 @@ def _get_domains_from_source(cls, source: str) -> Dict[str, str]:
                     domains.extend([row.get("domain") or "" for row in reader])
                 else:
                     domains.extend([domain.strip() for domain in src.readlines()])
+
+                total_domains_found = len(domains)
+                if total_domains_found > 100:
+                    raise typer.BadParameter(
+                        f"Domains in source file exceeds the maximum count of 100. Current source file domain count: {total_domains_found}"
+                    )
+
         except FileNotFoundError:
             raise typer.BadParameter(f"File '{source}' not found.")
 
@@ -198,13 +202,11 @@ def run(cls, name: str, params: Optional[Dict] = {}, **kwargs):
 
                 if isinstance(out_file, _io.TextIOWrapper):
                     # use rich `print` command to prettify the ouput in sys.stdout
-                    print(output)
+                    print(response)
                 else:
                     # if it's a file then write
                     out_file.write(output if output.endswith("\n") else output + "\n")
-                time.sleep(0.5)
-
-            name = typer.style(name, fg=typer.colors.CYAN, bold=True)
+                time.sleep(0.25)
         except Exception as e:
             if isinstance(e, ServiceException):
                 code = typer.style(getattr(e, "code", 400), fg=typer.colors.BRIGHT_RED)
@@ -218,7 +220,7 @@ def run(cls, name: str, params: Optional[Dict] = {}, **kwargs):
 
                 reason = typer.style(_reason, bg=typer.colors.RED)
 
-                err_msg_format = f"Error occured while fetching data from API: [{code}] Reason: {reason}"
+                err_msg_format = f"Error occured while fetching data from the API: [{code}] Reason: {reason}"
                 typer.echo(message=err_msg_format)
             else:
                 typer.echo(message=e)

domaintools/cli/commands/iris.py

@@ -30,8 +30,9 @@ def iris_investigate(
     ),
     src_file: str = typer.Option(
         None,
+        "-s",
         "--source-file",
-        help="Comma-separated list of domains. Supports only {.csv, .txt} format",
+        help="Comma-separated list of maximum 100 domains. Supports only {.csv, .txt} format",
         callback=DTCLICommand.validate_source_file_extension,
     ),
     user: str = typer.Option(None, "-u", "--user", help="Domaintools API Username."),
@@ -81,8 +82,9 @@ def iris_enrich(
     domains: str = typer.Option(None, "-d", "--domains", help="Domains to use."),
     src_file: str = typer.Option(
         None,
+        "-s",
         "--source-file",
-        help="Comma-separated list of domains. Supports only {.csv, .txt} format",
+        help="Comma-separated list of maximum 100 domains. Supports only {.csv, .txt} format",
         callback=DTCLICommand.validate_source_file_extension,
     ),
     user: str = typer.Option(None, "-u", "--user", help="Domaintools API Username."),