widdlZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl Z da de de eeffdZdefdZd eddfd Zd edeee eefffd Zd eedeefdZdS)N)partial)CallableDictListSequenceTupleUnion port_rangereturnczdtdtfd}t}|D]$}||r||n%t |D]$}||r||n%t |dkrt d|tt|S)a Get two available ports, one from the start and one from the end of the range :param port_range: Port range to try. Reverses the list and will then pick the first one available :raises OSError: If there are not enough ports available :return: Two available ports port_nrr c tjtjtj}|d|f|dS#t $rYdSwxYw)z Checks if a port is available :param port_nr: Port number to check :return: True if available, False otherwise TF)socketAF_INET SOCK_STREAMbindcloseOSError)rss m/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/mpire/dashboard/utils.py_port_availablez0get_two_available_ports.._port_availablesg  fnf.@AAA FFB= ! ! ! GGIII4   55 sAA A&%A&z@Dashboard Manager Server: there are not enough ports available: ) intboolsetaddreversedlenrtuplesorted)r ravailable_portsrs rget_two_available_portsr$ s      eeO ?7 # #     ( ( ( E J'' ?7 # #     ( ( ( E  ?q  eYceefff (( ) ))ctS)zx Gets the stack level to use when obtaining function details (used for the dashboard) :return: Stack level DASHBOARD_FUNCTION_STACKLEVELr%rget_stacklevelr*3s )(r% stacklevelc |adS)z Sets the stack level to use when obtaining function details (used for the dashboard) :param stacklevel: Stack level Nr')r+s rset_stacklevelr-<s%/!!!r%funccd}d}tjD]K}|jjddddks|dkr|}|dz }|t krnLtj|jdj}|t|d||jddz}t|}|j t|dz z }d d |D}nd }t|tr|j}nCt!|d r3t|t"t$jt$jfs|j} tj|}|jj}|j}n #d }d }d }YnxYw t5j} n#t8$rd } YnwxYw| dt;j||||j||d} | S)aq Obtain function details, including: - function filename - function line number - function name - invoked from filename - invoked from line number - invoked code context :param func: Function to call each time new task arguments become available. When passing on the worker ID the function should receive the worker ID as its first argument. If shared objects are provided the function should receive those as the next argument. If the worker state has been enabled it should receive a state variable as the next argument :return: Function details dictionary Nr__name__.mpirer )context c3>K|]}|VdS)N)strip).0lines r z'get_function_details..ms*FF FFFFFFr%zN/A__call__zn/a@)userfunction_filenamefunction_line_no function_nameinvoked_filenameinvoked_line_noinvoked_code_context) inspectstackframe f_globalssplitr( getframeinfo code_contextindexfind_calling_lineslinenor join isinstancerr.hasattrtypetypes FunctionType MethodTyper; getabsfile__code__co_firstlinenor0getpassgetuserKeyErrorr gethostnamefilename) r. invoked_framer+ frame_inforJrBr>r?r@r= func_detailss rget_function_detailsr`Fs+(MJmoo   %j 1 7 7 < '6,8 ::L s'FFF'' F65F6rJcd}d}d}d}d}tt|dD]N\}}t|D]/}|r ||krd} |dvrd}|}|dkr|dz }d}$|d kr|dz}0|r|dkrnO|| dS) z Tries to find the lines corresponding to the calling function :param code_context: List of code lines :return: List of code lines rFNr )start>"'T)() enumerater)rJn_parentheses_groupsfound_parentheses_group inside_stringinside_string_chline_nrr9chs rrLrLs#MG"8L#9#9CCC 4.. * *B *)))$)Mz!! $ #%  s$)$*.''s$)$ # ';q'@'@ E  ""r%)rXrDr functoolsrtypingrrrrrr rRr(rr$r*r-strr`rLr)r%rrqsH ???????????????? !%*%*U38_%*%*%*%*P)))))/s/t////MxMDeCHo1E,FMMMM`)#T#Y)#49)#)#)#)#)#)#r%