mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-04-02 18:36:19 +00:00
Reformat everything.
This commit is contained in:
Felix Fontein
parent
3f2213791a
commit
340ff8586d
1008 changed files with 61301 additions and 58309 deletions
|
|
@ -138,6 +138,7 @@ try:
|
|||
from pypureomapi import Omapi, OmapiMessage, OmapiError, OmapiErrorNotFound
|
||||
from pypureomapi import pack_ip, unpack_ip, pack_mac, unpack_mac
|
||||
from pypureomapi import OMAPI_OP_STATUS, OMAPI_OP_UPDATE
|
||||
|
||||
pureomapi_found = True
|
||||
except ImportError:
|
||||
PUREOMAPI_IMP_ERR = traceback.format_exc()
|
||||
|
|
@ -155,19 +156,25 @@ class OmapiHostManager:
|
|||
|
||||
def connect(self):
|
||||
try:
|
||||
self.omapi = Omapi(self.module.params['host'], self.module.params['port'], to_bytes(self.module.params['key_name']),
|
||||
self.module.params['key'])
|
||||
self.omapi = Omapi(
|
||||
self.module.params["host"],
|
||||
self.module.params["port"],
|
||||
to_bytes(self.module.params["key_name"]),
|
||||
self.module.params["key"],
|
||||
)
|
||||
except binascii.Error:
|
||||
self.module.fail_json(msg="Unable to open OMAPI connection. 'key' is not a valid base64 key.")
|
||||
except OmapiError as e:
|
||||
self.module.fail_json(msg=f"Unable to open OMAPI connection. Ensure 'host', 'port', 'key' and 'key_name' are valid. Exception was: {e}")
|
||||
self.module.fail_json(
|
||||
msg=f"Unable to open OMAPI connection. Ensure 'host', 'port', 'key' and 'key_name' are valid. Exception was: {e}"
|
||||
)
|
||||
except socket.error as e:
|
||||
self.module.fail_json(msg=f"Unable to connect to OMAPI server: {e}")
|
||||
|
||||
def get_host(self, macaddr):
|
||||
msg = OmapiMessage.open(to_bytes("host", errors='surrogate_or_strict'))
|
||||
msg.obj.append((to_bytes("hardware-address", errors='surrogate_or_strict'), pack_mac(macaddr)))
|
||||
msg.obj.append((to_bytes("hardware-type", errors='surrogate_or_strict'), struct.pack("!I", 1)))
|
||||
msg = OmapiMessage.open(to_bytes("host", errors="surrogate_or_strict"))
|
||||
msg.obj.append((to_bytes("hardware-address", errors="surrogate_or_strict"), pack_mac(macaddr)))
|
||||
msg.obj.append((to_bytes("hardware-type", errors="surrogate_or_strict"), struct.pack("!I", 1)))
|
||||
response = self.omapi.query_server(msg)
|
||||
if response.opcode != OMAPI_OP_UPDATE:
|
||||
return None
|
||||
|
|
@ -179,53 +186,56 @@ class OmapiHostManager:
|
|||
for k, v in dict(obj).items():
|
||||
result[to_text(k)] = v
|
||||
|
||||
if 'hardware-address' in result:
|
||||
result['hardware-address'] = to_native(unpack_mac(result['hardware-address']))
|
||||
if "hardware-address" in result:
|
||||
result["hardware-address"] = to_native(unpack_mac(result["hardware-address"]))
|
||||
|
||||
if 'ip-address' in result:
|
||||
result['ip-address'] = to_native(unpack_ip(result['ip-address']))
|
||||
if "ip-address" in result:
|
||||
result["ip-address"] = to_native(unpack_ip(result["ip-address"]))
|
||||
|
||||
if 'hardware-type' in result:
|
||||
result['hardware-type'] = struct.unpack("!I", result['hardware-type'])
|
||||
if "hardware-type" in result:
|
||||
result["hardware-type"] = struct.unpack("!I", result["hardware-type"])
|
||||
|
||||
return result
|
||||
|
||||
def setup_host(self):
|
||||
if self.module.params['hostname'] is None or len(self.module.params['hostname']) == 0:
|
||||
if self.module.params["hostname"] is None or len(self.module.params["hostname"]) == 0:
|
||||
self.module.fail_json(msg="name attribute could not be empty when adding or modifying host.")
|
||||
|
||||
msg = None
|
||||
host_response = self.get_host(self.module.params['macaddr'])
|
||||
host_response = self.get_host(self.module.params["macaddr"])
|
||||
# If host was not found using macaddr, add create message
|
||||
if host_response is None:
|
||||
msg = OmapiMessage.open(to_bytes('host', errors='surrogate_or_strict'))
|
||||
msg.message.append((to_bytes('create'), struct.pack('!I', 1)))
|
||||
msg.message.append((to_bytes('exclusive'), struct.pack('!I', 1)))
|
||||
msg.obj.append((to_bytes('hardware-address'), pack_mac(self.module.params['macaddr'])))
|
||||
msg.obj.append((to_bytes('hardware-type'), struct.pack('!I', 1)))
|
||||
msg.obj.append((to_bytes('name'), to_bytes(self.module.params['hostname'])))
|
||||
if self.module.params['ip'] is not None:
|
||||
msg.obj.append((to_bytes("ip-address", errors='surrogate_or_strict'), pack_ip(self.module.params['ip'])))
|
||||
msg = OmapiMessage.open(to_bytes("host", errors="surrogate_or_strict"))
|
||||
msg.message.append((to_bytes("create"), struct.pack("!I", 1)))
|
||||
msg.message.append((to_bytes("exclusive"), struct.pack("!I", 1)))
|
||||
msg.obj.append((to_bytes("hardware-address"), pack_mac(self.module.params["macaddr"])))
|
||||
msg.obj.append((to_bytes("hardware-type"), struct.pack("!I", 1)))
|
||||
msg.obj.append((to_bytes("name"), to_bytes(self.module.params["hostname"])))
|
||||
if self.module.params["ip"] is not None:
|
||||
msg.obj.append(
|
||||
(to_bytes("ip-address", errors="surrogate_or_strict"), pack_ip(self.module.params["ip"]))
|
||||
)
|
||||
|
||||
stmt_join = ""
|
||||
if self.module.params['ddns']:
|
||||
stmt_join += f"ddns-hostname \"{self.module.params['hostname']}\"; "
|
||||
if self.module.params["ddns"]:
|
||||
stmt_join += f'ddns-hostname "{self.module.params["hostname"]}"; '
|
||||
|
||||
try:
|
||||
if len(self.module.params['statements']) > 0:
|
||||
stmt_join += "; ".join(self.module.params['statements'])
|
||||
if len(self.module.params["statements"]) > 0:
|
||||
stmt_join += "; ".join(self.module.params["statements"])
|
||||
stmt_join += "; "
|
||||
except TypeError as e:
|
||||
self.module.fail_json(msg=f"Invalid statements found: {e}")
|
||||
|
||||
if len(stmt_join) > 0:
|
||||
msg.obj.append((to_bytes('statements'), to_bytes(stmt_join)))
|
||||
msg.obj.append((to_bytes("statements"), to_bytes(stmt_join)))
|
||||
|
||||
try:
|
||||
response = self.omapi.query_server(msg)
|
||||
if response.opcode != OMAPI_OP_UPDATE:
|
||||
self.module.fail_json(msg="Failed to add host, ensure authentication and host parameters "
|
||||
"are valid.")
|
||||
self.module.fail_json(
|
||||
msg="Failed to add host, ensure authentication and host parameters are valid."
|
||||
)
|
||||
self.module.exit_json(changed=True, lease=self.unpack_facts(response.obj))
|
||||
except OmapiError as e:
|
||||
self.module.fail_json(msg=f"OMAPI error: {e}")
|
||||
|
|
@ -234,14 +244,17 @@ class OmapiHostManager:
|
|||
response_obj = self.unpack_facts(host_response.obj)
|
||||
fields_to_update = {}
|
||||
|
||||
if 'ip-address' not in response_obj or \
|
||||
response_obj['ip-address'] != self.module.params['ip']:
|
||||
fields_to_update['ip-address'] = pack_ip(self.module.params['ip'])
|
||||
if "ip-address" not in response_obj or response_obj["ip-address"] != self.module.params["ip"]:
|
||||
fields_to_update["ip-address"] = pack_ip(self.module.params["ip"])
|
||||
|
||||
# Name cannot be changed
|
||||
if 'name' not in response_obj or response_obj['name'] != self.module.params['hostname']:
|
||||
self.module.fail_json(msg=(f"Changing hostname is not supported. Old was {response_obj['name']}, "
|
||||
f"new is {self.module.params['hostname']}. Please delete host and add new."))
|
||||
if "name" not in response_obj or response_obj["name"] != self.module.params["hostname"]:
|
||||
self.module.fail_json(
|
||||
msg=(
|
||||
f"Changing hostname is not supported. Old was {response_obj['name']}, "
|
||||
f"new is {self.module.params['hostname']}. Please delete host and add new."
|
||||
)
|
||||
)
|
||||
|
||||
"""
|
||||
# It seems statements are not returned by OMAPI, then we cannot modify them at this moment.
|
||||
|
|
@ -260,15 +273,16 @@ class OmapiHostManager:
|
|||
try:
|
||||
response = self.omapi.query_server(msg)
|
||||
if response.opcode != OMAPI_OP_STATUS:
|
||||
self.module.fail_json(msg="Failed to modify host, ensure authentication and host parameters "
|
||||
"are valid.")
|
||||
self.module.fail_json(
|
||||
msg="Failed to modify host, ensure authentication and host parameters are valid."
|
||||
)
|
||||
self.module.exit_json(changed=True)
|
||||
except OmapiError as e:
|
||||
self.module.fail_json(msg=f"OMAPI error: {e}")
|
||||
|
||||
def remove_host(self):
|
||||
try:
|
||||
self.omapi.del_host(self.module.params['macaddr'])
|
||||
self.omapi.del_host(self.module.params["macaddr"])
|
||||
self.module.exit_json(changed=True)
|
||||
except OmapiErrorNotFound:
|
||||
self.module.exit_json()
|
||||
|
|
@ -279,38 +293,38 @@ class OmapiHostManager:
|
|||
def main():
|
||||
module = AnsibleModule(
|
||||
argument_spec=dict(
|
||||
state=dict(type='str', required=True, choices=['absent', 'present']),
|
||||
host=dict(type='str', default="localhost"),
|
||||
port=dict(type='int', default=7911),
|
||||
key_name=dict(type='str', required=True),
|
||||
key=dict(type='str', required=True, no_log=True),
|
||||
macaddr=dict(type='str', required=True),
|
||||
hostname=dict(type='str', aliases=['name']),
|
||||
ip=dict(type='str'),
|
||||
ddns=dict(type='bool', default=False),
|
||||
statements=dict(type='list', elements='str', default=[]),
|
||||
state=dict(type="str", required=True, choices=["absent", "present"]),
|
||||
host=dict(type="str", default="localhost"),
|
||||
port=dict(type="int", default=7911),
|
||||
key_name=dict(type="str", required=True),
|
||||
key=dict(type="str", required=True, no_log=True),
|
||||
macaddr=dict(type="str", required=True),
|
||||
hostname=dict(type="str", aliases=["name"]),
|
||||
ip=dict(type="str"),
|
||||
ddns=dict(type="bool", default=False),
|
||||
statements=dict(type="list", elements="str", default=[]),
|
||||
),
|
||||
supports_check_mode=False,
|
||||
)
|
||||
|
||||
if not pureomapi_found:
|
||||
module.fail_json(msg=missing_required_lib('pypureomapi'), exception=PUREOMAPI_IMP_ERR)
|
||||
module.fail_json(msg=missing_required_lib("pypureomapi"), exception=PUREOMAPI_IMP_ERR)
|
||||
|
||||
if module.params['key'] is None or len(module.params["key"]) == 0:
|
||||
if module.params["key"] is None or len(module.params["key"]) == 0:
|
||||
module.fail_json(msg="'key' parameter cannot be empty.")
|
||||
|
||||
if module.params['key_name'] is None or len(module.params["key_name"]) == 0:
|
||||
if module.params["key_name"] is None or len(module.params["key_name"]) == 0:
|
||||
module.fail_json(msg="'key_name' parameter cannot be empty.")
|
||||
|
||||
host_manager = OmapiHostManager(module)
|
||||
try:
|
||||
if module.params['state'] == 'present':
|
||||
if module.params["state"] == "present":
|
||||
host_manager.setup_host()
|
||||
elif module.params['state'] == 'absent':
|
||||
elif module.params["state"] == "absent":
|
||||
host_manager.remove_host()
|
||||
except ValueError as e:
|
||||
module.fail_json(msg=f"OMAPI input value error: {e}")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue