simono41/pyicloud
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix response is not defined (#233)

Browse source
This commit is contained in:
Quentame 2020-03-04 00:03:14 +01:00 committed by GitHub
parent 5368081b6b
commit 5efcbc2d2b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pyicloud/base.py
View file

@ -114,7 +114,7 @@ class PyiCloudSession(requests.Session):
def _raise_error(self, code, reason):
if self.service.requires_2sa and \
reason == 'Missing X-APPLE-WEBAUTH-TOKEN cookie':
raise PyiCloud2SARequiredError(response.url)
raise PyiCloud2SARequiredError(self.service.user['apple_id'])
if code == 'ZONE_NOT_FOUND' or code == 'AUTHENTICATION_FAILED':
reason = 'Please log into https://icloud.com/ to manually ' \
'finish setting up your iCloud service'

4
pyicloud/exceptions.py
View file

@ -25,8 +25,8 @@ class PyiCloudFailedLoginException(PyiCloudException):
class PyiCloud2SARequiredError(PyiCloudException):
def __init__(self, url):
message = "Two-step authentication required for %s" % url
def __init__(self, apple_id):
message = "Two-step authentication required for account: %s" % apple_id
super(PyiCloud2SARequiredError, self).__init__(message)