1 files changed, 65 insertions, 0 deletions
diff --git a/mirrors/management/commands/mirrorresolv.py b/mirrors/management/commands/mirrorresolv.py
new file mode 100644
index 00000000..0e71894b
--- /dev/null
+++ b/mirrors/management/commands/mirrorresolv.py
@@ -0,0 +1,65 @@
+# -*- coding: utf-8 -*-
+"""
+mirrorresolv command
+
+Poll all mirror URLs and determine whether they have IPv4 and/or IPv6 addresses
+available.
+
+Usage: ./manage.py mirrorresolv
+"""
+
+from django.core.management.base import NoArgsCommand
+
+import sys
+import logging
+import socket
+
+from mirrors.models import MirrorUrl
+
+logging.basicConfig(
+    level=logging.WARNING,
+    format='%(asctime)s -> %(levelname)s: %(message)s',
+    datefmt='%Y-%m-%d %H:%M:%S',
+    stream=sys.stderr)
+logger = logging.getLogger()
+
+class Command(NoArgsCommand):
+    help = "Runs a check on all active mirror URLs to determine if they are reachable via IPv4 and/or v6."
+
+    def handle_noargs(self, **options):
+        v = int(options.get('verbosity', 0))
+        if v == 0:
+            logger.level = logging.ERROR
+        elif v == 1:
+            logger.level = logging.WARNING
+        elif v >= 2:
+            logger.level = logging.DEBUG
+
+        return resolve_mirrors()
+
+def resolve_mirrors():
+    logger.debug("requesting list of mirror URLs")
+    for mirrorurl in MirrorUrl.objects.filter(active=True, mirror__active=True):
+        try:
+            # save old values, we can skip no-op updates this way
+            oldvals = (mirrorurl.has_ipv4, mirrorurl.has_ipv6)
+            logger.debug("resolving %3i (%s)", mirrorurl.id, mirrorurl.hostname)
+            families = mirrorurl.address_families()
+            mirrorurl.has_ipv4 = socket.AF_INET in families
+            mirrorurl.has_ipv6 = socket.AF_INET6 in families
+            logger.debug("%s: v4: %s v6: %s", mirrorurl.hostname,
+                    mirrorurl.has_ipv4, mirrorurl.has_ipv6)
+            # now check new values, only update if new != old
+            newvals = (mirrorurl.has_ipv4, mirrorurl.has_ipv6)
+            if newvals != oldvals:
+                logger.debug("values changed for %s", mirrorurl)
+                mirrorurl.save(update_fields=('has_ipv4', 'has_ipv6'))
+        except socket.gaierror, e:
+            if e.errno == socket.EAI_NONAME:
+                logger.debug("gaierror resolving %s: %s", mirrorurl.hostname, e)
+            else:
+                logger.warn("gaierror resolving %s: %s", mirrorurl.hostname, e)
+        except socket.error, e:
+            logger.warn("error resolving %s: %s", mirrorurl.hostname, e)
+
+# vim: set ts=4 sw=4 et: