Get, set, or list environment variables. Usage:\\ %env: lists all environment variables/values %env var: get value for var %env var val: set value for var %env var=val: set value for var %env var=$val: set value for var, using python expans
(self, parameter_s='')
| 432 | |
| 433 | @line_magic |
| 434 | def env(self, parameter_s=''): |
| 435 | """Get, set, or list environment variables. |
| 436 | |
| 437 | Usage:\\ |
| 438 | |
| 439 | %env: lists all environment variables/values |
| 440 | %env var: get value for var |
| 441 | %env var val: set value for var |
| 442 | %env var=val: set value for var |
| 443 | %env var=$val: set value for var, using python expansion if possible |
| 444 | """ |
| 445 | if parameter_s.strip(): |
| 446 | split = '=' if '=' in parameter_s else ' ' |
| 447 | bits = parameter_s.split(split) |
| 448 | if len(bits) == 1: |
| 449 | key = parameter_s.strip() |
| 450 | if key in os.environ: |
| 451 | return os.environ[key] |
| 452 | else: |
| 453 | err = "Environment does not have key: {0}".format(key) |
| 454 | raise UsageError(err) |
| 455 | if len(bits) > 1: |
| 456 | return self.set_env(parameter_s) |
| 457 | env = dict(os.environ) |
| 458 | # hide likely secrets when printing the whole environment |
| 459 | for key in list(env): |
| 460 | if any(s in key.lower() for s in ('key', 'token', 'secret')): |
| 461 | env[key] = '<hidden>' |
| 462 | |
| 463 | return env |
| 464 | |
| 465 | @line_magic |
| 466 | def set_env(self, parameter_s): |
nothing calls this directly
no test coverage detected