wi # ddlZddlZ ddlmZn#e$r ddlmZYnwxYwddlZddlZddl Z ddl Z ddl m Z ddl m Z mZddlmZddlmZmZmZmZmZddlmZmZmZmZddlmZdd lmZdd l m!Z!dd l"m#a#m$Z$m%Z%m&Z&dd l'm(Z(ej)e*Z+ej)d Z,e,-ej.ee*Z/da0e1eddz dz d5Z2e23Z4dddn #1swxYwYda5da6da7e!Z8e/9dde:fdZ;e/9dde:fdZddfdedee:ee?e:fffdZ@ejAd%dZBd&de?deeeCe:fddfd ZDd!e d"e?d#eeeefddfd$ZEdS)'N)files)datetime)EventProcess) BaseProxy)DictOptionalSequenceTupleUnion)Flaskjsonifyrender_templaterequestescape) make_server)DashboardStartedEvent)$DASHBOARD_MANAGER_CONNECTION_DETAILSget_manager_client_dictsshutdown_manager_serverstart_manager_server)get_two_available_portswerkzeugzmpire.dashboard templateszprogress_bar.htmlr/returnc tj}n#t$rd}YnwxYwtd|t jt jpdt jS)z2 Obtain the index HTML :return: HTML zn/az index.html localhost)usernamehostname manager_hostmanager_port_nr) getpassgetuserKeyErrorrsocket gethostnamerhostport)users q/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/mpire/dashboard/dashboard.pyindexr.(sv    <$ASAUAU(L(Q(`U`+O+T V V VVs  %%z/_progress_bar_updatecJtj}g}ttD]}t|}|dd|d<|dr|d|dkrt ||dz ddd |d <|d >|d |kr2t |d |z ddd nd |d <||t|S)z Obtain progress bar updates (should be called through AJAX) :return: JSON string containing progress bar updates totalN?successn started_raw.rduration finished_raw- remainingresult) rnowsorted_DASHBOARD_TQDM_DICTkeysgetstrrsplitappendr)r=r<pb_idprogresss r-progress_bar_updaterG:s7 ,..C F,113344   '++E22 G  $ #HW  I  /8C=HW4E#E#E#&sXm-D'D#E#E#L#LSRS#T#TUV#WHZ (0(@(LQYZhQiloQoQo&).)AC)G%H%H%O%OPSUV%W%WXY%Z%Z*- [ !  h & ! ! !!z/_progress_bar_newcttjd}tjddk}t|}|dt jdtjkrd|d<nd |d|d<g}|rUt|dD]?}| d |d |d |d |d |d |d |d|d |d|d |d|d |d@d |}ttj d|||rdnddd|DS)z Obtain a piece of HTML for a new progress bar (should be called through AJAX) :return: JSON string containing new progress bar HTML rE has_insightstruer,@z{}:n_jobszz blocknone)idinsights_workersrJc4i|]\}}|t|Sr).0kvs r- z$progress_bar_new..us$6m6m6m1q&))6m6m6mrHr;rV)intrargs_DASHBOARD_TQDM_DETAILS_DICTrAr%r&r(r)formatrangerDjoinr_progress_bar_htmlitems)rErJprogress_bar_detailsrT worker_ids r-progress_bar_newreSs1  W% & &E</69L8;;EBBF#'/*;*;'T'Tf>P>R>R'T'TTT')V$$',||4H4P'Q'QV$ .3H=>> . .I  # #%-y%-%-27%-%-JS%-%-27%-%-JS%-%-38%-%-KT%-%-38 %-%-KT %-%- 38 %-%- KT %-%- 38 %-%- KT %-%-%- . . . .yy!122 ,3ouO_LXAd^doo6m6mPdPjPjPlPl6m6m6moo p p pprHii port_rangec>tstt|\}}t |at jdt j ttt|tfdda tt|t jpt%jt jdSt+d)z Starts a new MPIRE dashboard :param port_range: Port range to try. :return: A dictionary containing the dashboard port number and manager host and port number being used rTzdashboard-process)targetr\daemonname)dashboard_port_nrr#r$z$You already have a running dashboard)DASHBOARD_STARTED_EVENTis_setinitrr_DASHBOARD_MANAGERlogging getLoggersetLevelWARNr_runr_server_processstartwaitrr*r(r)r+ RuntimeError)rfrkr$s r-start_dashboardryxs  # ) ) + +C$$&&&-DZ-P-P*?2/BB *%%..w|<<<!5LN_8P8R8R5T+/6IKKK $$&&&&7 D I aVM_MaMa#G#LNN N ABBBrHc4tr~tBtjtjt jtttda da da t dSdS)z Shuts down the dashboard N)rlrmruoskillpidsignalSIGINTr`rror?r]resetrVrHr-shutdown_dashboardrs%%'' (  & GO' 7 7 7  " " " 2333!#'+$%%''''' ( (rHr$r#cJtrtd|pd}|t_|t_ t n5#t$r(td|d|wxYwt dS)z Connects to an existing MPIRE dashboard :param manager_port_nr: Port to use when connecting to a manager :param manager_host: Host to use when connecting to a manager. If ``None`` it will use localhost z/You're already connected to a running dashboardz 127.0.0.1z*Could not connect to dashboard manager at :N) rlrmrxrr*r+rConnectionRefusedErrordecodeset)r$r#s r-connect_to_dashboardrs%%''NLMMM .;L0<(-0?(-S """" !SSS$&R(4(;(;(=(=&R&R@O&R&RSS SS!!!!!s A2Bstartedrkmanager_client_dictsc|\aa}td|t}|t d||dS)a  Starts a dashboard server :param started: Event that signals the dashboard server has started :param manager_host: Dashboard manager host :param manager_port_nr: Dashboard manager port number :param dashboard_port_nr: Dashboard port number z0.0.0.0zServer started on 0.0.0.0:N)r?r]rapprloggerinfo serve_forever)rrkr_servers r-rtrtsh=Q96$5s ; ;F KKMMM KK@->@@AAA rH)rN)N)Fatexitr%importlib.resourcesrresource ImportErrorimportlib_resourcesrpr{r~r(rmultiprocessingrrmultiprocessing.managersrtypingrr r r r flaskr rrr markupsaferwerkzeug.servingr"mpire.dashboard.connection_classesrmpire.dashboard.managerrrrrmpire.dashboard.utilsrrq__name__rlogger_werkzeugrrERRORrruopenfpreadraror?r]rlrouterBr.rGrer_r[ryregisterrbytesrrtrVrHr-rs@ 65555555666555555556 ********......99999999999999::::::::::::((((((DDDDDDnnnnnnnnnnnn999999  8 $ $#'#J//''' eHoo T(($ % % 36I I3OO#SU################//113VsVVVV" "##"S"""$#"0   !p#!p!p!p! !pH,15t+<+< C C Cc5QTVYQY?FZA[ C C C CF ( ( ( (""#"XeESVJFW=X"dh""""6%CuYXaclMlGmrvs &DD D