From f17367711d51329e5612eb2d6ace890fb79d5ca6 Mon Sep 17 00:00:00 2001 From: Dongkuo Ma Date: Fri, 25 Oct 2019 14:38:08 +0800 Subject: [PATCH] populate MetaJob from job (#2) --- src/varanus/models.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/varanus/models.py b/src/varanus/models.py index bf55e5d..a18ca72 100644 --- a/src/varanus/models.py +++ b/src/varanus/models.py @@ -102,6 +102,11 @@ def __init__(self, job: dict): self.metadata = job or {} self.__dict__.update(self.metadata) + def populate_from_job(self, job): + self.items = job.items.stats()['totals']['input_values'] + self.pages = job.requests.stats()['totals']['input_values'] + self.logs = job.logs.stats()['totals']['input_values'] + class Job(MetaJob): @@ -118,6 +123,7 @@ def __init__(self, project, key=None, defaults=None, data=None): except StopIteration: pass self.metajob = MetaJob(dict(metadata)) + self.metajob.populate_from_job(data) self.__dict__.update(self.metajob.__dict__) for key in defaults: setattr(self, key, defaults[key])