make the exchange manager use treq instead of requests

This commit is contained in:
Victor Shyba 2018-05-05 02:33:37 -03:00
parent 8bfc35753a
commit c90140f701

View file

@ -1,8 +1,9 @@
import time import time
import requests
import logging import logging
import json import json
from twisted.internet import defer, threads
import treq
from twisted.internet import defer
from twisted.internet.task import LoopingCall from twisted.internet.task import LoopingCall
from lbrynet.core.Error import InvalidExchangeRateResponse from lbrynet.core.Error import InvalidExchangeRateResponse
@ -52,9 +53,10 @@ class MarketFeed(object):
def is_online(self): def is_online(self):
return self._online return self._online
@defer.inlineCallbacks
def _make_request(self): def _make_request(self):
r = requests.get(self.url, self.params, timeout=self.REQUESTS_TIMEOUT) response = yield treq.get(self.url, params=self.params, timeout=self.REQUESTS_TIMEOUT)
return r.text defer.returnValue((yield response.content()))
def _handle_response(self, response): def _handle_response(self, response):
return NotImplementedError return NotImplementedError
@ -75,7 +77,7 @@ class MarketFeed(object):
self._online = False self._online = False
def _update_price(self): def _update_price(self):
d = threads.deferToThread(self._make_request) d = self._make_request()
d.addCallback(self._handle_response) d.addCallback(self._handle_response)
d.addCallback(self._subtract_fee) d.addCallback(self._subtract_fee)
d.addCallback(self._save_price) d.addCallback(self._save_price)