From 93708bb31e2edd71f4cb8ac922b8afae263de293 Mon Sep 17 00:00:00 2001
From: schneefux <schneefux+commit@schneefux.xyz>
Date: Wed, 8 Mar 2017 18:00:21 +0100
Subject: allow batched priorities

---
 joblib.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'joblib.py')

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:
-- 
cgit v1.3.1