Skip to content

Commit

Permalink
Merge pull request #83 from Dynatrace-James-Kitson/management-zone-ru…
Browse files Browse the repository at this point in the history 
…le-type

correct rule type enums for management zones
  • Loading branch information
@Dynatrace-James-Kitson
Dynatrace-James-Kitson authored Mar 20, 2024
2 parents 699b3ed + 29b094a commit 10550c6
Show file tree
Hide file tree
Showing 2 changed files with 83 additions and 16 deletions.
97 changes: 82 additions & 15 deletions dynatrace/configuration_v1/management_zones.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
"""

from typing import List, Dict, Any
from enum import Enum

from dynatrace.environment_v2.schemas import ConfigurationMetadata
from dynatrace.dynatrace_object import DynatraceObject
Expand All @@ -26,25 +27,68 @@
ConditionKeyAttribute,
ConditionKeyType,
PropagationType,
RuleType,
)


class RuleType(Enum):
APPMON_SERVER = "APPMON_SERVER"
APPMON_SYSTEM_PROFILE = "APPMON_SYSTEM_PROFILE"
AWS_ACCOUNT = "AWS_ACCOUNT"
AWS_APPLICATION_LOAD_BALANCER = "AWS_APPLICATION_LOAD_BALANCER"
AWS_AUTO_SCALING_GROUP = "AWS_AUTO_SCALING_GROUP"
AWS_CLASSIC_LOAD_BALANCER = "AWS_CLASSIC_LOAD_BALANCER"
AWS_NETWORK_LOAD_BALANCER = "AWS_NETWORK_LOAD_BALANCER"
AWS_RELATIONAL_DATABASE_SERVICE = "AWS_RELATIONAL_DATABASE_SERVICE"
AZURE = "AZURE"
BROWSER_MONITOR = "BROWSER_MONITOR"
CLOUD_APPLICATION = "CLOUD_APPLICATION"
CLOUD_APPLICATION_NAMESPACE = "CLOUD_APPLICATION_NAMESPACE"
CLOUD_FOUNDRY_FOUNDATION = "CLOUD_FOUNDRY_FOUNDATION"
CUSTOM_APPLICATION = "CUSTOM_APPLICATION"
CUSTOM_DEVICE = "CUSTOM_DEVICE"
CUSTOM_DEVICE_GROUP = "CUSTOM_DEVICE_GROUP"
DATA_CENTER_SERVICE = "DATA_CENTER_SERVICE"
ENTERPRISE_APPLICATION = "ENTERPRISE_APPLICATION"
ESXI_HOST = "ESXI_HOST"
EXTERNAL_MONITOR = "EXTERNAL_MONITOR"
HOST = "HOST"
HOST_GROUP = "HOST_GROUP"
HTTP_MONITOR = "HTTP_MONITOR"
KUBERNETES_CLUSTER = "KUBERNETES_CLUSTER"
KUBERNETES_SERVICE = "KUBERNETES_SERVICE"
MOBILE_APPLICATION = "MOBILE_APPLICATION"
MULTIPROTOCOL_MONITOR = "MULTIPROTOCOL_MONITOR"
OPENSTACK_ACCOUNT = "OPENSTACK_ACCOUNT"
PROCESS_GROUP = "PROCESS_GROUP"
QUEUE = "QUEUE"
SERVICE = "SERVICE"
WEB_APPLICATION = "WEB_APPLICATION"
NONE = None


class ManagementZoneService:
ENDPOINT = "/api/config/v1/managementZones"

def __init__(self, http_client: HttpClient):
self.__http_client = http_client

def list(self, page_size: int = 200) -> PaginatedList["ManagementZoneShortRepresentation"]:
def list(
self, page_size: int = 200
) -> PaginatedList["ManagementZoneShortRepresentation"]:
"""
List all management zones.
:param page_size: The number of results per result page. Must be between 1 and 500
Default value : 200
"""
params = {"pageSize": page_size}
return PaginatedList(ManagementZoneShortRepresentation, self.__http_client, f"{self.ENDPOINT}", params, list_item="values")
return PaginatedList(
ManagementZoneShortRepresentation,
self.__http_client,
f"{self.ENDPOINT}",
params,
list_item="values",
)

def get(self, management_zone_id: str) -> "ManagementZone":
"""Gets the description of a management zone referenced by ID.
Expand All @@ -53,16 +97,22 @@ def get(self, management_zone_id: str) -> "ManagementZone":
:returns Event: the requested management zone
"""
response = self.__http_client.make_request(path=f"{self.ENDPOINT}/{management_zone_id}")
return ManagementZone(raw_element=response.json(), http_client=self.__http_client)
response = self.__http_client.make_request(
path=f"{self.ENDPOINT}/{management_zone_id}"
)
return ManagementZone(
raw_element=response.json(), http_client=self.__http_client
)

def delete(self, management_zone_id: str):
"""Deletes the specified management zone
:param networkzone_id: the ID of the management zone
:return: HTTP response
"""
return self.__http_client.make_request(path=f"{self.ENDPOINT}/{management_zone_id}", method="DELETE")
return self.__http_client.make_request(
path=f"{self.ENDPOINT}/{management_zone_id}", method="DELETE"
)


class ComparisonBasic(DynatraceObject):
Expand All @@ -76,14 +126,18 @@ def _create_from_raw_data(self, raw_element: Dict[str, Any]):

class ConditionKey(DynatraceObject):
def _create_from_raw_data(self, raw_element: Dict[str, Any]):
self.attribute: ConditionKeyAttribute = ConditionKeyAttribute(raw_element.get("attribute"))
self.attribute: ConditionKeyAttribute = ConditionKeyAttribute(
raw_element.get("attribute")
)
self.type: ConditionKeyType = ConditionKeyType(raw_element.get("type"))


class EntityRuleEngineCondition(DynatraceObject):
def _create_from_raw_data(self, raw_element: Dict[str, Any]):
self.key: ConditionKey = ConditionKey(raw_element=raw_element.get("key"))
self.comparison_info: ComparisonBasic = ComparisonBasic(raw_element=raw_element.get("comparisonInfo"))
self.comparison_info: ComparisonBasic = ComparisonBasic(
raw_element=raw_element.get("comparisonInfo")
)


class EntitySelectorBasedManagementZoneRule(DynatraceObject):
Expand All @@ -98,21 +152,32 @@ def _create_from_raw_data(self, raw_element: Dict[str, Any]):
self.type: RuleType = RuleType(raw_element.get("type"))
self.enabled: bool = raw_element.get("enabled")
self.value_format: str = raw_element.get("valueFormat")
self.propagation_types: List[PropagationType] = [PropagationType(prop_type) for prop_type in (raw_element.get("propagationTypes") or [])]
self.propagation_types: List[PropagationType] = [
PropagationType(prop_type)
for prop_type in (raw_element.get("propagationTypes") or [])
]
self.conditions: List[EntityRuleEngineCondition] = [
EntityRuleEngineCondition(raw_element=condition) for condition in (raw_element.get("conditions") or [])
EntityRuleEngineCondition(raw_element=condition)
for condition in (raw_element.get("conditions") or [])
]


class ManagementZone(DynatraceObject):
def _create_from_raw_data(self, raw_element: Dict[str, Any]):
self.metadata: ConfigurationMetadata = ConfigurationMetadata(self._http_client, None, raw_element.get("metadata"))
self.metadata: ConfigurationMetadata = ConfigurationMetadata(
self._http_client, None, raw_element.get("metadata")
)
self.id: str = raw_element.get("id")
self.name: str = raw_element.get("name")
self.description: str = raw_element.get("description")
self.rules: List[ManagementZoneRule] = [ManagementZoneRule(raw_element=rule) for rule in raw_element.get("rules")]
self.entity_selector_based_rules: List[EntitySelectorBasedManagementZoneRule] = [
EntitySelectorBasedManagementZoneRule(raw_element=rule) for rule in (raw_element.get("entitySelectorBasedRules") or [])
self.rules: List[ManagementZoneRule] = [
ManagementZoneRule(raw_element=rule) for rule in raw_element.get("rules")
]
self.entity_selector_based_rules: List[
EntitySelectorBasedManagementZoneRule
] = [
EntitySelectorBasedManagementZoneRule(raw_element=rule)
for rule in (raw_element.get("entitySelectorBasedRules") or [])
]


Expand All @@ -121,5 +186,7 @@ def get_full_configuration(self):
"""
Get the full configuration for this management zone short representation.
"""
response = self._http_client.make_request(f"{ManagementZoneService.ENDPOINT}/{self.id}").json()
response = self._http_client.make_request(
f"{ManagementZoneService.ENDPOINT}/{self.id}"
).json()
return ManagementZone(http_client=self._http_client, raw_element=response)
2 changes: 1 addition & 1 deletion setup.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

setup(
name="dt",
version="1.1.62",
version="1.1.63",
packages=find_packages(),
install_requires=["requests>=2.22"],
tests_require=["pytest", "mock", "tox"],
Expand Down

0 comments on commit 10550c6

Please sign in to comment.