@@ -14,10 +14,11 @@ class TelegraphApi:
1414 :type access_token: str
1515 """
1616
17- __slots__ = ('access_token' , 'session' )
17+ __slots__ = ('access_token' , 'base_url' , ' session' )
1818
19- def __init__ (self , access_token = None ):
19+ def __init__ (self , access_token = None , base_url = 'telegra.ph' ):
2020 self .access_token = access_token
21+ self .base_url = base_url
2122 self .session = requests .Session ()
2223
2324 def method (self , method , values = None , path = '' ):
@@ -27,7 +28,7 @@ def method(self, method, values=None, path=''):
2728 values ['access_token' ] = self .access_token
2829
2930 response = self .session .post (
30- 'https://api.telegra.ph /{}/{}' .format (method , path ),
31+ 'https://api.{} /{}/{}' .format (self . base_url , method , path ),
3132 data = values
3233 ).json ()
3334
@@ -51,7 +52,7 @@ def upload_file(self, f):
5152 """
5253 with FilesOpener (f ) as files :
5354 response = self .session .post (
54- 'https://telegra.ph /upload' ,
55+ 'https://{} /upload' . format ( self . base_url ) ,
5556 files = files
5657 ).json ()
5758
@@ -78,8 +79,8 @@ class Telegraph:
7879
7980 __slots__ = ('_telegraph' ,)
8081
81- def __init__ (self , access_token = None ):
82- self ._telegraph = TelegraphApi (access_token )
82+ def __init__ (self , access_token = None , base_url = 'telegra.ph' ):
83+ self ._telegraph = TelegraphApi (access_token , base_url )
8384
8485 def get_access_token (self ):
8586 """Get current access_token"""
0 commit comments