Skip to content

Commit

Permalink
style: formatting black
Browse files Browse the repository at this point in the history
  • Loading branch information
@Nemental
Nemental committed Jan 4, 2024
1 parent 45ddbd5 commit 681b266
Showing 1 changed file with 130 additions and 73 deletions.
203 changes: 130 additions & 73 deletions plugins/modules/grafana_datasource.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -617,7 +617,6 @@ def get_datasource_payload(data):

# datasource type related parameters
if data["ds_type"] == "elasticsearch":

json_data["maxConcurrentShardRequests"] = data["max_concurrent_shard_requests"]
json_data["timeField"] = data["time_field"]
if data.get("interval"):
Expand Down Expand Up @@ -695,8 +694,14 @@ def __init__(self, module):
"Bearer %s" % module.params["grafana_api_key"]
)
else:
self.headers["Authorization"] = basic_auth_header(module.params['url_username'], module.params['url_password'])
org_id = self.organization_by_name(module.params['org_name']) if module.params['org_name'] else module.params['org_id']
self.headers["Authorization"] = basic_auth_header(
module.params["url_username"], module.params["url_password"]
)
org_id = (
self.organization_by_name(module.params["org_name"])
if module.params["org_name"]
else module.params["org_id"]
)
self.switch_organization(org_id)
# }}}

Expand Down Expand Up @@ -734,12 +739,14 @@ def switch_organization(self, org_id):

def organization_by_name(self, org_name):
url = "/api/user/orgs"
organizations = self._send_request(url, headers=self.headers, method='GET')
orga = next((org for org in organizations if org['name'] == org_name))
organizations = self._send_request(url, headers=self.headers, method="GET")

Check warning on line 742 in plugins/modules/grafana_datasource.py

View check run for this annotation

Codecov / codecov/patch

plugins/modules/grafana_datasource.py#L741-L742 

Added lines #L741 - L742 were not covered by tests
orga = next((org for org in organizations if org["name"] == org_name))
if orga:
return orga['orgId']
return orga["orgId"]

Check warning on line 745 in plugins/modules/grafana_datasource.py

View check run for this annotation

Codecov / codecov/patch

plugins/modules/grafana_datasource.py#L745 

Added line #L745 was not covered by tests

return self._module.fail_json(failed=True, msg="Current user isn't member of organization: %s" % org_name)
return self._module.fail_json(

Check warning on line 747 in plugins/modules/grafana_datasource.py

View check run for this annotation

Codecov / codecov/patch

plugins/modules/grafana_datasource.py#L747 

Added line #L747 was not covered by tests
failed=True, msg="Current user isn't member of organization: %s" % org_name
)

def datasource_by_name(self, name):
datasource_exists = False
Expand All @@ -764,77 +771,127 @@ def setup_module_object():
argument_spec = base.grafana_argument_spec()

argument_spec.update(
name=dict(required=True, type='str'),
uid=dict(type='str'),
ds_type=dict(choices=['graphite',
'prometheus',
'elasticsearch',
'influxdb',
'opentsdb',
'mysql',
'postgres',
'cloudwatch',
'alexanderzobnin-zabbix-datasource',
'grafana-azure-monitor-datasource',
'camptocamp-prometheus-alertmanager-datasource',
'sni-thruk-datasource',
'redis-datasource',
'loki',
'tempo']),
ds_url=dict(type='str'),
access=dict(default='proxy', choices=['proxy', 'direct']),
database=dict(type='str', default=""),
user=dict(default='', type='str'),
password=dict(default='', no_log=True, type='str'),
basic_auth_user=dict(type='str'),
basic_auth_password=dict(type='str', no_log=True),
with_credentials=dict(default=False, type='bool'),
tls_client_cert=dict(type='str', no_log=True),
tls_client_key=dict(type='str', no_log=True),
tls_ca_cert=dict(type='str', no_log=True),
tls_skip_verify=dict(type='bool', default=False),
is_default=dict(default=False, type='bool'),
org_id=dict(default=1, type='int'),
org_name=dict(type='str'),
es_version=dict(type='str', default="7.10+", choices=["2", "5", "56", "60",
"70", "7.7+", "7.10+",
"8.0+"]),
max_concurrent_shard_requests=dict(type='int', default=256),
time_field=dict(default='@timestamp', type='str'),
time_interval=dict(type='str'),
interval=dict(type='str', choices=['', 'Hourly', 'Daily', 'Weekly', 'Monthly', 'Yearly'], default=''),
tsdb_version=dict(type='int', default=1, choices=[1, 2, 3]),
tsdb_resolution=dict(type='str', default='second', choices=['second', 'millisecond']),
sslmode=dict(default='disable', choices=['disable', 'require', 'verify-ca', 'verify-full']),
trends=dict(default=False, type='bool'),
aws_auth_type=dict(default='keys', choices=['keys', 'credentials', 'arn', 'default']),
aws_default_region=dict(default='us-east-1', choices=['ap-northeast-1', 'ap-northeast-2', 'ap-southeast-1', 'ap-southeast-2', 'ap-south-1',
'ca-central-1',
'cn-north-1', 'cn-northwest-1',
'eu-central-1', 'eu-west-1', 'eu-west-2', 'eu-west-3',
'sa-east-1',
'us-east-1', 'us-east-2', 'us-gov-west-1', 'us-west-1', 'us-west-2']),
aws_access_key=dict(default='', no_log=True, type='str'),
aws_secret_key=dict(default='', no_log=True, type='str'),
aws_credentials_profile=dict(default='', type='str'),
aws_assume_role_arn=dict(default='', type='str'),
aws_custom_metrics_namespaces=dict(type='str'),
azure_cloud=dict(type='str', default='azuremonitor', choices=['azuremonitor', 'chinaazuremonitor', 'govazuremonitor', 'germanyazuremonitor']),
azure_tenant=dict(type='str'),
azure_client=dict(type='str'),
azure_secret=dict(type='str', no_log=True),
zabbix_user=dict(type='str'),
zabbix_password=dict(type='str', no_log=True),
additional_json_data=dict(type='dict', default={}, required=False),
additional_secure_json_data=dict(type='dict', default={}, required=False),
enforce_secure_data=dict(type='bool', default=False, required=False)
name=dict(required=True, type="str"),
uid=dict(type="str"),
ds_type=dict(
choices=[
"graphite",
"prometheus",
"elasticsearch",
"influxdb",
"opentsdb",
"mysql",
"postgres",
"cloudwatch",
"alexanderzobnin-zabbix-datasource",
"grafana-azure-monitor-datasource",
"camptocamp-prometheus-alertmanager-datasource",
"sni-thruk-datasource",
"redis-datasource",
"loki",
"tempo",
]
),
ds_url=dict(type="str"),
access=dict(default="proxy", choices=["proxy", "direct"]),
database=dict(type="str", default=""),
user=dict(default="", type="str"),
password=dict(default="", no_log=True, type="str"),
basic_auth_user=dict(type="str"),
basic_auth_password=dict(type="str", no_log=True),
with_credentials=dict(default=False, type="bool"),
tls_client_cert=dict(type="str", no_log=True),
tls_client_key=dict(type="str", no_log=True),
tls_ca_cert=dict(type="str", no_log=True),
tls_skip_verify=dict(type="bool", default=False),
is_default=dict(default=False, type="bool"),
org_id=dict(default=1, type="int"),
org_name=dict(type="str"),
es_version=dict(
type="str",
default="7.10+",
choices=["2", "5", "56", "60", "70", "7.7+", "7.10+", "8.0+"],
),
max_concurrent_shard_requests=dict(type="int", default=256),
time_field=dict(default="@timestamp", type="str"),
time_interval=dict(type="str"),
interval=dict(
type="str",
choices=["", "Hourly", "Daily", "Weekly", "Monthly", "Yearly"],
default="",
),
tsdb_version=dict(type="int", default=1, choices=[1, 2, 3]),
tsdb_resolution=dict(
type="str", default="second", choices=["second", "millisecond"]
),
sslmode=dict(
default="disable",
choices=["disable", "require", "verify-ca", "verify-full"],
),
trends=dict(default=False, type="bool"),
aws_auth_type=dict(
default="keys", choices=["keys", "credentials", "arn", "default"]
),
aws_default_region=dict(
default="us-east-1",
choices=[
"ap-northeast-1",
"ap-northeast-2",
"ap-southeast-1",
"ap-southeast-2",
"ap-south-1",
"ca-central-1",
"cn-north-1",
"cn-northwest-1",
"eu-central-1",
"eu-west-1",
"eu-west-2",
"eu-west-3",
"sa-east-1",
"us-east-1",
"us-east-2",
"us-gov-west-1",
"us-west-1",
"us-west-2",
],
),
aws_access_key=dict(default="", no_log=True, type="str"),
aws_secret_key=dict(default="", no_log=True, type="str"),
aws_credentials_profile=dict(default="", type="str"),
aws_assume_role_arn=dict(default="", type="str"),
aws_custom_metrics_namespaces=dict(type="str"),
azure_cloud=dict(
type="str",
default="azuremonitor",
choices=[
"azuremonitor",
"chinaazuremonitor",
"govazuremonitor",
"germanyazuremonitor",
],
),
azure_tenant=dict(type="str"),
azure_client=dict(type="str"),
azure_secret=dict(type="str", no_log=True),
zabbix_user=dict(type="str"),
zabbix_password=dict(type="str", no_log=True),
additional_json_data=dict(type="dict", default={}, required=False),
additional_secure_json_data=dict(type="dict", default={}, required=False),
enforce_secure_data=dict(type="bool", default=False, required=False),
)

module = AnsibleModule(
argument_spec=argument_spec,
supports_check_mode=False,
required_together=[['url_username', 'url_password', 'org_id'], ['tls_client_cert', 'tls_client_key']],
mutually_exclusive=[['url_username', 'grafana_api_key'], ['tls_ca_cert', 'tls_skip_verify'], ['org_id', 'org_name']],
required_together=[
["url_username", "url_password", "org_id"],
["tls_client_cert", "tls_client_key"],
],
mutually_exclusive=[
["url_username", "grafana_api_key"],
["tls_ca_cert", "tls_skip_verify"],
["org_id", "org_name"],
],
required_if=[
["state", "present", ["ds_type", "ds_url"]],
["ds_type", "opentsdb", ["tsdb_version", "tsdb_resolution"]],
Expand Down

0 comments on commit 681b266

Please sign in to comment.