From 1f8af16a2ad223bce1ae998206d7f1eaf16d86f9 Mon Sep 17 00:00:00 2001 From: Barbara Miller Date: Fri, 25 Jul 2025 14:11:46 -0700 Subject: [PATCH] make tests pass, even with account_id --- brozzler/job_schema.yaml | 5 +++++ brozzler/model.py | 12 ++++++++---- 2 files changed, 13 insertions(+), 4 deletions(-) diff --git a/brozzler/job_schema.yaml b/brozzler/job_schema.yaml index 59b831f..e28c329 100644 --- a/brozzler/job_schema.yaml +++ b/brozzler/job_schema.yaml @@ -105,3 +105,8 @@ max_claimed_sites: pdfs_only: type: boolean + +account_id: + type: + - string + - integer diff --git a/brozzler/model.py b/brozzler/model.py index ac4c109..7f2612a 100644 --- a/brozzler/model.py +++ b/brozzler/model.py @@ -98,10 +98,14 @@ def new_job(frontier, job_conf): frontier.rr, {"conf": job_conf, "status": "ACTIVE", "started": doublethink.utcnow()}, ) - job.id = job_conf.get("id") - job.account_id = job_conf.get("account_id") - job.max_claimed_sites = job_conf.get("max_claimed_sites") - job.pdfs_only = job_conf.get("pdfs_only") + if "id" in job_conf: + job.id = job_conf["id"] + if "max_claimed_sites" in job_conf: + job.max_claimed_sites = job_conf["max_claimed_sites"] + if "pdfs_only" in job_conf: + job.pdfs_only = job_conf["pdfs_only"] + if "account_id" in job_conf: + job.account_id = job_conf.["account_id"] job.save() sites = []