Skip to content

Commit e5af447

Browse files
committed
Change __init__/connect scheme with create GitHub and upgrade to GHE model.
1 parent e48a3b7 commit e5af447

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

git_repo/services/ext/github.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,15 +13,24 @@
1313

1414
from datetime import datetime
1515

16+
GITHUB_COM_FQDN = 'github.com'
17+
1618
@register_target('hub', 'github')
1719
class GithubService(RepositoryService):
18-
fqdn = 'github.com'
20+
fqdn = GITHUB_COM_FQDN
1921

2022
def __init__(self, *args, **kwarg):
2123
self.gh = github3.GitHub()
2224
super(GithubService, self).__init__(*args, **kwarg)
2325

2426
def connect(self):
27+
if self.fqdn != GITHUB_COM_FQDN:
28+
# upgrade self.gh from a GitHub object to a GitHubEnterprise object
29+
gh = github3.GitHubEnterprise(RepositoryService.build_url(self))
30+
self.gh._session.base_url = gh._session.base_url
31+
self.gh = gh
32+
# propagate ssl certificate parameter
33+
self.gh._session.verify = self.session_certificate or not self.session_insecure
2534
try:
2635
self.gh.login(token=self._privatekey)
2736
self.username = self.gh.user().login

0 commit comments

Comments
 (0)