From 36a7bea9ad5b02d920af9639828f94b1608c52d5 Mon Sep 17 00:00:00 2001 From: shirshasa Date: Wed, 20 May 2020 15:24:26 +0300 Subject: [PATCH] small fix --- src/access.py | 2 +- src/consumer.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/access.py b/src/access.py index f96971f..f481aea 100644 --- a/src/access.py +++ b/src/access.py @@ -40,7 +40,7 @@ class Access: params = json.load(f) self.refreshing.clear() - self.url = url or params['url'] + self.url = url self.username = params['username'] self.password = params['password'] self.refreshing.set() diff --git a/src/consumer.py b/src/consumer.py index 019a6be..74dcfda 100644 --- a/src/consumer.py +++ b/src/consumer.py @@ -47,10 +47,11 @@ def callback(ch, method, properties, body, logger): logger.info(f'Message: {body}.') try: data = json.loads(body) - api_url = data['apiURL'] if 'apiURL' in data else None + assert 'apiURL' in data, 'No apiURL field in received message.' + params = { 'book_id': data['id'], - 'access': Access(url=api_url), + 'access': Access(url=data['apiURL']), 'logger': logger }