ED/EDMarketConnector
1
0
Fork 0
mirror of https://github.com/EDCD/EDMarketConnector.git synced 2025-04-13 07:47:14 +03:00
Code Issues Projects Releases Wiki Activity

Correctly report when Companion server is down

Browse Source
This commit is contained in:
Jonathan Harris 2016-10-25 10:49:34 +01:00
parent d5e327e8e2
commit 23fcf5d950
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
companion.py
View File

@ -208,11 +208,7 @@ class Session:
if __debug__: print_exc()
raise ServerError()
if r.status_code != requests.codes.ok:
self.dump(r)
r.raise_for_status()
if 'server error' in r.text:
if r.status_code != requests.codes.ok or 'server error' in r.text:
self.dump(r)
raise ServerError()
elif r.url == URL_LOGIN: # would have redirected away if success
@ -229,8 +225,7 @@ class Session:
if not code:
raise VerificationRequired()
r = self.session.post(URL_CONFIRM, data = {'code' : code}, timeout=timeout)
r.raise_for_status()
if r.url == URL_CONFIRM: # would have redirected away if success
if r.status_code != requests.codes.ok or r.url == URL_CONFIRM: # would have redirected away if success
raise VerificationRequired()
self.save() # Save cookies now for use by command-line app
self.login()
@ -248,14 +243,15 @@ class Session:
if __debug__: print_exc()
raise ServerError()
if r.status_code != requests.codes.ok:
self.dump(r)
if r.status_code == requests.codes.forbidden or r.url == URL_LOGIN:
# Start again - maybe our session cookie expired?
self.state = Session.STATE_INIT
return self.query()
r.raise_for_status()
if r.status_code != requests.codes.ok:
self.dump(r)
raise ServerError()
try:
data = r.json()
except: