diff options
author | zdc <zdc@users.noreply.github.com> | 2022-03-26 15:41:59 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-03-26 15:41:59 +0200 |
commit | aa60d48c2711cdcd9f88a4e5c77379adb0408231 (patch) | |
tree | 349631a02467dae0158f6f663cc8aa8537974a97 /cloudinit/dhclient_hook.py | |
parent | 5c4b3943343a85fbe517e5ec1fc670b3a8566b4b (diff) | |
parent | 31448cccedd8f841fb3ac7d0f2e3cdefe08a53ba (diff) | |
download | vyos-cloud-init-aa60d48c2711cdcd9f88a4e5c77379adb0408231.tar.gz vyos-cloud-init-aa60d48c2711cdcd9f88a4e5c77379adb0408231.zip |
Merge pull request #51 from zdc/T2117-sagitta-22.1
T2117: Cloud-init updated to 22.1
Diffstat (limited to 'cloudinit/dhclient_hook.py')
-rw-r--r-- | cloudinit/dhclient_hook.py | 21 |
1 files changed, 13 insertions, 8 deletions
diff --git a/cloudinit/dhclient_hook.py b/cloudinit/dhclient_hook.py index 72b51b6a..46b2e8d9 100644 --- a/cloudinit/dhclient_hook.py +++ b/cloudinit/dhclient_hook.py @@ -19,7 +19,7 @@ EVENTS = (UP, DOWN) def _get_hooks_dir(): i = stages.Init() - return os.path.join(i.paths.get_runpath(), 'dhclient.hooks') + return os.path.join(i.paths.get_runpath(), "dhclient.hooks") def _filter_env_vals(info): @@ -28,15 +28,16 @@ def _filter_env_vals(info): new_info = {} for k, v in info.items(): if k.startswith("DHCP4_") or k.startswith("new_"): - key = (k.replace('DHCP4_', '').replace('new_', '')).lower() + key = (k.replace("DHCP4_", "").replace("new_", "")).lower() new_info[key] = v return new_info def run_hook(interface, event, data_d=None, env=None): if event not in EVENTS: - raise ValueError("Unexpected event '%s'. Expected one of: %s" % - (event, EVENTS)) + raise ValueError( + "Unexpected event '%s'. Expected one of: %s" % (event, EVENTS) + ) if data_d is None: data_d = _get_hooks_dir() if env is None: @@ -58,9 +59,11 @@ def get_parser(parser=None): if parser is None: parser = argparse.ArgumentParser(prog=NAME, description=__doc__) parser.add_argument( - "event", help='event taken on the interface', choices=EVENTS) + "event", help="event taken on the interface", choices=EVENTS + ) parser.add_argument( - "interface", help='the network interface being acted upon') + "interface", help="the network interface being acted upon" + ) # cloud-init main uses 'action' parser.set_defaults(action=(NAME, handle_args)) return parser @@ -72,12 +75,14 @@ def handle_args(name, args, data_d=None): return run_hook(interface=args.interface, event=args.event, data_d=data_d) -if __name__ == '__main__': +if __name__ == "__main__": import sys + parser = get_parser() args = parser.parse_args(args=sys.argv[1:]) return_value = handle_args( - NAME, args, data_d=os.environ.get('_CI_DHCP_HOOK_DATA_D')) + NAME, args, data_d=os.environ.get("_CI_DHCP_HOOK_DATA_D") + ) if return_value: sys.exit(return_value) |