34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from flask_script import Manager
|
|
|
|
from lemur.constants import SUCCESS_METRIC_STATUS
|
|
from lemur.dns_providers.service import get_all_dns_providers, set_domains
|
|
from lemur.extensions import metrics
|
|
from lemur.plugins.base import plugins
|
|
|
|
manager = Manager(
|
|
usage="Iterates through all DNS providers and sets DNS zones in the database."
|
|
)
|
|
|
|
|
|
@manager.command
|
|
def get_all_zones():
|
|
"""
|
|
Retrieves all DNS providers from the database. Refreshes the zones associated with each DNS provider
|
|
"""
|
|
print("[+] Starting dns provider zone lookup and configuration.")
|
|
dns_providers = get_all_dns_providers()
|
|
acme_plugin = plugins.get("acme-issuer")
|
|
|
|
for dns_provider in dns_providers:
|
|
try:
|
|
zones = acme_plugin.get_all_zones(dns_provider)
|
|
set_domains(dns_provider, zones)
|
|
except Exception as e:
|
|
print("[+] Error with DNS Provider {}: {}".format(dns_provider.name, e))
|
|
set_domains(dns_provider, [])
|
|
|
|
status = SUCCESS_METRIC_STATUS
|
|
|
|
metrics.send("get_all_zones", "counter", 1, metric_tags={"status": status})
|
|
print("[+] Done with dns provider zone lookup and configuration.")
|