p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

Agent: Add HTTPIslandAPIClient._api_url attribute

This commit is contained in:
Mike Salvatore 2022-09-19 19:57:28 -04:00
parent 841183d8e7
commit cb8fda0bec
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/infection_monkey/island_api_client/http_island_api_client.py
View File

@ -39,11 +39,12 @@ class HTTPIslandAPIClient(IIslandAPIClient):
) )
self._island_server = island_server self._island_server = island_server
self._api_url = f"https://{self._island_server}/api"
@handle_island_errors @handle_island_errors
def send_log(self, log_contents: str): def send_log(self, log_contents: str):
requests.post( # noqa: DUO123 requests.post( # noqa: DUO123
f"https://{self._island_server}/api/log", f"{self._api_url}/log",
json=log_contents, json=log_contents,
verify=False, verify=False,
timeout=MEDIUM_REQUEST_TIMEOUT, timeout=MEDIUM_REQUEST_TIMEOUT,
@ -52,7 +53,7 @@ class HTTPIslandAPIClient(IIslandAPIClient):
@handle_island_errors @handle_island_errors
def get_pba_file(self, filename: str): def get_pba_file(self, filename: str):
response = requests.get( # noqa: DUO123 response = requests.get( # noqa: DUO123
f"https://{self._island_server}/api/pba/download/{filename}", f"{self._api_url}/pba/download/{filename}",
verify=False, verify=False,
timeout=LONG_REQUEST_TIMEOUT, timeout=LONG_REQUEST_TIMEOUT,
) )