def inventory_ajax(env): """Backend endpoint for inventory table""" draw = int(request.args.get('draw', 0)) envs = environments() check_env(env, envs) headers, fact_names = inventory_facts() query = AndOperator() fact_query = OrOperator() fact_query.add([EqualsOperator("name", name) for name in fact_names]) query.add(fact_query) if env != '*': query.add(EqualsOperator("environment", env)) facts = puppetdb.facts(query=query) fact_data = {} for fact in facts: if fact.node not in fact_data: fact_data[fact.node] = {} fact_data[fact.node][fact.name] = fact.value total = len(fact_data) return render_template('inventory.json.tpl', draw=draw, total=total, total_filtered=total, fact_data=fact_data, columns=fact_names)
def nodes(env): """Fetch all (active) nodes from PuppetDB and stream a table displaying those nodes. Downside of the streaming aproach is that since we've already sent our headers we can't abort the request if we detect an error. Because of this we'll end up with an empty table instead because of how yield_or_stop works. Once pagination is in place we can change this but we'll need to provide a search feature instead. :param env: Search for nodes in this (Catalog and Fact) environment :type env: :obj:`string` """ envs = environments() status_arg = request.args.get('status', '') check_env(env, envs) query = AndOperator() if env != '*': query.add(EqualsOperator("catalog_environment", env)) if status_arg in ['failed', 'changed', 'unchanged']: query.add(EqualsOperator('latest_report_status', status_arg)) elif status_arg == 'unreported': unreported = datetime.utcnow() unreported = (unreported - timedelta(hours=app.config['UNRESPONSIVE_HOURS'])) unreported = unreported.replace(microsecond=0).isoformat() unrep_query = OrOperator() unrep_query.add(NullOperator('report_timestamp', True)) unrep_query.add(LessEqualOperator('report_timestamp', unreported)) query.add(unrep_query) if len(query.operations) == 0: query = None nodelist = puppetdb.nodes( query=query, unreported=app.config['UNRESPONSIVE_HOURS'], with_status=True, with_event_numbers=app.config['WITH_EVENT_NUMBERS']) nodes = [] for node in yield_or_stop(nodelist): if status_arg: if node.status == status_arg: nodes.append(node) else: nodes.append(node) return Response( stream_with_context( stream_template('nodes.html', nodes=nodes, envs=envs, current_env=env)))
def test_or_with_no_operations(self): op = OrOperator() with pytest.raises(APIError): repr(op) with pytest.raises(APIError): str(op) with pytest.raises(APIError): str(op)
def _puppetdb_nodes(rules, master_id, master_address): if "facts" not in rules or not rules["facts"]: return set() if rules["match_type"] == "ALL": operator = AndOperator() else: operator = OrOperator() for fact_rule in rules["facts"]: rule_op = fact_rule["operator"] lhs = ["fact", fact_rule["fact"]] rhs = fact_rule["value"] if rule_op == "=": operator.add(EqualsOperator(lhs, rhs)) elif rule_op == "!=": notop = NotOperator() notop.add(EqualsOperator(lhs, rhs)) operator.add(notop) elif rule_op == "~": operator.add(RegexOperator(lhs, rhs)) elif rule_op == "!~": notop = NotOperator() notop.add(RegexOperator(lhs, rhs)) operator.add(notop) elif rule_op == ">": operator.add(GreaterOperator(lhs, rhs)) elif rule_op == ">=": operator.add(GreaterEqualOperator(lhs, rhs)) elif rule_op == "<": operator.add(LessOperator(lhs, rhs)) elif rule_op == "<=": operator.add(LessEqualOperator(lhs, rhs)) query_str = str(operator).replace("'", '"') cert_instance = CertsClass(master_id) cert_path = cert_instance.get_cert() private_key_path = cert_instance.get_key() req = requests.get( f"https://{master_address}:8081/pdb/meta/v1/version", verify=False, cert=(cert_path, private_key_path), ) if req.status_code == 404: # PuppetDB <= 2.x nodes_uri = f"https://{master_address}:8081/v4/nodes?query={query_str}" else: # PuppetDB >= 3.x nodes_uri = ( f"https://{master_address}:8081/pdb/query/v4/nodes?query={query_str}" ) # Fetch the matching nodes req = requests.get(nodes_uri, verify=False, cert=(cert_path, private_key_path)) return {node["certname"] for node in req.json()}
def report(env, node_name, report_id): """Displays a single report including all the events associated with that report and their status. The report_id may be the puppetdb's report hash or the configuration_version. This allows for better integration into puppet-hipchat. :param env: Search for reports in this environment :type env: :obj:`string` :param node_name: Find the reports whose certname match this value :type node_name: :obj:`string` :param report_id: The hash or the configuration_version of the desired report :type report_id: :obj:`string` """ envs = environments() check_env(env, envs) query = AndOperator() report_id_query = OrOperator() report_id_query.add(EqualsOperator("hash", report_id)) report_id_query.add(EqualsOperator("configuration_version", report_id)) if env != '*': query.add(EqualsOperator("environment", env)) query.add(EqualsOperator("certname", node_name)) query.add(report_id_query) reports = puppetdb.reports(query=query) try: report = next(reports) except StopIteration: abort(404) report.version = commonmark.commonmark(report.version) return render_template('report.html', report=report, events=yield_or_stop(report.events()), logs=report.logs, metrics=report.metrics, envs=envs, current_env=env)
def test_or_operator(self): op = OrOperator() op.add(EqualsOperator("operatingsystem", "CentOS")) op.add([EqualsOperator("architecture", "x86_64"), GreaterOperator("operatingsystemmajrelease", 6)]) assert str(op) == '["or", ["=", "operatingsystem", "CentOS"], '\ '["=", "architecture", "x86_64"], '\ '[">", "operatingsystemmajrelease", 6]]' assert repr(op) == 'Query: ["or", '\ '["=", "operatingsystem", "CentOS"], '\ '["=", "architecture", "x86_64"], '\ '[">", "operatingsystemmajrelease", 6]]' assert str(op) == '["or", ["=", "operatingsystem", "CentOS"], ' \ '["=", "architecture", "x86_64"], '\ '[">", "operatingsystemmajrelease", 6]]' with pytest.raises(APIError): op.add({"query1": '["=", "catalog_environment", "production"]'})
def reports_ajax(env, node_name): """Query and Return JSON data to reports Jquery datatable :param env: Search for all reports in this environment :type env: :obj:`string` """ draw = int(request.args.get('draw', 0)) start = int(request.args.get('start', 0)) length = int(request.args.get('length', app.config['NORMAL_TABLE_COUNT'])) paging_args = {'limit': length, 'offset': start} search_arg = request.args.get('search[value]') order_column = int(request.args.get('order[0][column]', 0)) order_filter = REPORTS_COLUMNS[order_column].get( 'filter', REPORTS_COLUMNS[order_column]['attr']) order_dir = request.args.get('order[0][dir]', 'desc') order_args = '[{"field": "%s", "order": "%s"}]' % (order_filter, order_dir) status_args = request.args.get('columns[1][search][value]', '').split('|') date_args = request.args.get('columns[0][search][value]', '') max_col = len(REPORTS_COLUMNS) for i in range(len(REPORTS_COLUMNS)): if request.args.get("columns[%s][data]" % i, None): max_col = i + 1 envs = environments() check_env(env, envs) reports_query = AndOperator() if env != '*': reports_query.add(EqualsOperator("environment", env)) if node_name: reports_query.add(EqualsOperator("certname", node_name)) if search_arg: search_query = OrOperator() search_query.add(RegexOperator("certname", r"%s" % search_arg)) search_query.add(RegexOperator("puppet_version", r"%s" % search_arg)) search_query.add( RegexOperator("configuration_version", r"%s" % search_arg)) reports_query.add(search_query) if date_args: dates = json.loads(date_args) if len(dates) > 0: date_query = AndOperator() if 'min' in dates: date_query.add(GreaterEqualOperator('end_time', dates['min'])) if 'max' in dates: date_query.add(LessEqualOperator('end_time', dates['max'])) reports_query.add(date_query) status_query = OrOperator() for status_arg in status_args: if status_arg in ['failed', 'changed', 'unchanged']: arg_query = AndOperator() arg_query.add(EqualsOperator('status', status_arg)) arg_query.add(EqualsOperator('noop', False)) status_query.add(arg_query) if status_arg == 'unchanged': arg_query = AndOperator() arg_query.add(EqualsOperator('noop', True)) arg_query.add(EqualsOperator('noop_pending', False)) status_query.add(arg_query) elif status_arg == 'noop': arg_query = AndOperator() arg_query.add(EqualsOperator('noop', True)) arg_query.add(EqualsOperator('noop_pending', True)) status_query.add(arg_query) if len(status_query.operations) == 0: if len(reports_query.operations) == 0: reports_query = None else: reports_query.add(status_query) if status_args[0] != 'none': reports = get_or_abort(puppetdb.reports, query=reports_query, order_by=order_args, include_total=True, **paging_args) reports, reports_events = tee(reports) total = None else: reports = [] reports_events = [] total = 0 # Convert metrics to relational dict metrics = {} for report in reports_events: if total is None: total = puppetdb.total metrics[report.hash_] = {} for m in report.metrics: if m['category'] not in metrics[report.hash_]: metrics[report.hash_][m['category']] = {} metrics[report.hash_][m['category']][m['name']] = m['value'] if total is None: total = 0 return render_template('reports.json.tpl', draw=draw, total=total, total_filtered=total, reports=reports, metrics=metrics, envs=envs, current_env=env, columns=REPORTS_COLUMNS[:max_col])
def report(env, node_name, report_id, show_error_as): """Displays a single report including all the events associated with that report and their status. The report_id may be the puppetdb's report hash or the configuration_version. This allows for better integration into puppet-hipchat. :param env: Search for reports in this environment :type env: :obj:`string` :param node_name: Find the reports whose certname match this value :type node_name: :obj:`string` :param report_id: The hash or the configuration_version of the desired report :type report_id: :obj:`string` :param show_error_as: 'friendly' or 'raw', the former means that messages will be show in a mode transformed for human-readability, the latter that the messages will be unchanged :param show_error_as: :obj:`string` """ envs = environments() check_env(env, envs) query = AndOperator() report_id_query = OrOperator() report_id_query.add(EqualsOperator("hash", report_id)) report_id_query.add(EqualsOperator("configuration_version", report_id)) if env != '*': query.add(EqualsOperator("environment", env)) query.add(EqualsOperator("certname", node_name)) query.add(report_id_query) reports = puppetdb.reports(query=query) try: report = next(reports) except StopIteration: abort(404) if show_error_as not in ['friendly', 'raw']: abort(404) report.version = commonmark.commonmark(report.version) events = [{ 'resource': f"{event.item['type']}[{event.item['title']}]", 'status': event.status, 'old': event.item['old'], 'new': event.item['new'], 'failed': event.failed, } for event in report.events()] logs = [ { 'timestamp': log['time'], 'level': log["level"], 'source': log['source'], 'tags': ', '.join(log['tags']), 'message': get_message(node_name, log, show_error_as), 'location': get_location(log), # this could be also done with a different rendered in DataTables, # - feel free to refactor it into that if you know how 'short_location': get_short_location(get_location(log)), } for log in report.logs ] return render_template('report.html', report=report, events=events, logs=logs, metrics=report.metrics, envs=envs, current_env=env, current_show_error_as=show_error_as)