diff --git a/lemur/__init__.py b/lemur/__init__.py index 3c8923a5..cb05a768 100644 --- a/lemur/__init__.py +++ b/lemur/__init__.py @@ -82,5 +82,5 @@ def configure_hook(app): return response return json_handler - for code in default_exceptions.iterkeys(): + for code, value in default_exceptions.items(): app.error_handler_spec[None][code] = make_json_handler(code) diff --git a/lemur/manage.py b/lemur/manage.py index 5c5c4a6e..1fd5601b 100755 --- a/lemur/manage.py +++ b/lemur/manage.py @@ -7,7 +7,6 @@ import os import sys import base64 import time -import arrow import requests import json @@ -770,46 +769,17 @@ def publish_verisign_units(): requests.post('http://localhost:8078/metrics', data=json.dumps(metric)) -class Rolling(Command): +@manager.command +def publish_unapproved_verisign_certificates(): """ - Rotates existing certificates to a new one on an ELB + Query the Verisign for any certificates that need to be approved. + :return: """ - option_list = ( - Option('-w', '--window', dest='window', default=24), - ) - - def run(self, window): - """ - Simple function that queries verisign for API units and posts the mertics to - Atlas API for other teams to consume. - :return: - """ - end = arrow.utcnow() - start = end.replace(hours=-window) - items = Certificate.query.filter(Certificate.not_before <= end.format('YYYY-MM-DD')) \ - .filter(Certificate.not_before >= start.format('YYYY-MM-DD')).all() - - metrics = {} - for i in items: - name = "{0},{1}".format(i.owner, i.issuer) - if metrics.get(name): - metrics[name] += 1 - else: - metrics[name] = 1 - - for name, value in metrics.iteritems(): - owner, issuer = name.split(",") - metric = [ - { - "timestamp": 1321351651, - "type": "GAUGE", - "name": "Issued Certificates", - "tags": {"owner": owner, "issuer": issuer, "window": window}, - "value": value - } - ] - - requests.post('http://localhost:8078/metrics', data=json.dumps(metric)) + from lemur.plugins import plugins + from lemur.extensions import metrics + v = plugins.get('verisign-issuer') + certs = v.get_pending_certificates() + metrics.send('pending_certificates', 'gauge', certs) class Report(Command): @@ -951,7 +921,6 @@ def main(): manager.add_command("create_role", CreateRole()) manager.add_command("provision_elb", ProvisionELB()) manager.add_command("rotate_elbs", RotateELBs()) - manager.add_command("rolling", Rolling()) manager.add_command("sources", Sources()) manager.add_command("report", Report()) manager.run() diff --git a/lemur/plugins/lemur_verisign/plugin.py b/lemur/plugins/lemur_verisign/plugin.py index c8042a89..b2ea30bc 100644 --- a/lemur/plugins/lemur_verisign/plugin.py +++ b/lemur/plugins/lemur_verisign/plugin.py @@ -188,6 +188,25 @@ class VerisignIssuerPlugin(IssuerPlugin): response = self.session.post(url, headers={'content-type': 'application/x-www-form-urlencoded'}) return handle_response(response.content)['Response']['Order'] + def get_pending_certificates(self): + """ + Uses Verisign to fetch the number of certificate awaiting approval. + + :return: + """ + url = current_app.config.get("VERISIGN_URL") + '/reportingws' + + end = arrow.now() + start = end.replace(days=-7) + data = { + 'reportType': 'summary', + 'certProductType': 'Server', + 'startDate': start.format("MM/DD/YYYY"), + 'endDate': end.format("MM/DD/YYYY"), + } + response = self.session.post(url, data=data) + return response.json()['certificateSummary'][0]['Pending'] + class VerisignSourcePlugin(SourcePlugin): title = 'Verisign'