summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--joblib.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/joblib.py b/joblib.py
index 3b97fe3..22f51d1 100644
--- a/joblib.py
+++ b/joblib.py
@@ -47,13 +47,17 @@ class JobQueue(object):
payloads = payload
else:
payloads = [payload]
+ if isinstance(priority, list):
+ priorities = priority
+ else:
+ priorities = [priority] * len(payloads)
ids = []
for pl in payloads:
ids.append(await con.fetchval("""
INSERT INTO jobs(type, payload, priority)
VALUES($1, $2, $3)
RETURNING id
- """, jobtype, json.dumps(pl), priority))
+ """, jobtype, json.dumps(pl), priorities))
if isinstance(payload, list):
return ids
else: