Hopiu/portainer-cli
1
0
Fork 0
mirror of https://github.com/Hopiu/portainer-cli.git synced 2026-03-16 22:10:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix update_stack request url mount, add env-file param in update_stack, fix clear-env in update_stack

Browse source
This commit is contained in:
Douglas Paz 2018-12-03 13:32:00 -03:00
parent 5dae82b37d
commit ddcc8e7887
1 changed files with 43 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
portainer_cli/__init__.py
View file

@ -22,10 +22,7 @@ env_arg_regex = r'--env\.(.+)=(.+)'
def env_arg_to_dict(s):
split = re.split(env_arg_regex, s)
return {
'name': split[1],
'value': split[2],
}
return (split[1], split[2],)
class PortainerCLI(object):
@ -119,11 +116,12 @@ class PortainerCLI(object):
self.jwt = jwt
@plac.annotations(
env_file=('Environment Variable file', 'option'),
prune=('Prune services', 'flag', 'p'),
clear_env=('Clear all env vars', 'flag', 'c'),
)
def update_stack(self, id, endpoint_id, stack_file='', prune=False,
clear_env=False, *args):
def update_stack(self, id, endpoint_id, stack_file='', env_file='',
prune=False, clear_env=False, *args):
stack_url = 'stacks/{}?endpointId={}'.format(
id,
endpoint_id,
@ -134,24 +132,53 @@ class PortainerCLI(object):
stack_file_content = open(stack_file).read()
else:
stack_file_content = self.request(
'stacks/{}/file?endpointId={}').format(
'stacks/{}/file?endpointId={}'.format(
id,
endpoint_id,
).json().get('StackFileContent')
env_args = filter(
lambda x: re.match(env_arg_regex, x),
args,
)
).json().get('StackFileContent')
if env_file:
env = {}
for env_line in open(env_file).readlines():
env_line = env_line.strip()
if not env_line \
or env_line.startswith('#') \
or '=' not in env_line:
continue
k, v = env_line.split('=', 1)
k, v = k.strip(), v.strip()
env[k] = v
else:
env_args = filter(
lambda x: re.match(env_arg_regex, x),
args,
)
env = dict(map(
lambda x: env_arg_to_dict(x),
env_args,
))
if not clear_env:
current_env = dict(
map(
lambda x: (x.get('name'), x.get('value'),),
current.get('Env'),
),
)
current_env.update(env)
env = current_env
final_env = list(
map(
lambda x: {'name': x[0], 'value': x[1]},
env.items()
),
)
env = list(map(
lambda x: env_arg_to_dict(x),
env_args,
))
data = {
'Id': id,
'StackFileContent': stack_file_content,
'Prune': prune,
'Env': env if len(env) > 0 or clear_env else current.get('Env'),
'Env': final_env if len(final_env) > 0 else current.get('Env'),
}
logger.debug('update stack data: {}'.format(data))
self.request(
stack_url,
self.METHOD_PUT,