Merge pull request #12 from mattvaldes/master

Add support for custom DefectDojo URL prefix

Author: aaronweaver

defectdojo_api/defectdojo.py

@@ -33,6 +33,11 @@ def __init__(self, host, api_key, user, api_version='v1', verify_ssl=True, timeo
         self.verify_ssl = verify_ssl
         self.proxies = proxies
         self.timeout = timeout
+        
+        if host.split('/') > 3:
+            self.custom_path = '/' + '/'.join(host.split('/')[3:]).rstrip('/')
+        else:
+            self.custom_path = ''
 
         if not user_agent:
             self.user_agent = 'DefectDojo_api/' + version
@@ -46,46 +51,47 @@ def __init__(self, host, api_key, user, api_version='v1', verify_ssl=True, timeo
             requests.packages.urllib3.disable_warnings()  # Disabling SSL warning messages if verification is disabled.
 
     ###### Helper Functions ######
+
     def get_user_uri(self, user_id):
         """Returns the DefectDojo API URI for a user.
 
         :param user_id: Id of the user
 
         """
 
-        return "/api/" + self.api_version + "/users/" + str(user_id) + "/"
+        return self.custom_path + "/api/" + self.api_version + "/users/" + str(user_id) + "/"
 
     def get_engagement_uri(self, engagement_id):
         """Returns the DefectDojo API URI for an engagement.
 
         :param engagement_id: Id of the engagement
 
         """
-        return "/api/" + self.api_version + "/engagements/" + str(engagement_id) + "/"
+        return self.custom_path + "/api/" + self.api_version + "/engagements/" + str(engagement_id) + "/"
 
     def get_product_uri(self, product_id):
         """Returns the DefectDojo API URI for a product.
 
         :param product_id: Id of the product
 
         """
-        return "/api/" + self.api_version + "/products/" + str(product_id) + "/"
+        return self.custom_path + "/api/" + self.api_version + "/products/" + str(product_id) + "/"
 
     def get_test_uri(self, test_id):
         """Returns the DefectDojo API URI for a test.
 
         :param test_id: Id of the test
 
         """
-        return "/api/" + self.api_version + "/tests/" + str(test_id) + "/"
+        return self.custom_path + "/api/" + self.api_version + "/tests/" + str(test_id) + "/"
 
     def get_language_uri(self, language_type_id):
         """Returns the DefectDojo API URI for a langauge.
 
         :param test_id: Id of the language
 
         """
-        return "/api/" + self.api_version + "/language_types/" + str(language_type_id) + "/"
+        return self.custom_path + "/api/" + self.api_version + "/language_types/" + str(language_type_id) + "/"
 
     def get_tool_configuration_uri(self, tool_configuration_id):
         """Returns the DefectDojo API URI for a tool.