commit bcb93b889d55a22347f2d1f2d01a63c707fb7ef1 Author: Arturo Filastò arturo@filasto.net Date: Thu Jan 12 12:54:17 2017 +0000
Include the deck_id in the returned value --- ooni/deck/deck.py | 8 +++++--- ooni/measurements.py | 13 ++++++++----- ooni/ui/web/server.py | 8 +++++++- ooni/utils/__init__.py | 6 ++++-- 4 files changed, 24 insertions(+), 11 deletions(-)
diff --git a/ooni/deck/deck.py b/ooni/deck/deck.py index fbbb8c3d..c57ccb63 100644 --- a/ooni/deck/deck.py +++ b/ooni/deck/deck.py @@ -234,7 +234,7 @@ class NGDeck(object): # XXX-REFACTOR we do this so late to avoid the collision between the # same id and hence generating the same filename. test_details = net_test_loader.getTestDetails() - task.id = generate_filename(test_details) + task.id = generate_filename(test_details, deck_id=self.id)
measurement_id = None report_filename = task.output_path @@ -330,7 +330,8 @@ class DeckTask(object):
self.ooni = { 'bouncer_client': None, - 'test_details': {} + 'test_details': {}, + 'test_name': None } self.output_path = None
@@ -353,8 +354,9 @@ class DeckTask(object): if required_key not in task_data: raise MissingTaskDataKey(required_key)
+ self.ooni['test_name'] = task_data.pop('test_name') # This raises e.NetTestNotFound, we let it go onto the caller - nettest_path = nettest_to_path(task_data.pop("test_name"), + nettest_path = nettest_to_path(self.ooni['test_name'], self._arbitrary_paths)
annotations = self._pop_option('annotations', task_data, {}) diff --git a/ooni/measurements.py b/ooni/measurements.py index 9197ee75..68b5c555 100644 --- a/ooni/measurements.py +++ b/ooni/measurements.py @@ -106,8 +106,11 @@ def get_measurement(measurement_id, compute_size=False): if compute_size is True: size = directory_usage(measurement.path)
- test_start_time, country_code, asn, test_name = \ - measurement_id.split("-")[:4] + measurement_metadata = measurement_id.split("-") + test_start_time, country_code, asn, test_name = measurement_metadata[:4] + deck_id = "none" + if len(measurement_metadata) > 4: + deck_id = '-'.join(measurement_metadata[4:]) return { "test_name": test_name, "country_code": country_code, @@ -119,8 +122,7 @@ def get_measurement(measurement_id, compute_size=False): "running": running, "stale": stale, "size": size, - # XXX we need the deck ID in here - "deck_id": "none" + "deck_id": deck_id }
@@ -156,8 +158,9 @@ def list_measurements(compute_size=False, order=None): for measurement_id in measurement_path.listdir(): try: measurements.append(get_measurement(measurement_id, compute_size)) - except: + except Exception as exc: log.err("Failed to get metadata for measurement {0}".format(measurement_id)) + log.exception(exc)
if order is None: return measurements diff --git a/ooni/ui/web/server.py b/ooni/ui/web/server.py index c8f191c7..e009cb74 100644 --- a/ooni/ui/web/server.py +++ b/ooni/ui/web/server.py @@ -370,6 +370,12 @@ class WebUIAPI(object): def api_deck_list(self, request): deck_list = {'decks': []} for deck_id, deck in self.director.deck_store.list(): + nettests = [] + for task in deck.tasks: + if task.type == 'ooni': + assert task.ooni['test_name'] is not None + nettests.append(task.ooni['test_name']) + deck_list['decks'].append({ 'id': deck_id, 'name': deck.name, @@ -378,7 +384,7 @@ class WebUIAPI(object): deck_id, from_schedule=False), 'running_scheduled': self.director.isDeckRunning( deck_id, from_schedule=True), - 'nettests': [], #XXX + 'nettests': nettests, 'description': deck.description, 'schedule': deck.schedule, 'enabled': self.director.deck_store.is_enabled(deck_id) diff --git a/ooni/utils/__init__.py b/ooni/utils/__init__.py index 6abd04fd..34b9fbba 100644 --- a/ooni/utils/__init__.py +++ b/ooni/utils/__init__.py @@ -1,7 +1,6 @@ import shutil import string import random -import signal import errno import gzip import os @@ -103,7 +102,7 @@ def randomDate(start, end): LONG_DATE = "%Y-%m-%d %H:%M:%S" SHORT_DATE = "%Y%m%dT%H%M%SZ"
-def generate_filename(test_details, prefix=None, extension=None): +def generate_filename(test_details, prefix=None, extension=None, deck_id=None): """ Returns a filename for every test execution.
@@ -116,6 +115,9 @@ def generate_filename(test_details, prefix=None, extension=None): kwargs["prefix"] = prefix filename_format += "{prefix}-" filename_format += "{timestamp}-{probe_cc}-{probe_asn}-{test_name}" + if deck_id is not None: + kwargs["deck_id"] = deck_id + filename_format += "-{deck_id}" if extension is not None: kwargs["extension"] = extension filename_format += ".{extension}"