From 7a6bc2f7749c10c1d3344ea0196e81ae95b7f0d7 Mon Sep 17 00:00:00 2001
From: Lars Wirzenius <liw@liw.fi>
Date: Sun, 21 Jul 2019 11:45:29 +0300
Subject: Add: TokenGetter

---
 ick2/workapi.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'ick2/workapi.py')

diff --git a/ick2/workapi.py b/ick2/workapi.py
index 3eeb0fa..7ff6f93 100644
--- a/ick2/workapi.py
+++ b/ick2/workapi.py
@@ -41,6 +41,7 @@ class WorkAPI(ick2.APIbase):
         ]
 
     def get_work(self, token=None, **kwargs):
+        token = self._token_getter.get_token()
         worker_id = self._get_client_id(**kwargs)
         ick2.log.log(
             'trace', msg_text='Worker wants work', worker_id=worker_id)
@@ -140,6 +141,8 @@ class WorkAPI(ick2.APIbase):
         return None
 
     def update_work(self, update, token=None, **kwargs):
+        token = self._token_getter.get_token()
+
         try:
             worker_id = update['worker']
             build_id = update['build_id']
-- 
cgit v1.2.1