Merge pull request #9 from dynatrace-api-framework/test

Propogating to Master

Author: Philipose

Pipfile

@@ -5,6 +5,7 @@ verify_ssl = true
 
 [dev-packages]
 pylint = "*"
+autopep8 = "*"
 
 [packages]
 requests = "*"

Pipfile.lock

@@ -1,15 +1,25 @@
 """Replace active variables with another set"""
-import sys
+from platform import system
+import argparse
 import os
 
-def replace_env():
+def replace_set(set_file):
   """Replace Variable File"""
-  print("Enter Set to Import: ", end='')
-  new_env = input()
-
-  if "windows" in sys.platform:
-    os.system("copy variable_sets\\" + str(new_env) + ".py user_variables.py")
+  # Options are Darwin, Linux, Java and Windows. Java not supported
+  if "Windows" in system():
+    os.system("copy variable_sets\\" + str(set_file) + ".py user_variables.py")
   else:
-    os.system("cp variable_sets/" + str(new_env) + ".py user_variables.py")
+    os.system("cp variable_sets/" + str(set_file) + ".py user_variables.py")
+
+def get_variable_set_file(variable_set_arg):
+  """Checks if the set file was provided via arg else prompt"""
+  if variable_set_arg:
+    return variable_set_arg
+  return input("Enter Set to Import: ")
 
-replace_env()
+if __name__ == '__main__':
+  parser = argparse.ArgumentParser()
+  parser.add_argument('--set-file', '-s')
+  args = parser.parse_args()
+  set_file = get_variable_set_file(args.set_file)
+  replace_set(set_file)

change_variables.py

@@ -1,6 +1,8 @@
 """SSO Operations for Dynatrace"""
 import dynatrace.requests.request_handler as rh
 
+ENDPOINT = "sso/ssoProvider"
+
 def disable_sso (cluster):
   """Disable SSO Sign-in"""
   disable_payload = {
@@ -10,7 +12,7 @@ def disable_sso (cluster):
       "ssoGroupsEnabled": False,
       "ssoLoginDisabled": True
   }
-  response = rh.cluster_post(cluster, "sso/ssoProvider", json=disable_payload)
+  response = rh.cluster_post(cluster, ENDPOINT, json=disable_payload)
   return response.status_code
 
 def enable_sso(cluster, disable_local=False, groups_enabled=False, is_openid=False):
@@ -30,9 +32,9 @@ def enable_sso(cluster, disable_local=False, groups_enabled=False, is_openid=Fal
   if is_openid:
     enable_payload['ssoProvider'] = "OIDC"
 
-  response = rh.cluster_post(cluster, "sso/ssoProvider", json=enable_payload)
+  response = rh.cluster_post(cluster, ENDPOINT, json=enable_payload)
   return response.status_code
 
 def get_sso_status (cluster):
-  response = rh.cluster_get(cluster, "sso/ssoProvider")
+  response = rh.cluster_get(cluster, ENDPOINT)
   return response.json()

dynatrace/cluster/sso.py

@@ -63,7 +63,6 @@ def delete_app_groups (cluster, app_name):
           cluster,
           "groups/" + group_id
       )
-  return
 
 def delete_app_groups_setwide(app_name):
   """Create Dynatrace User Groups for Applications"""

dynatrace/cluster/user_groups.py

@@ -4,6 +4,11 @@
 import requests
 from urllib3.exceptions import InsecureRequestWarning
 
+HTTPS_STR = "https://"
+CLUSTER_V1_PATH = "/api/v1.0/onpremise/"
+ENV_API_V1 = "/api/v1/"
+CONFIG_API_V1 = "/api/config/v1/"
+
 OLD_MERGE_ENVIRONMENT_SETTINGS = requests.Session.merge_environment_settings
 
 @contextlib.contextmanager
@@ -55,7 +60,7 @@ def sanitize_endpoint (endpoint):
 
 def generate_tenant_url(cluster, tenant):
   """Generate URL based on SaaS or Managed"""
-  url = "https://"
+  url = HTTPS_STR
   if cluster["is_managed"]:
     url = url + cluster['url'] + "/e/" + cluster['tenant'][tenant]
   else:
@@ -75,7 +80,7 @@ def cluster_get(cluster, endpoint, params=None):
     params['Api-Token'] = cluster['cluster_token']
 
     response = requests.get(
-        "https://" + cluster['url'] + "/api/v1.0/onpremise/" + endpoint,
+        HTTPS_STR + cluster['url'] + CLUSTER_V1_PATH + endpoint,
         params=params,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"])
     )
@@ -96,7 +101,7 @@ def cluster_post(cluster, endpoint, params=None, json=None):
     params['Api-Token'] = cluster['cluster_token']
 
     response =  requests.post(
-        "https://" + cluster['url'] + "/api/v1.0/onpremise/" + endpoint,
+        HTTPS_STR + cluster['url'] + CLUSTER_V1_PATH + endpoint,
         params=params,
         json=json,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"])
@@ -117,7 +122,7 @@ def cluster_put(cluster, endpoint, params=None, json=None):
     params['Api-Token'] = cluster['cluster_token']
 
     response =  requests.put(
-        "https://" + cluster['url'] + "/api/v1.0/onpremise/" + endpoint,
+        HTTPS_STR + cluster['url'] + CLUSTER_V1_PATH + endpoint,
         params=params,
         json=json,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"])
@@ -137,7 +142,7 @@ def cluster_delete(cluster, endpoint, params=None, json=None):
   with no_ssl_verification():
     params['Api-Token'] = cluster['cluster_token']
     response =  requests.delete(
-        "https://" + cluster['url'] + "/api/v1.0/onpremise/" + endpoint,
+        HTTPS_STR + cluster['url'] + CLUSTER_V1_PATH + endpoint,
         params=params,
         json=json,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"])
@@ -155,7 +160,7 @@ def env_get(cluster, tenant, endpoint, params=None):
   with no_ssl_verification():
     params['Api-Token'] = cluster['api_token'][tenant]
     response = requests.get(
-        generate_tenant_url(cluster, tenant) + "/api/v1/" + endpoint,
+        generate_tenant_url(cluster, tenant) + ENV_API_V1 + endpoint,
         params=params,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"])
     )
@@ -173,7 +178,7 @@ def env_post(cluster, tenant, endpoint, params=None, json=None):
     params['Api-Token'] = cluster['api_token'][tenant]
 
     response = requests.post(
-        generate_tenant_url(cluster, tenant) + "/api/v1/" + endpoint,
+        generate_tenant_url(cluster, tenant) + ENV_API_V1 + endpoint,
         params=params,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"]),
         json=json
@@ -192,7 +197,7 @@ def env_put(cluster, tenant, endpoint, params=None, json=None):
     params['Api-Token'] = cluster['api_token'][tenant]
 
     response = requests.put(
-        generate_tenant_url(cluster, tenant) + "/api/v1/" + endpoint,
+        generate_tenant_url(cluster, tenant) + ENV_API_V1 + endpoint,
         params=params,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"]),
         json=json
@@ -210,7 +215,7 @@ def env_delete(cluster, tenant, endpoint, params=None):
   with no_ssl_verification():
     params['Api-Token'] = cluster['api_token'][tenant]
     response = requests.delete(
-        generate_tenant_url(cluster, tenant) + "/api/v1/" + endpoint,
+        generate_tenant_url(cluster, tenant) + ENV_API_V1 + endpoint,
         params=params,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"])
     )
@@ -229,7 +234,7 @@ def config_get(cluster, tenant, endpoint, params=None, json=None):
     params['Api-Token'] = cluster['api_token'][tenant]
 
     response = requests.get(
-        generate_tenant_url(cluster, tenant) + "/api/config/v1/" + endpoint,
+        generate_tenant_url(cluster, tenant) + CONFIG_API_V1 + endpoint,
         params=params,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"]),
         json=json
@@ -248,7 +253,7 @@ def config_post(cluster, tenant, endpoint, params=None, json=None):
     params['Api-Token'] = cluster['api_token'][tenant]
 
     response = requests.post(
-        generate_tenant_url(cluster, tenant) + "/api/config/v1/" + endpoint,
+        generate_tenant_url(cluster, tenant) + CONFIG_API_V1 + endpoint,
         params=params,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"]),
         json=json
@@ -267,7 +272,7 @@ def config_put(cluster, tenant, endpoint, params=None, json=None):
     params['Api-Token'] = cluster['api_token'][tenant]
 
     response = requests.put(
-        generate_tenant_url(cluster, tenant) + "/api/config/v1/" + endpoint,
+        generate_tenant_url(cluster, tenant) + CONFIG_API_V1 + endpoint,
         params=params,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"]),
         json=json
@@ -286,7 +291,7 @@ def config_delete(cluster, tenant, endpoint, params=None, json=None):
     params['Api-Token'] = cluster['api_token'][tenant]
 
     response = requests.delete(
-        generate_tenant_url(cluster, tenant) + "/api/config/v1/" + endpoint,
+        generate_tenant_url(cluster, tenant) + CONFIG_API_V1 + endpoint,
         params=params,
         verify=(True if "verify_ssl" not in cluster else cluster ["verify_ssl"]),
         json=json

dynatrace/requests/request_handler.py

@@ -4,6 +4,9 @@
 import dynatrace.requests.request_handler as rh
 import user_variables as uv
 
+
+MZ_ENDPOINT = "/maintenanceWindows/"
+
 class InvalidDateFormatException(ValueError):
   def __init__(self, required_format):
     self.required_format = required_format
@@ -113,28 +116,28 @@ def generate_schedule(recurrence_type, start_time, duration, range_start, range_
 
 def create_window (cluster, tenant, json):
   """Create Maintenance Window"""
-  response = rh.config_post(cluster, tenant, '/maintenanceWindows', json=json)
+  response = rh.config_post(cluster, tenant, MZ_ENDPOINT, json=json)
   return response.status_code
 
 def update_window (cluster, tenant, window_id, json):
   """Update Maintenance Window"""
-  response = rh.config_put(cluster, tenant, '/maintenanceWindows/' + window_id, json=json)
+  response = rh.config_put(cluster, tenant, MZ_ENDPOINT + window_id, json=json)
   return response.status_code
 
 def delete_window (cluster, tenant, window_id):
   """Delete Maintenance Window"""
-  response = rh.config_delete(cluster, tenant, '/maintenanceWindows/' + window_id)
+  response = rh.config_delete(cluster, tenant, MZ_ENDPOINT + window_id)
   return response.status_code
 
 def get_windows (cluster, tenant):
   """Return List of Maintenance Windows in Effect"""
-  response = rh.config_get(cluster, tenant, '/maintenanceWindows')
+  response = rh.config_get(cluster, tenant, MZ_ENDPOINT)
   return response.json()
 
 
 def get_window (cluster, tenant, window_id):
   """Return Maintenance Window Details"""
-  response = rh.config_get(cluster, tenant, '/maintenanceWindows/' + window_id)
+  response = rh.config_get(cluster, tenant, MZ_ENDPOINT + window_id)
   return response.json()
 
 def parse_tag(tag_string):

dynatrace/tenant/maintenance.py

@@ -3,10 +3,12 @@
 import json
 from dynatrace.requests import request_handler as rh
 
+ENDPOINT = "/service/requestAttributes/"
+
 def pull_to_files(cluster, tenant, ignore_disabled=True):
   """Pull files from an environment to local"""
   # API Calls needed: Pull RA, take the ID and pull the details of each RA
-  all_ra_call = rh.config_get(cluster, tenant, "/service/requestAttributes")
+  all_ra_call = rh.config_get(cluster, tenant, ENDPOINT)
   all_ra_json = all_ra_call.json()
   all_ra_json = all_ra_json['values']
   #print (json.dumps(all_ra_json, indent=2))
@@ -15,7 +17,7 @@ def pull_to_files(cluster, tenant, ignore_disabled=True):
     single_ra_call = rh.config_get(
         cluster,
         tenant,
-        "/service/requestAttributes/" + str(request_attribute['id'])
+        ENDPOINT + str(request_attribute['id'])
     )
     if single_ra_call.status_code == 200:
       single_ra_json = single_ra_call.json()
@@ -34,7 +36,7 @@ def push_from_files(file_list, cluster, tenant):
   """Push Request Attributes in JSONs to a tenant"""
 
   # Checks for Existing RAs to update them put request rather than a post that would fail
-  existing_ra_get = rh.config_get(cluster, tenant, "/service/requestAttributes")
+  existing_ra_get = rh.config_get(cluster, tenant, ENDPOINT)
   existing_ra_json = existing_ra_get.json()
   existing_ra_json = existing_ra_json['values']
   existing_ra_list = {}
@@ -48,14 +50,14 @@ def push_from_files(file_list, cluster, tenant):
         single_ra_post = rh.config_put(
             cluster,
             tenant,
-            "/service/requestAttributes/" + existing_ra_list[file],
+            ENDPOINT + existing_ra_list[file],
             json=ra_json
         )
       else:
         single_ra_post = rh.config_post(
             cluster,
             tenant,
-            "/service/requestAttributes",
+            ENDPOINT,
             json=ra_json
         )
       if single_ra_post.status_code >= 400:

dynatrace/tenant/request_attributes.py

@@ -1,20 +1,22 @@
 from dynatrace.requests import request_handler as rh
 
+ENDPOINT = "timeseries/"
+
 def get_timeseries_list(cluster, tenant, params=None):
   """Get List of Timeseries Metics"""
-  response = rh.env_get(cluster, tenant, "timeseries", params=params)
+  response = rh.env_get(cluster, tenant, ENDPOINT, params=params)
   return response.json()
 
 def get_timeseries_metric (cluster, tenant, metric, params=None):
   """Get Timeseries Metric"""
   #Chose to do GET, but could also be done as POST. Don't think there are any advantages to post
-  response = rh.env_get(cluster, tenant, "timeseries/" + metric, params=params)
+  response = rh.env_get(cluster, tenant, ENDPOINT + metric, params=params)
   return response.json()
 
 def create_custom_metric (cluster, tenant, metric, json, params=None):
-  response = rh.env_put(cluster, tenant, "timeseries/" + metric, params=params, json=json)
+  response = rh.env_put(cluster, tenant, ENDPOINT + metric, params=params, json=json)
   return response.status_code
 
 def delete_custom_metic (cluster, tenant, metric):
-  response = rh.env_delete(cluster, tenant, "timeseries/" + metric)
+  response = rh.env_delete(cluster, tenant, ENDPOINT + metric)
   return response.status_code