checker: Make it possible to manually check a domain again

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/dbl/checker.py b/src/dbl/checker.py
index 8960c66619053ab97582779c88eed419e78d4a19..0e4a5ed7132e8e755a291748f8923879ddf3b7da 100644 (file)
--- a/src/dbl/checker.py
+++ b/src/dbl/checker.py
@@ -28,6 +28,7 @@ import sqlmodel
 
 from . import database
 from . import domains
+from . import util
 
 # Setup logging
 log = logging.getLogger(__name__)
@@ -54,7 +55,9 @@ class Checker(object):
                threshold = 1024
 
                # If no domains have been passed, fetch some from the database
-               if not domains:
+               if domains:
+                       domains = util.aiterable(domains)
+               else:
                        domains = self.get_domains()
 
                with self.executor:

diff --git a/src/dbl/util.py b/src/dbl/util.py
index b43f64d4f5acd505c8ab8ff179d400b0fbcd651b..021360b1df3651d3f5feec130a12de738c005e53 100644 (file)
--- a/src/dbl/util.py
+++ b/src/dbl/util.py
@@ -31,6 +31,10 @@ from .i18n import _
 # Setup logging
 log = logging.getLogger(__name__)
 
+async def aiterable(iterable):
+       for item in iterable:
+               yield item
+
 class Stopwatch(object):
        def __init__(self, task):
                self.task = task