forked from LiveCarta/BookConverter
[LAW-3158] domen defining via api
- automatically define domen for uploaded books from queue's message
This commit is contained in:
@@ -23,19 +23,24 @@ class Access:
|
|||||||
|
|
||||||
refreshing = Event()
|
refreshing = Event()
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self, url):
|
||||||
self.read_credentials()
|
"""
|
||||||
|
|
||||||
|
:param url: str, url received from queue message, if field apiURL exists
|
||||||
|
else None
|
||||||
|
"""
|
||||||
|
self.set_credentials(url)
|
||||||
self.get_token()
|
self.get_token()
|
||||||
self.refreshing.set()
|
self.refreshing.set()
|
||||||
|
|
||||||
def read_credentials(self):
|
def set_credentials(self, url):
|
||||||
folder_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
folder_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||||||
config_path = os.path.join(folder_path, "config/api_config.json")
|
config_path = os.path.join(folder_path, "config/api_config.json")
|
||||||
with open(config_path, "r") as f:
|
with open(config_path, "r") as f:
|
||||||
params = json.load(f)
|
params = json.load(f)
|
||||||
|
|
||||||
self.refreshing.clear()
|
self.refreshing.clear()
|
||||||
self.url = params['url']
|
self.url = url or params['url']
|
||||||
self.username = params['username']
|
self.username = params['username']
|
||||||
self.password = params['password']
|
self.password = params['password']
|
||||||
self.refreshing.set()
|
self.refreshing.set()
|
||||||
|
|||||||
@@ -42,14 +42,15 @@ def convert_book(book_id, access, logger):
|
|||||||
# print('Book has been proceeded.')
|
# print('Book has been proceeded.')
|
||||||
|
|
||||||
|
|
||||||
def callback(ch, method, properties, body, access, logger):
|
def callback(ch, method, properties, body, logger):
|
||||||
print(f'Message: {body}.')
|
print(f'Message: {body}.')
|
||||||
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
|
||||||
params = {
|
params = {
|
||||||
'book_id': data['id'],
|
'book_id': data['id'],
|
||||||
'access': access,
|
'access': Access(url=api_url),
|
||||||
'logger': logger
|
'logger': logger
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -114,10 +115,9 @@ if __name__ == '__main__':
|
|||||||
logger.log(logging.ERROR, f'Queue {conf_param["queue"]} is not declared.')
|
logger.log(logging.ERROR, f'Queue {conf_param["queue"]} is not declared.')
|
||||||
raise exc
|
raise exc
|
||||||
|
|
||||||
acs = Access()
|
|
||||||
channel.basic_consume(queue=conf_param['queue'],
|
channel.basic_consume(queue=conf_param['queue'],
|
||||||
auto_ack=True,
|
auto_ack=True,
|
||||||
on_message_callback=partial(callback, access=acs, logger=logger))
|
on_message_callback=partial(callback, logger=logger))
|
||||||
logger.info('Connection has been established.')
|
logger.info('Connection has been established.')
|
||||||
print('Waiting for messages...')
|
print('Waiting for messages...')
|
||||||
logger.info('Waiting for messages...')
|
logger.info('Waiting for messages...')
|
||||||
|
|||||||
Reference in New Issue
Block a user