Exemplo n.º 1
0
def get_config(c):
    config = {
        'client_id':
        c.get('clientId') or
        CONFIG['APP_NAME'] + '@worker-' + str(toolkit.gen_time_serial_seq()),
        'protocol':
        mqtt.MQTTv5,
        'transport':
        c.get('transport') or 'tcp',
    }
    return config
Exemplo n.º 2
0
    def __init__(self,
                 logger,
                 tmp_folder=None,
                 date_time_sub_tmp_folder=False,
                 third_part_tool_root_path=None):

        self.logger = logger

        self.tmp_folder = os.path.join('/tmp', tmp_folder or str(uuid.uuid4()))
        if date_time_sub_tmp_folder:
            date_time_string = str(toolkit.gen_time_serial_seq(rand_length=10))
            self.tmp_folder = os.path.join(self.tmp_folder, date_time_string)

        os.makedirs(self.tmp_folder)

        self.third_part_tool_root_path = third_part_tool_root_path or DEFAULT_THIRD_PART_TOOLS_ROOT_PATH
Exemplo n.º 3
0
import redis
import psutil

# Project Modules
from worker.utils import yaml_resources, toolkit

# Configure
print_detail = sys.argv[0] == '_celery.py'

base_path = os.path.dirname(os.path.abspath(__file__))
CONFIG = yaml_resources.load_config(os.path.join(base_path, '../config.yaml'),
                                    print_detail=print_detail)
IMAGE_INFO = yaml_resources.load_file(
    'IMAGE_INFO', os.path.join(base_path, '../image-info.json'))

WORKER_ID = toolkit.gen_time_serial_seq()

# For monitor
MAIN_PROCESS = psutil.Process()
MAIN_PROCESS.cpu_percent(interval=1)

CHILD_PROCESS_MAP = {}  # PID -> Process

HEARTBEAT_EXEC_TIMESTAMP = 0

from worker.app_init import before_app_create, after_app_created

# Disable InsecureRequestWarning
import requests
from requests.packages.urllib3.exceptions import InsecureRequestWarning
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)