File tree Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Original file line number Diff line number Diff line change 13
13
14
14
from datetime import datetime
15
15
16
+ GITHUB_COM_FQDN = 'github.com'
17
+
16
18
@register_target ('hub' , 'github' )
17
19
class GithubService (RepositoryService ):
18
- fqdn = 'github.com'
20
+ fqdn = GITHUB_COM_FQDN
19
21
20
22
def __init__ (self , * args , ** kwarg ):
21
23
self .gh = github3 .GitHub ()
22
24
super (GithubService , self ).__init__ (* args , ** kwarg )
23
25
24
26
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
25
34
try :
26
35
self .gh .login (token = self ._privatekey )
27
36
self .username = self .gh .user ().login
You can’t perform that action at this time.
0 commit comments