o Á¿i– ã@sXdZddlZddlZddlmZmZGdd„deƒZGdd„dƒZde d e fd d „Z dS) zR Control Socket Client Client library for connecting to gunicorn control socket. éN)ÚControlProtocolÚ make_requestc@seZdZdZdS)ÚControlClientErrorzControl client error.N)Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r úG/home/ubuntu/.local/lib/python3.10/site-packages/gunicorn/ctl/client.pyrsrc@s\eZdZdZddedefdd„Zdd„Zd d „Zdd ed e de fdd„Z dd„Z dd„Z d S)Ú ControlClientzÛ Client for connecting to gunicorn control socket. Can be used as a context manager: with ControlClient('/path/to/gunicorn.ctl') as client: result = client.send_command('show workers') ç>@Ú socket_pathÚtimeoutcCs||_||_d|_d|_dS)z¨ Initialize control client. Args: socket_path: Path to the Unix socket timeout: Socket timeout in seconds (default 30) Nr)r rÚ_sockÚ _request_id)Úselfr rr r r Ú__init__"s zControlClient.__init__c Csv|jrdSzt tjtj¡|_|j |j¡|j |j¡WdStjy:}zd|_t d|j›d|›ƒ‚d}~ww)zq Connect to control socket. Raises: ControlClientError: If connection fails NzFailed to connect to z: ) rÚsocketÚAF_UNIXÚ SOCK_STREAMÚ settimeoutrÚconnectr Úerrorr)rÚer r r r/s€þzControlClient.connectcCs6|jrz|j ¡Wn tyYnwd|_dSdS)zClose connection.N)rÚcloseÚ Exception©rr r r rAs ÿ ûzControlClient.closeNÚcommandÚargsÚreturnc Cs¢|js| ¡|jd7_t|j||ƒ}zt |j|¡t |j¡}Wnty;}z | ¡t d|›ƒ‚d}~ww|  d¡dkrKt |  dd¡ƒ‚|  di¡S)a. Send command and wait for response. Args: command: Command string (e.g., "show workers") args: Optional additional arguments Returns: Response data dictionary Raises: ControlClientError: If communication fails ézCommunication error: NÚstatusrz Unknown errorÚdata) rrrrrÚ write_messageÚ read_messagerrrÚget)rrrÚrequestÚresponserr r r Ú send_commandJs€þ zControlClient.send_commandcCs | ¡|S©N)rrr r r Ú __enter__jszControlClient.__enter__cGs | ¡dSr))r)rrr r r Ú__exit__ns zControlClient.__exit__)r r))rrrrÚstrÚfloatrrrÚlistÚdictr(r*r+r r r r r s    r ÚlinercCsbt |¡}|s dgfSg}g}|D]}|s$| ¡s$| d¡s$| |¡q| |¡qd |¡|fS)z Parse a command line into command and args. Args: line: Command line string Returns: Tuple of (command_string, args_list) Úú-ú )ÚshlexÚsplitÚisdigitÚ startswithÚappendÚjoin)r0ÚpartsÚ command_partsrÚpartr r r Ú parse_commandrs   r=) rr4rÚgunicorn.ctl.protocolrrrrr r,Útupler=r r r r ÚsZ