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

Added some documentation

This commit is contained in:
Shay Nehmad 2020-01-19 16:28:04 +02:00
parent 422fe6ff06
commit d52672f4d7
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/common/cloud/gcp/gcp_instance.py
View File

@ -5,10 +5,15 @@ from common.cloud.environment_names import GCP
from common.cloud.instance import CloudInstance from common.cloud.instance import CloudInstance
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
GCP_METADATA_SERVICE_URL = "http://metadata.google.internal/" GCP_METADATA_SERVICE_URL = "http://metadata.google.internal/"
class GcpInstance(CloudInstance): class GcpInstance(CloudInstance):
"""
Used to determine if on GCP. See https://cloud.google.com/compute/docs/storing-retrieving-metadata#runninggce
"""
def is_instance(self): def is_instance(self):
return self.on_gcp return self.on_gcp
@ -16,9 +21,6 @@ class GcpInstance(CloudInstance):
return GCP return GCP
def __init__(self): def __init__(self):
"""
Determines if on GCP.
"""
self.on_gcp = False self.on_gcp = False
try: try:
@ -26,7 +28,7 @@ class GcpInstance(CloudInstance):
response = requests.get(GCP_METADATA_SERVICE_URL) response = requests.get(GCP_METADATA_SERVICE_URL)
if response: if response:
logger.debug("Got response, so probably on GCP. Trying to parse.") logger.debug("Got ok metadata response: on GCP")
self.on_gcp = True self.on_gcp = True
if "Metadata-Flavor" not in response.headers: if "Metadata-Flavor" not in response.headers:
@ -37,5 +39,5 @@ class GcpInstance(CloudInstance):
else: else:
logger.warning("On GCP, but metadata response not ok: {}".format(response.status_code)) logger.warning("On GCP, but metadata response not ok: {}".format(response.status_code))
except requests.RequestException: except requests.RequestException:
logger.debug("Failed to get response from GCP metadata service: This instance is not on GCP.") logger.debug("Failed to get response from GCP metadata service: This instance is not on GCP")
self.on_gcp = False self.on_gcp = False