RiskIQPassiveTotalHostPairsChildrenWidgetScript

Source

import traceback
from typing import Any

from CommonServerPython import *


def set_arguments_for_widget_view(indicator_data: dict[str, Any]) -> dict[str, str] | str:
    """
    Prepare argument for commands or message to set custom layout of indicator
    """
    indicator_type = indicator_data.get("indicator_type", "").lower()
    if indicator_type == "riskiqasset":
        riskiq_asset_type = indicator_data.get("CustomFields", {}).get("riskiqassettype", "")
        if riskiq_asset_type == "":
            return 'Please provide value in the "RiskIQAsset Type" field to fetch detailed information of the asset.'
        if riskiq_asset_type == "Domain" or riskiq_asset_type == "IP Address":
            return {"direction": "children", "query": indicator_data.get("value", "")}
        else:
            return "No host pair(s) were found for the given argument(s)."
    else:
        return {"direction": "children", "query": indicator_data.get("value", "")}


def main() -> None:
    try:
        arguments = set_arguments_for_widget_view(demisto.args().get("indicator"))
        if isinstance(arguments, str):
            demisto.results(arguments)
        else:
            demisto.results(demisto.executeCommand("pt-get-host-pairs", arguments))
    except Exception as e:
        demisto.error(traceback.format_exc())  # print the traceback
        return_error(f"Could not load widget:\n{e}")


# python2 uses __builtin__ python3 uses builtins
if __name__ == "__builtin__" or __name__ == "builtins":
    main()

README