small fix

This commit is contained in:
shirshasa
2020-05-20 15:24:26 +03:00
parent a9e8023804
commit 36a7bea9ad
2 changed files with 4 additions and 3 deletions

View File

@@ -40,7 +40,7 @@ class Access:
params = json.load(f) params = json.load(f)
self.refreshing.clear() self.refreshing.clear()
self.url = url or params['url'] self.url = url
self.username = params['username'] self.username = params['username']
self.password = params['password'] self.password = params['password']
self.refreshing.set() self.refreshing.set()

View File

@@ -47,10 +47,11 @@ def callback(ch, method, properties, body, logger):
logger.info(f'Message: {body}.') logger.info(f'Message: {body}.')
try: try:
data = json.loads(body) 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 = { params = {
'book_id': data['id'], 'book_id': data['id'],
'access': Access(url=api_url), 'access': Access(url=data['apiURL']),
'logger': logger 'logger': logger
} }