added: batch runners

Author: adamcupial

.gitignore

@@ -122,3 +122,4 @@ dmypy.json
 .venv/
 
 # End of https://www.gitignore.io/api/python
+*.csv

lighthouse/__init__.py

@@ -1 +1,2 @@
-from .runner import LighthouseRunner
+from .runner import LighthouseRunner, LighthouseRepeatRunner
+from .batch import BatchRunner

lighthouse/batch.py

@@ -0,0 +1,25 @@
+from runner import LighthouseRepeatRunner
+from itertools import product
+from tqdm import tqdm
+
+class BatchRunner(object):
+    def __init__(self, urls, form_factors, quiet=True,
+                 additional_settings=None, repeats=3):
+
+        if not isinstance(form_factors, (list, )):
+            form_factors = list(form_factors)
+        if not isinstance(urls, (list, )):
+            urls = list(urls)
+
+        all_combinations = tqdm(list(product(urls, form_factors)),
+                                desc='Running urls')
+
+        self.reports = []
+
+        for url, factor in all_combinations:
+            all_combinations.set_description('{0} | {1}'.format(url, factor))
+            report = LighthouseRepeatRunner(url, factor, quiet,
+                                            additional_settings,
+                                            repeats).report
+            self.reports.append((url, factor, report))
+

lighthouse/runner.py

@@ -5,6 +5,9 @@
 import os
 import subprocess
 import tempfile
+from collections import namedtuple
+from functools import reduce
+from tqdm import tqdm
 
 # Own
 from report import LighthouseReport
@@ -70,3 +73,30 @@ def _get_report(self):
 
     def _clean(self):
         os.remove(self.__report_path)
+
+
+class LighthouseRepeatRunner(object):
+    def __init__(self, url, form_factor='mobile', quiet=True,
+                 additional_settings=None, repeats=3):
+        reports = []
+
+        progress = tqdm(range(1, repeats + 1), desc='Repeating test')
+
+        for i in progress:
+            progress.set_description('Run {0}/{1}'.format(i, repeats))
+            reports.append(LighthouseRunner(url, form_factor=form_factor,
+                                            quiet=quiet,
+                                            additional_settings=additional_settings).report)   # noqa: E501
+
+        report = namedtuple('LighthouseAveragedReport', 'timings, score')
+        self.report = report(
+            timings=self._get_average([getattr(x, 'timings') for x in reports]),  # noqa: E501
+            score=self._get_average([getattr(x, 'score') for x in reports])
+        )
+
+    def _get_average(self, obj_lst):
+        ret = {}
+        for key in obj_lst[0].keys():
+            lst = [x.get(key) for x in obj_lst]
+            ret[key] = reduce(lambda a, b: a + b, lst) / len(lst)
+        return ret

setup.py

@@ -9,5 +9,7 @@
     author_email='cupial.adam@gmail.com',
     url='n/a',
     packages=find_packages(),
-    install_requires=[]
+    install_requires=[
+        'tqdm>=4.30',
+    ]
 )