o i @sdZddlZddlmZmZddlmZmZddlm Z e e Z dZ dZ d Zd ZGd d d eZeGd ddZdedefddZd!dededededef ddZd"dededeedefddZdefddZdeeeeefffdd ZdS)#z;Configuration management for supervisor process management.N) dataclassfield)DictOptional) get_loggerPROCESS_AUTO_RECOVERYPROCESS_MAX_START_RETRIES LOG_LEVELSUPERVISOR_CONFIG_PATHc@seZdZdZdS)ConfigurationErrorz5Exception raised for configuration validation errors.N)__name__ __module__ __qualname____doc__rrg/home/ubuntu/.local/lib/python3.10/site-packages/model_hosting_container_standards/supervisor/models.pyr sr c@sdeZdZUdZdZeed<dZeed<dZ e ed<dZ e ed <e e d Zee ee e ffed <d S) SupervisorConfigaConfiguration for supervisor process management system. Hybrid Environment Variable Design: - Application config: PROCESS_ prefixed names (PROCESS_AUTO_RECOVERY, PROCESS_MAX_START_RETRIES, LOG_LEVEL) - Supervisord config: SUPERVISOR_{SECTION}_{KEY} pattern for custom overrides - Section names with colons: Use double underscore __ to represent colon : Examples: - PROCESS_AUTO_RECOVERY=false (application behavior) - PROCESS_MAX_START_RETRIES=5 (application behavior) - LOG_LEVEL=debug (application behavior) - SUPERVISOR_PROGRAM_STARTSECS=10 (supervisord [program] section override) - SUPERVISOR_SUPERVISORD_LOGLEVEL=debug (supervisord [supervisord] section override) - SUPERVISOR_PROGRAM__WEB_COMMAND="gunicorn app:app" (supervisord [program:web] section) - SUPERVISOR_RPCINTERFACE__SUPERVISOR_FACTORY=... (supervisord [rpcinterface:supervisor] section) T auto_recoverymax_start_retries/tmp/supervisord.conf config_pathinfo log_level)default_factorycustom_sectionsN)r rrrrbool__annotations__rintrstrrrdictrrrrrrrs     &rvaluereturncCs |dvS)zParse boolean from string.)true1yeson)lower)r"rrr _parse_bool2s r)dnamedefaultmin_valmax_valcCsxt|}|s |Sz!t|}||kr|ks(nt|d|d|d||WSty;t|d|dw)z-Get integer from environment with validation.z must be between z and z, got z must be an integer, got '')osgetenvrr ValueError)r+r,r-r.r"parsedrrr _get_env_int7s  r4allowedcCs>t||}|r||vrt|d|d|d|S)z,Get string from environment with validation.z must be one of z, got 'r/)r0r1stripr(r )r+r,r5r"rrr _get_env_strHsr7c Csjzt}ttttdttdtt dtt dgd|dWSt y4}z t d|d}~ww) zAParse environment variables and return SupervisorConfig instance.r$rrr)debugrwarnerrorcritical)rrrrrz!Configuration validation failed: N)!_parse_supervisor_custom_sectionsrr)r0r1rr4r r7r r r loggerr:)rerrrparse_environment_variablesPs$ r?c Csddl}|d}i}tjD]Q\}}|dkrq||}|s/|dr.td|dq| d}| d  }| d d  }||vrKi||<| |||<t d |d |d|q|S)a Parse SUPERVISOR_{SECTION}_{KEY}=VALUE environment variables for supervisord configuration. Pattern: SUPERVISOR_SECTION_KEY -> [section] key=value Special handling for section names with colons: - Double underscore __ in section name becomes colon : Examples: - SUPERVISOR_PROGRAM_STARTSECS=10 -> [program] startsecs=10 - SUPERVISOR_SUPERVISORD_LOGLEVEL=debug -> [supervisord] loglevel=debug - SUPERVISOR_PROGRAM__WEB_COMMAND="gunicorn app:app" -> [program:web] command=gunicorn app:app - SUPERVISOR_RPCINTERFACE__SUPERVISOR_FACTORY=... -> [rpcinterface:supervisor] factory=... Skips SUPERVISOR_CONFIG_PATH (used for file path, not supervisord config). Returns: Dictionary mapping section names to their key-value configurations rNz`^SUPERVISOR_(?P
[A-Z0-9]+(?:__[A-Z0-9]+|_[A-Z0-9]+)*)_(?P[A-Z0-9]+(?:_[A-Z0-9]+)*)$r SUPERVISOR_z2Invalid SUPERVISOR_ environment variable format: 'z'. Expected format: SUPERVISOR_SECTION_KEY=value (alphanumeric with underscores, no leading/trailing underscores, use __ for section colons)sectionkey__:z(Found custom supervisor configuration: [z] =)recompiler0environitemsmatch startswithr=warninggroupr(replacer6r8) rFpatternrenv_varr"rJ section_partkey_name section_namerrrr<fs2    r<)rr*)N)rr0 dataclassesrrtypingrrlogging_configrr r=rr r r Exceptionr rr rr)rr4listr7r?r<rrrrs$   "