hastic-server/analytics/bin/server

#!/usr/bin/env python3

import sys
import os

#TODO: make wrapper script that set PYTHONPATH instead
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'analytics'))

import config
import json
import logging
import asyncio
import traceback

import services
from analytic_unit_manager import AnalyticUnitManager


root = logging.getLogger()
logger = logging.getLogger('SERVER')


server_service: services.ServerService = None
data_service: services.DataService = None
analytic_unit_manager: AnalyticUnitManager = None


root.setLevel(logging.DEBUG)


logging_formatter = logging.Formatter("%(asctime)s [Analytics] [%(levelname)-5.5s]  %(message)s")

logging_handler = logging.StreamHandler(sys.stdout)
#logging_handler = logging.FileHandler(config.DATA_FOLDER + '/analytics.log')
logging_handler.setLevel(logging.DEBUG)
logging_handler.setFormatter(logging_formatter)

root.addHandler(logging_handler)


async def handle_task(task: object):
    try:

        logger.info("Command is OK")

        task_result_payload = {
            '_id': task['_id'],
            'task': task['type'],
            'analyticUnitId': task['analyticUnitId'],
            'status': "IN_PROGRESS"
        }

        if not task['type'] == 'PUSH':
            message = services.server_service.ServerMessage('TASK_RESULT', task_result_payload)
            await server_service.send_message(message)

        res = await analytic_unit_manager.handle_analytic_task(task)
        res['_id'] = task['_id']

        if not task['type'] == 'PUSH':
            message = services.server_service.ServerMessage('TASK_RESULT', res)
            await server_service.send_message(message)

    except Exception as e:
        error_text = traceback.format_exc()
        logger.error("handle_task Exception: '%s'" % error_text)

async def handle_message(message: services.ServerMessage):
    payload = None
    if message.method == 'TASK':
        await handle_task(message.payload)

def init_services():
    logger.info("Starting services...")
    logger.info("Server...")
    server_service = services.ServerService(handle_message)
    logger.info("Ok")
    logger.info("Data service...")
    data_service = services.DataService(server_service)
    logger.info("Ok")
    logger.info("Analytic unit manager...")
    analytic_unit_manager = AnalyticUnitManager()
    logger.info("Ok")

    return server_service, data_service, analytic_unit_manager

async def app_loop():
    await server_service.handle_loop()


if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    logger.info("Ok")
    server_service, data_service, analytic_unit_manager = init_services()
    print('Analytics process is running') # we need to print to stdout and flush
    sys.stdout.flush()                    # because node.js expects it

    loop.run_until_complete(app_loop())