Fix build system to work with Github Enterprise

endpoints/trigger.py

@@ -100,7 +100,7 @@ class BuildTrigger(object):
     raise NotImplementedError
 
   def activate(self, trigger_uuid, standard_webhook_url, auth_token, config):
-    """ 
+    """
     Activates the trigger for the service, with the given new configuration.
     Returns new configuration that should be stored if successful.
     """
@@ -150,8 +150,10 @@ def raise_unsupported():
 class GithubBuildTrigger(BuildTrigger):
   @staticmethod
   def _get_client(auth_token):
-    return Github(auth_token, client_id=github_trigger.client_id(),
-                              client_secret=github_trigger.client_secret())
+    return Github(auth_token,
+                  base_url=github_trigger.api_endpoint(),
+                  client_id=github_trigger.client_id(),
+                  client_secret=github_trigger.client_secret())
 
   @classmethod
   def service_name(cls):
@@ -246,7 +248,7 @@ class GithubBuildTrigger(BuildTrigger):
     gh_client = self._get_client(auth_token)
     source = config['build_source']
 
-    try:      
+    try:
       repo = gh_client.get_repo(source)
 
       # Find the first matching branch.
@@ -381,7 +383,7 @@ class GithubBuildTrigger(BuildTrigger):
         raise SkipRequestException()
 
     if should_skip_commit(commit_message):
-      raise SkipRequestException() 
+      raise SkipRequestException()
 
     short_sha = GithubBuildTrigger.get_display_name(commit_sha)
 
@@ -419,7 +421,7 @@ class GithubBuildTrigger(BuildTrigger):
       branches = self.list_field_values(auth_token, config, 'branch_name')
       tags = self.list_field_values(auth_token, config, 'tag_name')
 
-      return ([{'kind': 'branch', 'name': b} for b in branches] + 
+      return ([{'kind': 'branch', 'name': b} for b in branches] +
               [{'kind': 'tag', 'name': tag} for tag in tags])
 
     if field_name == 'tag_name':