
    `f                     $   d Z ddlZddlZddlZddlZddlmZ ddlmZ dZ	dZ
ej                            d          pdZ eej                            d	          pe
          ZdgZ eed
          ZdZdZdZdZ G d de          Zd ZddZdS )a6  Remote Debugger.

Introduction
============

This is a remote debugger for Celery tasks running in multiprocessing
pool workers.  Inspired by a lost post on dzone.com.

Usage
-----

.. code-block:: python

    from celery.contrib import rdb
    from celery import task

    @task()
    def add(x, y):
        result = x + y
        rdb.set_trace()
        return result

Environment Variables
=====================

.. envvar:: CELERY_RDB_HOST

``CELERY_RDB_HOST``
-------------------

    Hostname to bind to.  Default is '127.0.0.1' (only accessible from
    localhost).

.. envvar:: CELERY_RDB_PORT

``CELERY_RDB_PORT``
-------------------

    Base port to bind to.  Default is 6899.
    The debugger will try to find an available port starting from the
    base port.  The selected port will be logged by the worker.
    N)Pdb)current_process)CELERY_RDB_HOSTCELERY_RDB_PORTDEFAULT_PORTRdbdebugger	set_tracei  r   z	127.0.0.1r   	_getframezs{self.ident}: Couldn't find an available port.

Please specify one using the CELERY_RDB_PORT environment variable.
z{self.ident}: Ready to connect: telnet {self.host} {self.port}

Type `exit` in session to continue.

{self.ident}: Waiting for client...
z5{self.ident}: Now in session with {self.remote_addr}.z4{self.ident}: Session with {self.remote_addr} ended.c                        e Zd ZdZdZdZdZeedde	j
        f fd	ZddZd Zd	 Zd
 Zd Zd ZexZZd ZexZZd Z xZS )r   zRemote debugger.zRemote DebuggerNd   r   c                    d| _         || _        t          j        t          j        f| _        |                     ||||          \  | _        }| j                            d           | j        	                    d           | j
         d| | _        || _        || _        |                     t                              |                      | j                                        \  | _        }| j                            d           d                    d |D                       | _        |                     t*                              |                      | j                            d          x| _        xt          _        t          _        t1                                          d| j        | j                   d S )	NT   :selfc              3   4   K   | ]}t          |          V  d S N)str).0vs     D/var/www/html/env/lib/python3.11/site-packages/celery/contrib/rdb.py	<genexpr>zRdb.__init__.<locals>.<genexpr>n   s(      #<#<qCFF#<#<#<#<#<#<    rwtab)completekeystdinstdout)activeoutsysr   r   _prev_handlesget_avail_port_socksetblockinglistenmeidenthostportsayBANNERformataccept_clientjoinremote_addrSESSION_STARTEDmakefile_handlesuper__init__)	r   r*   r+   port_search_limit	port_skewr!   	this_portaddress	__class__s	           r   r7   zRdb.__init__[   s    Y
2 $ 3 3$)9!
 !

I 	
q!!!
!--)--
		D))*** $
 1 1 3 3g  ###88#<#<G#<#<#<<<''T'22333040E0Ed0K0KKKsy3:U#|DL 	 	B 	B 	B 	B 	Br   c                 n   	 t                      j                            d          \  }}t          |          }n# t          $ r Y nw xY wd }t          |          D ]}t          j        t          j        t          j                  }|	                    t          j
        t          j        d           ||z   |z   }	 |                    ||f           ||fc S # t          $ r*}	|	j        t          j        t          j        fv rY d }	~	 d }	~	ww xY wt#          t$                              |                     )N-r   r   )r   namesplitint
ValueErrorrangesocketAF_INETSOCK_STREAM
setsockopt
SOL_SOCKETSO_REUSEADDRbindOSErrorerrno
EADDRINUSEEINVAL	ExceptionNO_AVAILABLE_PORTr.   )
r   r*   r+   search_limitskew_r:   ir%   excs
             r   r$   zRdb.get_avail_portt   sH   	%'',22377GAtt99DD 	 	 	D		|$$ 	( 	(AM&.&2DEEEV.0CQGGGtaI(

D),--- i''''    9!15< @@@HHHH )00d0;;<<<s-   8; 
AA;C
D"DDDc                 2    t          || j                   d S )N)file)printr!   )r   ms     r   r,   zRdb.say   s    adhr   c                     | S r    r   s    r   	__enter__zRdb.__enter__   s    r   c                 .    |                                   d S r   )_close_session)r   exc_infos     r   __exit__zRdb.__exit__   s    r   c                    | j         x\  | _        | _        \  t          _        t          _        | j        r| j        | j                                         | j        | j                                         | j        | j                                         d| _        | 	                    t                              |                      d S d S )NFr   )r#   r   r   r"   r    r5   closer0   r%   r,   SESSION_ENDEDr.   r   s    r   r^   zRdb._close_session   s    :>:LL
DK"7#)SZ; 	6|'""$$$|'""$$$z%
  """DKHH]))t)4455555	6 	6r   c                 V    |                                   |                                  dS Nr   )r^   set_continuer   args     r   do_continuezRdb.do_continue   s+    qr   c                 V    |                                   |                                  dS re   )r^   set_quitrg   s     r   do_quitzRdb.do_quit   s&    qr   c                 .    t          j        d            d S r   )r"   settracer   s    r   rk   zRdb.set_quit   s    Tr   )r   r   )__name__
__module____qualname____doc__r(   
_prev_outsr%   r   r   r"   r   r7   r$   r,   r\   r`   r^   ri   do_cdo_contrl   do_qdo_exitrk   __classcell__)r<   s   @r   r   r   T   s        	BJE+/#&"#*B B B B B B2= = = =*         
6 
6 
6   ! D7   D7      r   r   c                  b    t           d         } | | j        st                      x} t           d<   | S )z8Return the current debugger instance, or create if none.r   )_currentr    r   )rdbs    r   r	   r	      s-    
1+C
{#*{EE!hqkJr   c                 n    | t                      j        } t                                          |           S )z:Set break-point at current location, or a specified frame.)_framef_backr	   r
   )frames    r   r
   r
      s*    }::&&&r   r   )rr   rL   osrD   r"   pdbr   billiard.processr   __all__r   environgetr   rA   r   rz   getattrr}   rP   r-   r3   rc   r   r	   r
   r[   r   r   <module>r      s:  ) )T  				  



       , , , , , ,
 *..!233B{#bjnn%677G<HH 6	k	"	" 
 JFX X X X X# X X Xv  ' ' ' ' ' 'r   