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
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
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)