144 lines
4.7 KiB
Python
144 lines
4.7 KiB
Python
# =============================================================================
|
|
# Copyright (C) 2014 Ryan Holmes
|
|
#
|
|
# This file is part of pyfa.
|
|
#
|
|
# pyfa is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# pyfa is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with pyfa. If not, see <http://www.gnu.org/licenses/>.
|
|
# =============================================================================
|
|
|
|
|
|
import requests
|
|
import socket
|
|
from logbook import Logger
|
|
|
|
import config
|
|
from service.settings import NetworkSettings
|
|
|
|
pyfalog = Logger(__name__)
|
|
|
|
# network timeout, otherwise pyfa hangs for a long while if no internet connection
|
|
timeout = 3
|
|
socket.setdefaulttimeout(timeout)
|
|
|
|
|
|
class Error(Exception):
|
|
def __init__(self, msg=None):
|
|
self.message = msg
|
|
|
|
|
|
class RequestError(Exception):
|
|
pass
|
|
|
|
|
|
class AuthenticationError(Exception):
|
|
pass
|
|
|
|
|
|
class ServerError(Exception):
|
|
pass
|
|
|
|
|
|
class TimeoutError(Exception):
|
|
pass
|
|
|
|
|
|
class Network:
|
|
# Request constants - every request must supply this, as it is checked if
|
|
# enabled or not via settings
|
|
ENABLED = 1
|
|
EVE = 2 # Mostly API, but also covers CREST requests. update: might be useless these days, this Network class needs to be reviewed
|
|
PRICES = 4
|
|
UPDATE = 8
|
|
|
|
_instance = None
|
|
|
|
@classmethod
|
|
def getInstance(cls):
|
|
if cls._instance is None:
|
|
cls._instance = Network()
|
|
|
|
return cls._instance
|
|
|
|
def get(self, url, type, **kwargs):
|
|
self.__networkAccessCheck(type)
|
|
|
|
headers = self.__getHeaders()
|
|
proxies = self.__getProxies()
|
|
|
|
try:
|
|
resp = requests.get(url, headers=headers, proxies=proxies, **kwargs)
|
|
resp.raise_for_status()
|
|
return resp
|
|
except requests.exceptions.HTTPError as error:
|
|
pyfalog.warning('HTTPError:')
|
|
pyfalog.warning(error)
|
|
if error.response.status_code == 404:
|
|
raise RequestError()
|
|
elif error.response.status_code == 403:
|
|
raise AuthenticationError()
|
|
elif error.response.status_code >= 500:
|
|
raise ServerError()
|
|
raise Error(error)
|
|
except requests.exceptions.Timeout:
|
|
raise TimeoutError()
|
|
except (KeyboardInterrupt, SystemExit):
|
|
raise
|
|
except Exception as error:
|
|
raise Error(error)
|
|
|
|
def post(self, url, type, jsonData, **kwargs):
|
|
self.__networkAccessCheck(type)
|
|
|
|
headers = self.__getHeaders()
|
|
proxies = self.__getProxies()
|
|
|
|
try:
|
|
resp = requests.post(url, json=jsonData, headers=headers, proxies=proxies, **kwargs)
|
|
resp.raise_for_status()
|
|
return resp
|
|
except requests.exceptions.HTTPError as error:
|
|
pyfalog.warning('HTTPError:')
|
|
pyfalog.warning(error)
|
|
if error.response.status_code == 404:
|
|
raise RequestError()
|
|
elif error.response.status_code == 403:
|
|
raise AuthenticationError()
|
|
elif error.response.status_code >= 500:
|
|
raise ServerError()
|
|
raise Error(error)
|
|
except requests.exceptions.Timeout:
|
|
raise TimeoutError()
|
|
except (KeyboardInterrupt, SystemExit):
|
|
raise
|
|
except Exception as error:
|
|
raise Error(error)
|
|
|
|
def __networkAccessCheck(self, type):
|
|
# Make sure request is enabled
|
|
access = NetworkSettings.getInstance().getAccess()
|
|
if not self.ENABLED & access or not type & access:
|
|
pyfalog.warning('Access not enabled - please enable in Preferences > Network')
|
|
raise Error('Access not enabled - please enable in Preferences > Network')
|
|
|
|
def __getHeaders(self):
|
|
versionString = '{0}'.format(config.version)
|
|
return {'User-Agent': 'pyfa {0} (python-requests {1})'.format(versionString, requests.__version__)}
|
|
|
|
def __getProxies(self):
|
|
# python-requests supports setting proxy for request as parameter to get() / post()
|
|
# in a form like: proxies = { 'http': 'http://10.10.1.10:3128', 'https': 'http://10.10.1.10:1080' }
|
|
# or with HTTP Basic auth support: proxies = {'http': 'http://user:pass@10.10.1.10:3128/'}
|
|
# then you do: requests.get('http://example.org', proxies=proxies)
|
|
return NetworkSettings.getInstance().getProxySettingsInRequestsFormat()
|