diff options
Diffstat (limited to 'joblib.py')
| -rw-r--r-- | joblib.py | 6 |
1 files changed, 5 insertions, 1 deletions
@@ -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: |
