1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2026-05-06 10:05:54 +00:00

Reformat everything.

This commit is contained in:
Felix Fontein 2025-11-01 12:08:41 +01:00
parent 3f2213791a
commit 340ff8586d
1008 changed files with 61301 additions and 58309 deletions

View file

@ -76,29 +76,36 @@ from ansible.module_utils.common.text.converters import to_native
class NotSupportedError(Exception):
pass
# module specific functions
def get_schema_facts(cursor, schema=''):
def get_schema_facts(cursor, schema=""):
facts = {}
cursor.execute("""
cursor.execute(
"""
select schema_name, schema_owner, create_time
from schemata
where not is_system_schema and schema_name not in ('public')
and (? = '' or schema_name ilike ?)
""", schema, schema)
""",
schema,
schema,
)
while True:
rows = cursor.fetchmany(100)
if not rows:
break
for row in rows:
facts[row.schema_name.lower()] = {
'name': row.schema_name,
'owner': row.schema_owner,
'create_time': str(row.create_time),
'usage_roles': [],
'create_roles': []}
cursor.execute("""
"name": row.schema_name,
"owner": row.schema_owner,
"create_time": str(row.create_time),
"usage_roles": [],
"create_roles": [],
}
cursor.execute(
"""
select g.object_name as schema_name, r.name as role_name,
lower(g.privileges_description) privileges_description
from roles r join grants g
@ -106,23 +113,27 @@ def get_schema_facts(cursor, schema=''):
and g.privileges_description like '%USAGE%'
and g.grantee not in ('public', 'dbadmin')
and (? = '' or g.object_name ilike ?)
""", schema, schema)
""",
schema,
schema,
)
while True:
rows = cursor.fetchmany(100)
if not rows:
break
for row in rows:
schema_key = row.schema_name.lower()
if 'create' in row.privileges_description:
facts[schema_key]['create_roles'].append(row.role_name)
if "create" in row.privileges_description:
facts[schema_key]["create_roles"].append(row.role_name)
else:
facts[schema_key]['usage_roles'].append(row.role_name)
facts[schema_key]["usage_roles"].append(row.role_name)
return facts
def get_user_facts(cursor, user=''):
def get_user_facts(cursor, user=""):
facts = {}
cursor.execute("""
cursor.execute(
"""
select u.user_name, u.is_locked, u.lock_time,
p.password, p.acctexpired as is_expired,
u.profile_name, u.resource_pool,
@ -130,7 +141,10 @@ def get_user_facts(cursor, user=''):
from users u join password_auditor p on p.user_id = u.user_id
where not u.is_super_user
and (? = '' or u.user_name ilike ?)
""", user, user)
""",
user,
user,
)
while True:
rows = cursor.fetchmany(100)
if not rows:
@ -138,65 +152,73 @@ def get_user_facts(cursor, user=''):
for row in rows:
user_key = row.user_name.lower()
facts[user_key] = {
'name': row.user_name,
'locked': str(row.is_locked),
'password': row.password,
'expired': str(row.is_expired),
'profile': row.profile_name,
'resource_pool': row.resource_pool,
'roles': [],
'default_roles': []}
"name": row.user_name,
"locked": str(row.is_locked),
"password": row.password,
"expired": str(row.is_expired),
"profile": row.profile_name,
"resource_pool": row.resource_pool,
"roles": [],
"default_roles": [],
}
if row.is_locked:
facts[user_key]['locked_time'] = str(row.lock_time)
facts[user_key]["locked_time"] = str(row.lock_time)
if row.all_roles:
facts[user_key]['roles'] = row.all_roles.replace(' ', '').split(',')
facts[user_key]["roles"] = row.all_roles.replace(" ", "").split(",")
if row.default_roles:
facts[user_key]['default_roles'] = row.default_roles.replace(' ', '').split(',')
facts[user_key]["default_roles"] = row.default_roles.replace(" ", "").split(",")
return facts
def get_role_facts(cursor, role=''):
def get_role_facts(cursor, role=""):
facts = {}
cursor.execute("""
cursor.execute(
"""
select r.name, r.assigned_roles
from roles r
where (? = '' or r.name ilike ?)
""", role, role)
""",
role,
role,
)
while True:
rows = cursor.fetchmany(100)
if not rows:
break
for row in rows:
role_key = row.name.lower()
facts[role_key] = {
'name': row.name,
'assigned_roles': []}
facts[role_key] = {"name": row.name, "assigned_roles": []}
if row.assigned_roles:
facts[role_key]['assigned_roles'] = row.assigned_roles.replace(' ', '').split(',')
facts[role_key]["assigned_roles"] = row.assigned_roles.replace(" ", "").split(",")
return facts
def get_configuration_facts(cursor, parameter=''):
def get_configuration_facts(cursor, parameter=""):
facts = {}
cursor.execute("""
cursor.execute(
"""
select c.parameter_name, c.current_value, c.default_value
from configuration_parameters c
where c.node_name = 'ALL'
and (? = '' or c.parameter_name ilike ?)
""", parameter, parameter)
""",
parameter,
parameter,
)
while True:
rows = cursor.fetchmany(100)
if not rows:
break
for row in rows:
facts[row.parameter_name.lower()] = {
'parameter_name': row.parameter_name,
'current_value': row.current_value,
'default_value': row.default_value}
"parameter_name": row.parameter_name,
"current_value": row.current_value,
"default_value": row.default_value,
}
return facts
def get_node_facts(cursor, schema=''):
def get_node_facts(cursor, schema=""):
facts = {}
cursor.execute("""
select node_name, node_address, export_address, node_state, node_type,
@ -209,33 +231,36 @@ def get_node_facts(cursor, schema=''):
break
for row in rows:
facts[row.node_address] = {
'node_name': row.node_name,
'export_address': row.export_address,
'node_state': row.node_state,
'node_type': row.node_type,
'catalog_path': row.catalog_path}
"node_name": row.node_name,
"export_address": row.export_address,
"node_state": row.node_state,
"node_type": row.node_type,
"catalog_path": row.catalog_path,
}
return facts
# module logic
def main():
module = AnsibleModule(
argument_spec=dict(
cluster=dict(default='localhost'),
port=dict(default='5433'),
cluster=dict(default="localhost"),
port=dict(default="5433"),
db=dict(),
login_user=dict(default='dbadmin'),
login_user=dict(default="dbadmin"),
login_password=dict(no_log=True),
), supports_check_mode=True)
),
supports_check_mode=True,
)
if not pyodbc_found:
module.fail_json(msg=missing_required_lib('pyodbc'), exception=PYODBC_IMP_ERR)
module.fail_json(msg=missing_required_lib("pyodbc"), exception=PYODBC_IMP_ERR)
db = ''
if module.params['db']:
db = module.params['db']
db = ""
if module.params["db"]:
db = module.params["db"]
try:
dsn = (
@ -259,12 +284,14 @@ def main():
configuration_facts = get_configuration_facts(cursor)
node_facts = get_node_facts(cursor)
module.exit_json(changed=False,
vertica_schemas=schema_facts,
vertica_users=user_facts,
vertica_roles=role_facts,
vertica_configuration=configuration_facts,
vertica_nodes=node_facts)
module.exit_json(
changed=False,
vertica_schemas=schema_facts,
vertica_users=user_facts,
vertica_roles=role_facts,
vertica_configuration=configuration_facts,
vertica_nodes=node_facts,
)
except NotSupportedError as e:
module.fail_json(msg=to_native(e), exception=traceback.format_exc())
except SystemExit:
@ -274,5 +301,5 @@ def main():
module.fail_json(msg=to_native(e), exception=traceback.format_exc())
if __name__ == '__main__':
if __name__ == "__main__":
main()