update constructor

This commit is contained in:
Joel Gil Leon 2018-05-19 15:40:38 +03:00
pare c5b77beb04
commit 3d883a74b6

Veure arxiu

@ -174,7 +174,7 @@ class Mastodon:
###
# Authentication, including constructor
###
def __init__(self, client_id, client_secret=None, access_token=None,
def __init__(self, client_id=None, client_secret=None, access_token=None,
api_base_url=__DEFAULT_BASE_URL, debug_requests=False,
ratelimit_method="wait", ratelimit_pacefactor=1.1,
request_timeout=__DEFAULT_TIMEOUT, mastodon_version=None,
@ -248,6 +248,9 @@ class Mastodon:
raise MastodonIllegalArgumentError("Invalid ratelimit method.")
# Token loading
if self.client_id is None and access_token is None:
raise MastodonIllegalArgumentError('Credentials are needed')
if self.client_id is not None:
if os.path.isfile(self.client_id):
with open(self.client_id, 'r') as secret_file:
self.client_id = secret_file.readline().rstrip()
@ -255,7 +258,6 @@ class Mastodon:
else:
if self.client_secret is None:
raise MastodonIllegalArgumentError('Specified client id directly, but did not supply secret')
if self.access_token is not None and os.path.isfile(self.access_token):
with open(self.access_token, 'r') as token_file:
self.access_token = token_file.readline().rstrip()