update

requirements.txt

@@ -0,0 +1,7 @@
+certifi==2023.5.7 ; python_version >= "3.11" and python_version < "4.0"
+charset-normalizer==3.2.0 ; python_version >= "3.11" and python_version < "4.0"
+idna==3.4 ; python_version >= "3.11" and python_version < "4.0"
+mlogconfig==0.2.5 ; python_version >= "3.11" and python_version < "4.0"
+python-dotenv==1.0.0 ; python_version >= "3.11" and python_version < "4.0"
+requests==2.31.0 ; python_version >= "3.11" and python_version < "4.0"
+urllib3==2.0.3 ; python_version >= "3.11" and python_version < "4.0"

setup.py

@@ -0,0 +1,94 @@
+certifi==2023.5.7 ; python_version >= "3.11" and python_version < "4.0" \
+    --hash=sha256:0f0d56dc5a6ad56fd4ba36484d6cc34451e1c6548c61daad8c320169f91eddc7 \
+    --hash=sha256:c6c2e98f5c7869efca1f8916fed228dd91539f9f1b444c314c06eef02980c716
+charset-normalizer==3.2.0 ; python_version >= "3.11" and python_version < "4.0" \
+    --hash=sha256:04e57ab9fbf9607b77f7d057974694b4f6b142da9ed4a199859d9d4d5c63fe96 \
+    --hash=sha256:09393e1b2a9461950b1c9a45d5fd251dc7c6f228acab64da1c9c0165d9c7765c \
+    --hash=sha256:0b87549028f680ca955556e3bd57013ab47474c3124dc069faa0b6545b6c9710 \
+    --hash=sha256:1000fba1057b92a65daec275aec30586c3de2401ccdcd41f8a5c1e2c87078706 \
+    --hash=sha256:1249cbbf3d3b04902ff081ffbb33ce3377fa6e4c7356f759f3cd076cc138d020 \
+    --hash=sha256:1920d4ff15ce893210c1f0c0e9d19bfbecb7983c76b33f046c13a8ffbd570252 \
+    --hash=sha256:193cbc708ea3aca45e7221ae58f0fd63f933753a9bfb498a3b474878f12caaad \
+    --hash=sha256:1a100c6d595a7f316f1b6f01d20815d916e75ff98c27a01ae817439ea7726329 \
+    --hash=sha256:1f30b48dd7fa1474554b0b0f3fdfdd4c13b5c737a3c6284d3cdc424ec0ffff3a \
+    --hash=sha256:203f0c8871d5a7987be20c72442488a0b8cfd0f43b7973771640fc593f56321f \
+    --hash=sha256:246de67b99b6851627d945db38147d1b209a899311b1305dd84916f2b88526c6 \
+    --hash=sha256:2dee8e57f052ef5353cf608e0b4c871aee320dd1b87d351c28764fc0ca55f9f4 \
+    --hash=sha256:2efb1bd13885392adfda4614c33d3b68dee4921fd0ac1d3988f8cbb7d589e72a \
+    --hash=sha256:2f4ac36d8e2b4cc1aa71df3dd84ff8efbe3bfb97ac41242fbcfc053c67434f46 \
+    --hash=sha256:3170c9399da12c9dc66366e9d14da8bf7147e1e9d9ea566067bbce7bb74bd9c2 \
+    --hash=sha256:3b1613dd5aee995ec6d4c69f00378bbd07614702a315a2cf6c1d21461fe17c23 \
+    --hash=sha256:3bb3d25a8e6c0aedd251753a79ae98a093c7e7b471faa3aa9a93a81431987ace \
+    --hash=sha256:3bb7fda7260735efe66d5107fb7e6af6a7c04c7fce9b2514e04b7a74b06bf5dd \
+    --hash=sha256:41b25eaa7d15909cf3ac4c96088c1f266a9a93ec44f87f1d13d4a0e86c81b982 \
+    --hash=sha256:45de3f87179c1823e6d9e32156fb14c1927fcc9aba21433f088fdfb555b77c10 \
+    --hash=sha256:46fb8c61d794b78ec7134a715a3e564aafc8f6b5e338417cb19fe9f57a5a9bf2 \
+    --hash=sha256:48021783bdf96e3d6de03a6e39a1171ed5bd7e8bb93fc84cc649d11490f87cea \
+    --hash=sha256:4957669ef390f0e6719db3613ab3a7631e68424604a7b448f079bee145da6e09 \
+    --hash=sha256:5e86d77b090dbddbe78867a0275cb4df08ea195e660f1f7f13435a4649e954e5 \
+    --hash=sha256:6339d047dab2780cc6220f46306628e04d9750f02f983ddb37439ca47ced7149 \
+    --hash=sha256:681eb3d7e02e3c3655d1b16059fbfb605ac464c834a0c629048a30fad2b27489 \
+    --hash=sha256:6c409c0deba34f147f77efaa67b8e4bb83d2f11c8806405f76397ae5b8c0d1c9 \
+    --hash=sha256:7095f6fbfaa55defb6b733cfeb14efaae7a29f0b59d8cf213be4e7ca0b857b80 \
+    --hash=sha256:70c610f6cbe4b9fce272c407dd9d07e33e6bf7b4aa1b7ffb6f6ded8e634e3592 \
+    --hash=sha256:72814c01533f51d68702802d74f77ea026b5ec52793c791e2da806a3844a46c3 \
+    --hash=sha256:7a4826ad2bd6b07ca615c74ab91f32f6c96d08f6fcc3902ceeedaec8cdc3bcd6 \
+    --hash=sha256:7c70087bfee18a42b4040bb9ec1ca15a08242cf5867c58726530bdf3945672ed \
+    --hash=sha256:855eafa5d5a2034b4621c74925d89c5efef61418570e5ef9b37717d9c796419c \
+    --hash=sha256:8700f06d0ce6f128de3ccdbc1acaea1ee264d2caa9ca05daaf492fde7c2a7200 \
+    --hash=sha256:89f1b185a01fe560bc8ae5f619e924407efca2191b56ce749ec84982fc59a32a \
+    --hash=sha256:8b2c760cfc7042b27ebdb4a43a4453bd829a5742503599144d54a032c5dc7e9e \
+    --hash=sha256:8c2f5e83493748286002f9369f3e6607c565a6a90425a3a1fef5ae32a36d749d \
+    --hash=sha256:8e098148dd37b4ce3baca71fb394c81dc5d9c7728c95df695d2dca218edf40e6 \
+    --hash=sha256:94aea8eff76ee6d1cdacb07dd2123a68283cb5569e0250feab1240058f53b623 \
+    --hash=sha256:95eb302ff792e12aba9a8b8f8474ab229a83c103d74a750ec0bd1c1eea32e669 \
+    --hash=sha256:9bd9b3b31adcb054116447ea22caa61a285d92e94d710aa5ec97992ff5eb7cf3 \
+    --hash=sha256:9e608aafdb55eb9f255034709e20d5a83b6d60c054df0802fa9c9883d0a937aa \
+    --hash=sha256:a103b3a7069b62f5d4890ae1b8f0597618f628b286b03d4bc9195230b154bfa9 \
+    --hash=sha256:a386ebe437176aab38c041de1260cd3ea459c6ce5263594399880bbc398225b2 \
+    --hash=sha256:a38856a971c602f98472050165cea2cdc97709240373041b69030be15047691f \
+    --hash=sha256:a401b4598e5d3f4a9a811f3daf42ee2291790c7f9d74b18d75d6e21dda98a1a1 \
+    --hash=sha256:a7647ebdfb9682b7bb97e2a5e7cb6ae735b1c25008a70b906aecca294ee96cf4 \
+    --hash=sha256:aaf63899c94de41fe3cf934601b0f7ccb6b428c6e4eeb80da72c58eab077b19a \
+    --hash=sha256:b0dac0ff919ba34d4df1b6131f59ce95b08b9065233446be7e459f95554c0dc8 \
+    --hash=sha256:baacc6aee0b2ef6f3d308e197b5d7a81c0e70b06beae1f1fcacffdbd124fe0e3 \
+    --hash=sha256:bf420121d4c8dce6b889f0e8e4ec0ca34b7f40186203f06a946fa0276ba54029 \
+    --hash=sha256:c04a46716adde8d927adb9457bbe39cf473e1e2c2f5d0a16ceb837e5d841ad4f \
+    --hash=sha256:c0b21078a4b56965e2b12f247467b234734491897e99c1d51cee628da9786959 \
+    --hash=sha256:c1c76a1743432b4b60ab3358c937a3fe1341c828ae6194108a94c69028247f22 \
+    --hash=sha256:c4983bf937209c57240cff65906b18bb35e64ae872da6a0db937d7b4af845dd7 \
+    --hash=sha256:c4fb39a81950ec280984b3a44f5bd12819953dc5fa3a7e6fa7a80db5ee853952 \
+    --hash=sha256:c57921cda3a80d0f2b8aec7e25c8aa14479ea92b5b51b6876d975d925a2ea346 \
+    --hash=sha256:c8063cf17b19661471ecbdb3df1c84f24ad2e389e326ccaf89e3fb2484d8dd7e \
+    --hash=sha256:ccd16eb18a849fd8dcb23e23380e2f0a354e8daa0c984b8a732d9cfaba3a776d \
+    --hash=sha256:cd6dbe0238f7743d0efe563ab46294f54f9bc8f4b9bcf57c3c666cc5bc9d1299 \
+    --hash=sha256:d62e51710986674142526ab9f78663ca2b0726066ae26b78b22e0f5e571238dd \
+    --hash=sha256:db901e2ac34c931d73054d9797383d0f8009991e723dab15109740a63e7f902a \
+    --hash=sha256:e03b8895a6990c9ab2cdcd0f2fe44088ca1c65ae592b8f795c3294af00a461c3 \
+    --hash=sha256:e1c8a2f4c69e08e89632defbfabec2feb8a8d99edc9f89ce33c4b9e36ab63037 \
+    --hash=sha256:e4b749b9cc6ee664a3300bb3a273c1ca8068c46be705b6c31cf5d276f8628a94 \
+    --hash=sha256:e6a5bf2cba5ae1bb80b154ed68a3cfa2fa00fde979a7f50d6598d3e17d9ac20c \
+    --hash=sha256:e857a2232ba53ae940d3456f7533ce6ca98b81917d47adc3c7fd55dad8fab858 \
+    --hash=sha256:ee4006268ed33370957f55bf2e6f4d263eaf4dc3cfc473d1d90baff6ed36ce4a \
+    --hash=sha256:eef9df1eefada2c09a5e7a40991b9fc6ac6ef20b1372abd48d2794a316dc0449 \
+    --hash=sha256:f058f6963fd82eb143c692cecdc89e075fa0828db2e5b291070485390b2f1c9c \
+    --hash=sha256:f25c229a6ba38a35ae6e25ca1264621cc25d4d38dca2942a7fce0b67a4efe918 \
+    --hash=sha256:f2a1d0fd4242bd8643ce6f98927cf9c04540af6efa92323e9d3124f57727bfc1 \
+    --hash=sha256:f7560358a6811e52e9c4d142d497f1a6e10103d3a6881f18d04dbce3729c0e2c \
+    --hash=sha256:f779d3ad205f108d14e99bb3859aa7dd8e9c68874617c72354d7ecaec2a054ac \
+    --hash=sha256:f87f746ee241d30d6ed93969de31e5ffd09a2961a051e60ae6bddde9ec3583aa
+idna==3.4 ; python_version >= "3.11" and python_version < "4.0" \
+    --hash=sha256:814f528e8dead7d329833b91c5faa87d60bf71824cd12a7530b5526063d02cb4 \
+    --hash=sha256:90b77e79eaa3eba6de819a0c442c0b4ceefc341a7a2ab77d7562bf49f425c5c2
+mlogconfig==0.2.5 ; python_version >= "3.11" and python_version < "4.0" \
+    --hash=sha256:24f3a887684d1a5bc6643f4b42bf4db7b56d3356df1565cbe8ec26e4f9050514 \
+    --hash=sha256:db398281f298f857a8f011511ff67298b8a60d1a7f950f364a1006bf3966f3cb
+python-dotenv==1.0.0 ; python_version >= "3.11" and python_version < "4.0" \
+    --hash=sha256:a8df96034aae6d2d50a4ebe8216326c61c3eb64836776504fcca410e5937a3ba \
+    --hash=sha256:f5971a9226b701070a4bf2c38c89e5a3f0d64de8debda981d1db98583009122a
+requests==2.31.0 ; python_version >= "3.11" and python_version < "4.0" \
+    --hash=sha256:58cd2187c01e70e6e26505bca751777aa9f2ee0b7f4300988b709f44e013003f \
+    --hash=sha256:942c5a758f98d790eaed1a29cb6eefc7ffb0d1cf7af05c3d2791656dbd6ad1e1
+urllib3==2.0.3 ; python_version >= "3.11" and python_version < "4.0" \
+    --hash=sha256:48e7fafa40319d358848e1bc6809b208340fafe2096f1725d05d67443d0483d1 \
+    --hash=sha256:bee28b5e56addb8226c96f7f13ac28cb4c301dd5ea8a6ca179c0b9835e032825

src/client.py

@@ -2,5 +2,15 @@
 TODO: Update Summary Docstring
 """
 
-from src.rapid7 import Rapid7
-from src.paloalto import PaloAlto
+import os
+import src.rapid7
+import src.paloalto
+
+from src.rapid7.api_r7_asset_group import create_high_risk_asset_group
+
+class AssetGroupCreator:
+    def __init__(self):
+        pass
+
+    def create_high_risk_asset_group(self, name):
+        create_high_risk_asset_group(name)

src/paloalto/__init__.py

@@ -1 +0,0 @@
-import src.paloalto.cortex_xdr

src/rapid7/__init__.py

@@ -1,9 +0,0 @@
-"""
-This package provides functions for interacting with the Rapid7 InsightVM API.
-"""
-
-from .api_r7_auth import *
-
-from .api_r7_isvm import *
-
-from .api_r7_asset_group import *

src/rapid7/api_r7_asset_group.py

@@ -6,23 +6,42 @@
 import urllib3
 from dotenv import load_dotenv
 import requests
-from src.rapid7.api_r7_auth import load_r7_isvm_api_credentials, get_isvm_basic_auth_header
+from api_r7_auth import load_r7_isvm_api_credentials, get_isvm_basic_auth_header
+from src.rapid7.api_r7_isvm_get_assets import API_ENDPOINT
 
 # Load environment variables from .env file
 load_dotenv()
 
 # Set up logging
 logging.basicConfig(filename='api_r7_asset_group.log', level=logging.ERROR)
 
-def create_high_risk_asset_group():
+def create_asset_group():
     """
-    Creates a dynamic asset group based on criteria and prints out the ID with a URL.
+    Creates an asset group based on criteria and prints out the ID with a URL.
     """
+    url, headers, payload = _set_up_request()
+    try:
+        response = requests.post(
+            url,
+            headers=headers,
+            json=payload,
+            verify=False,
+            timeout=90
+        ).json()
+        ag_id = response["id"]
+        print(
+            f"Asset Group {ag_id} created and can be found at {url}/group.jsp?groupid={ag_id}"
+        )
+    except requests.exceptions.RequestException as error:
+        logging.exception("Error creating asset group: %s", error)
+        raise
+
+def _set_up_request():
     urllib3.disable_warnings()
     # Get the ISVM API credentials and base URL from environment variables
     _, _, isvm_base_url = load_r7_isvm_api_credentials()
     auth_headers = get_isvm_basic_auth_header()
-    url = f"{isvm_base_url}/api/3/asset_groups"
+    url = f"{isvm_base_url}/api/3/{api_endpoint}"
     headers = {
         "Content-Type": "application/json",
         **auth_headers,
@@ -46,18 +65,5 @@ def create_high_risk_asset_group():
         "type": "dynamic",
         "vulnerabilities": {},
     }
-    try:
-        response = requests.post(
-            url,
-            headers=headers,
-            json=payload,
-            verify=False,
-            timeout=90
-        ).json()
-        ag_id = response["id"]
-        print(
-            f"Asset Group {ag_id} created and can be found at {isvm_base_url}/group.jsp?groupid={ag_id}"
-        )
-    except requests.exceptions.RequestException as error:
-        logging.exception("Error creating asset group: %s", error)
-        raise
+    return url, headers, payload
+

src/rapid7/api_r7_endpoints.py

@@ -0,0 +1,30 @@
+"""
+This module provides a class that defines the API endpoints available for Rapid7 InsightVM API v3.
+"""
+
+class InsightVMAPIv3Endpoints:
+    """
+    A class that defines the API endpoints available for Rapid7 InsightVM API v3.
+    """
+    ASSET_GROUP = "asset-group"
+    ASSET = "asset"
+    AUTHENTICATION = "authentication"
+    CONFIGURATION = "configuration"
+    CREDENTIAL = "credential"
+    DISCOVERY_CONNECTION = "discovery-connection"
+    DISCOVERY_PROFILE = "discovery-profile"
+    ENGINE = "engine"
+    EXPORT = "export"
+    IMPORT = "import"
+    LICENSE = "license"
+    REPORT = "report"
+    ROLE = "role"
+    SCAN = "scan"
+    SITE = "site"
+    TAG = "tag"
+    USER = "user"
+    VULNERABILITY_EXCEPTION = "vulnerability-exception"
+    VULNERABILITY = "vulnerability"
+    VULNERABILITY_CHECK = "vulnerability-check"
+    VULNERABILITY_DEFINITION = "vulnerability-definition"
+    VULNERABILITY_SOLUTION = "vulnerability-solution"