certanet / firepyer Goto Github PK

def create_network_object(fdm_object: object, name: str, value: str, type: str) -> dict:
    """
    Create a network object in FDM, overwrite if already exists.
    """
    try:
        network_object = fdm_object.create_network(type=type, name=name, value=value)
        return network_object
    except FirepyerError as e:
        if "duplicate name" in str(e):
            existing_network_object = fdm_object.get_net_objects(name)
            pprint(existing_network_object)
            print(f"Network object {name} already exists")
            print("Updating network object")
            existing_network_object["value"] = value
            existing_network_object[
                "subType"
            ] = (
                type.upper()
            )  # Capitalization needed since we're sending directly to api
            pprint(existing_network_object)
            put_it = fdm_object.put_api(
                f"/object/networks/{existing_network_object['id']}",
                json.dumps(existing_network_object),
            )
            if put_it.status_code == 200 or put_it.status_code == 204:
                print("Network object updated")
                return existing_network_object
            else:
                raise FirepyerError(
                    f"Failed to update network object {name} with status code {put_it.status_code}\n{put_it.text}"
                )
        else:
            raise e