allow batched priorities

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: