Hopiu/portainer-cli
1
0
Fork 0
mirror of https://github.com/Hopiu/portainer-cli.git synced 2026-03-17 06:20:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add proxies property based on environment variables for HTTP requests

Browse source
This commit is contained in:
Bastien Bonnefoy 2019-03-13 18:07:14 +00:00
parent bb34e5ef5c
commit 4641728b17
1 changed files with 20 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
portainer_cli/__init__.py
View file

@ -46,6 +46,7 @@ class PortainerCLI(object):
local = False
_base_url = 'http://localhost:9000/'
_jwt = None
_proxies = {}
@property
def base_url(self):
@ -67,6 +68,23 @@ class PortainerCLI(object):
self._jwt = value
self.persist()
@property
def proxies(self):
return self._proxies
@proxies.setter
def proxies(self, value):
try:
self._proxies['http'] = os.environ['HTTP_PROXY']
except KeyError:
self._proxies['http'] = ''
try:
self._proxies['https'] = os.environ['HTTPS_PROXY']
except KeyError:
self._proxies['https'] = ''
if self._proxies['http'] == '' and self._proxies['https'] == '':
self._proxies = {}
@property
def data_path(self):
if self.local:
@ -231,7 +249,7 @@ class PortainerCLI(object):
prepped.headers['Content-Length'] = len(prepped.body)
if self.jwt:
prepped.headers['Authorization'] = 'Bearer {}'.format(self.jwt)
response = session.send(prepped)
response = session.send(prepped, proxies=self.proxies)
logger.debug('request response: {}'.format(response.content))
response.raise_for_status()
if printc:
@ -254,6 +272,7 @@ class PortainerCLI(object):
logging.basicConfig(level=logging.DEBUG)
self.local = local
self.load()
self.proxies = {}
if command == self.COMMAND_CONFIGURE:
plac.call(self.configure, args)
elif command == self.COMMAND_LOGIN: