1
0
mirror of https://github.com/EDCD/EDMarketConnector.git synced 2025-06-02 08:31:16 +03:00

Set connection timeout on Companion and EDDN network interactions.

This commit is contained in:
Jonathan Harris 2015-06-27 11:39:20 +01:00
parent 1110d978f4
commit 3cc6a7f54f
2 changed files with 7 additions and 5 deletions

View File

@ -19,6 +19,7 @@ if __debug__:
from config import config
holdoff = 120 # be nice
timeout = 10 # requests timeout
# Map values reported by the Companion interface to names displayed in-game and recognized by trade tools
@ -112,7 +113,7 @@ class Session:
raise CredentialsError()
else:
self.credentials = { 'email' : username, 'password' : password }
r = self.session.post('https://companion.orerve.net/user/login', data = self.credentials)
r = self.session.post('https://companion.orerve.net/user/login', data = self.credentials, timeout=timeout)
if r.status_code != requests.codes.ok:
self.dump(r)
r.raise_for_status()
@ -131,8 +132,7 @@ class Session:
return r.status_code
def verify(self, code):
r = self.session.post('https://companion.orerve.net/user/confirm',
data = { 'code' : code })
r = self.session.post('https://companion.orerve.net/user/confirm', data = {'code' : code}, timeout=timeout)
r.raise_for_status()
# verification doesn't actually return a yes/no, so log in again to determine state
try:
@ -147,7 +147,7 @@ class Session:
self.login()
elif self.state == Session.STATE_AUTH:
raise VerificationRequired()
r = self.session.get('https://companion.orerve.net/profile')
r = self.session.get('https://companion.orerve.net/profile', timeout=timeout)
if r.status_code != requests.codes.ok:
self.dump(r)

View File

@ -13,6 +13,8 @@ from config import applongname, appversion, config
upload = 'http://eddn-gateway.elite-markets.net:8080/upload/'
schema = 'http://schemas.elite-markets.net/eddn/commodity/2'
timeout= 10 # requests timeout
bracketmap = { 1: 'Low',
2: 'Med',
3: 'High', }
@ -20,7 +22,7 @@ bracketmap = { 1: 'Low',
def export(data):
def send(msg):
r = requests.post(upload, data=json.dumps(msg))
r = requests.post(upload, data=json.dumps(msg), timeout=timeout)
if __debug__ and r.status_code != requests.codes.ok:
print 'Status\t%s' % r.status_code
print 'URL\t%s' % r.url