U j8E@s<ddlmZddlmZmZmZddlmZddlmZddl m Z dZ dZ dd hZ ed hZd Zed d hZd Zd ZddZddZddZe fddZddZddZddZddZddZd>d!d"Zd#d$Zd%d&Zd'd(Z d?d*d+Z!d,d-Z"d.d/Z#d0d1Z$d2d3Z%d@d4d5Z&d6d7Z'd8d9Z(d:d;Z)dYnX||}|jdtfdtfdtfgdddd|jdtfdtfgd dd |jd tfdtfdtfgd dd |S) Nz Coleccion z creada como coleccion normal. centro_id camera_idr#TZ!idx_feeder_score_centro_camera_ts)uniquename backgroundZidx_feeder_score_centro_ts)r*r+scopeZ idx_feeder_score_scope_centro_ts)list_collection_namescreate_collectionrr create_indexr)scapp_db database_name collectionrrrensure_feeder_score_collection2s0  r3c Cst|tr|gn|}t|ts&tdg}|D]}t|tsJtd||d}|s`td|dd}|tkrtd||dd |D|t|d ||d pid q.|S) Nz&Payload debe ser dict o lista de dictszDocumento invalido: r(z-Falta 'camera_id' en alguno de los documentosr,r zscope invalido: cSsi|]\}}|dkr||qS)_idr).0keyrrrr bsz?validate_and_normalize_feeder_score_payload..r#data)r(r#r,r8) rdictlistrget VALID_SCOPESappenditemsr&)payloaddocs normalizeddocr(r,rrr+validate_and_normalize_feeder_score_payloadNs,       rCcCsnt|}t}d}|pgD]6}|j||d|dd|||ddd|d7}qtd |d |d |S) Nrr(r#)r'r(r#)r'updatedT)upsertr z FeederScore: z* documentos almacenados/actualizados para .)r3rnow replace_oner)r0r@r'r2Z updated_atsavedrBrrrsave_feeder_score_payloadls$  rJc Csz2||||ddddidtfdtfg}WnBtk rt}z$td|jjd|iWYSd}~XYnXtt}|D]L}| d}t |t sqd | d d i| d pi|| d | d<qtt |S) Nz$gtez$ltr'r#r4rr#r(zError leyendo feeder score (z): r,r r8%d/%m/%Y)findsortr Exceptionr __traceback__ tb_linenorr9r;rrstrftimesortedr>) r2r' start_dayend_daycursorr%groupedrBr#rrrfetch_feeder_score_datas.    rYcCsPt|krdS|tjd|iddddddgd}|sr5dayr>rrrr7sz0compute_feeder_score_by_area..) rrint_fetch_docs_by_day_and_centerrr9r>_build_center_rollup_cellrT) r0area_idrU day_rangesrV grouped_docsresponserdcentersr'r@cellrrrcompute_feeder_score_by_areas rnNcCs&t|}|tt|d}t|dd|di||dd}tt}|D]\}} tt} | D]R\} } t| | } | s|qd| d}t |dkrd |ddn|}| | | qd|dkr| D]}|D]} | ||| d <qqqL| D]\}}t|||||<qqLd d t|DS) Nr]r_r`rarKrLcenterr*cSs"i|]\}}|tt|qSrrbrcrrrr7sz2compute_feeder_score_by_region..)rrrerfrr9r>r:rgsplitlenjoinr=values_build_area_rollup_cellrT)r0 region_idrUri granularityrVrjrkrdrlZ area_cellsr'r@Z center_cellpartsrhZcenters_in_arearrrcompute_feeder_score_by_regions.  " rycCst|kriS|t|ddidtfdtfdtfdtfg}tdd}|D]@}|d}|d}t|trR|szqR|| d | |qR|S) Nr4rr#r'r,r(cSsttSr)rr:rrrrz/_fetch_docs_by_day_and_center..rM) rZr-rNrOrrr;rrrSr=)r0Z match_filterrWrXrBr#r'rrrrfs     rfc sTtdd|Ddfdd|D}dkr2dSdp>i}d}d}d}g}t}|D]} | dpji} |t| dd7}|ttt| d d7}|ttt| d d7}|t| d | d dk rZ|| d qZd||d|d |ddt |d||t ||dt ||dd } t |dt rP|d| d<| S)Ncss.|]&}|ddks"|ddkr|VqdS)r,r r( __CENTER__Nr;r5rBrrr sz,_build_center_rollup_cell..cs&g|]}|k r|ddkr|qS)r(r|r}r~Z center_docrr sz-_build_center_rollup_cell..r8rdelivered_food_kg alarmas_totalalarmas_no_atendidas reaction_timescorer(r status aggregation food_weightedn_jaulas_scoredn_jaulas_fasted) r,r*rrrrrrrrrrw)nextr;set_numberre _alarm_valueextend_reaction_samplesaddround_summarize_reactionrrrr9) r'r@Z cage_docsZ center_datafood alarm_totalalarm_unattendedreaction_samplesZ scored_cagesrBr8rmrrrrgsB  rgc Cstdd|D}d}d}d}d}g}d}|D]} t| d} tdt| dd} | dk r|d7}| dkr|| | 7}|| 7}|tt| dd7}|tt| d d7}|t| d q.d ||rd nd |dkrt||dnddt|d||t||t dd|Ddd S)Ncss$|]}tdt|ddVqdS)rrN)maxrr;r5rmrrrrsz*_build_area_rollup_cell..rrrrr rrrareafinalZfastedrorrcss|]}|dVqdS)rNr}rrrrr8s) r,r*rrrrrrrZn_centros_scoredr) sumrr;rrerrrr aggregate_rw) rhZ center_cellsrZ score_numZ score_denrrrZscored_centersrmrweightrrrrus<  rurc Csdd|p gD}|sdS|dk}d}d}d}t}t}t} t} t} |D]} tdt| dd} |rr| nd}|dkrd}||7}|| 7}||t| dd7}|t| |t| | d r| | d | t| d d| | d iqP|dks&t |d kr*dSt t |}|t krDdSi}t | D] }d}d}d}|D]} | d pxi|pi}tdt| dd} |r| nd}|dkrd}tdtdt|d d}tdtdt|dd}|||7}||||7}|||7}qf|dkr4t||dnd|dkrNt||dndd||<qPt |}t |d kr||dnt||t | d krt t | ntt | d krt t | ndt|r|n|dt||d|dS)NcSs4g|],}t|trt|dtr|dr|qS)dimsrr9r;)r5kernelrrrr>s   z aggregate_rw..rrfood_kg?r deliv_pctconfig_versionbaserr covqr)rr) algo_version algo_versionscontract_versionrrrrr)rrrr;update_kernel_algo_versionsr_rw_contract_versionkeysrrriterSUPPORTED_RW_CONTRACT_VERSIONSrTminrMIXED_ALGO_VERSIONMIXED_CONFIG_VERSION)ZkernelsrchildrenZweightedZ total_weightZ food_totalZ deliv_numrZcontract_versionsZconfig_versionsbasesZdim_keysrrrrrdimZcov_numZq_numZq_denitemrrZsorted_algo_versionsrrrr<s|       rc Csxt|s dS|pidpi}t|p&idt|dd}t|pFidd}t|pZidd}t|pnidd }d}d}|d piD]r\} } tdt|| d} tdtd t| d d} tdtd t| d d} || | | 7}|| | 7}q|dkr||nd }ttdt|ddd|||}|rTd|nd}t|t||d|||S)NweightsrrZsfrCapg.@Z sfrPerPctg?ZallowOverscoreTrrrrrrgY@)is_recomputabler;rboolr>rr)rparamsrrZsfr_capZ sfr_per_pctZallow_overscorenumZdenrrrrrZq_valueZbonusZceilingrrrreweights& (rcCs4t|to2t|tko2t|dto2t|dS)Nr)rr9rrr;r)rrrrrs   rcCs\t|tsdS|d}|dkr2|dtkr2tSt|d}|dk rXt|rXt|SdS)Nrr) rr9r;#LEGACY_RW_CONTRACT_V1_ALGO_VERSIONSLEGACY_RW_CONTRACT_VERSIONrfloat is_integerre)rversionnumberrrrrs   rcCsjt|tr|dnd}t|tttfr>dd|D}|r>|St|trR|dnd}|rdt|hSdhS)NrcSsh|]}|rt|qSr)r)r5rrrr sz(_kernel_algo_versions..rZ __unknown__)rr9r;r:tuplerr)rversionscleanedrrrrrsrc Cs.z t|}Wnttfk r(|YSX|Sr)r TypeErrorr)rfallbackrrrrrs   rcCsRt|tr|dni}t|tr:||dk r:||St|trN||SdS)NZreasons_summaryr)r8r6reasonsrrrrs rcCs|t|tsgS|d}t|tr0dd|DS|d|d}tt|dd}t|}|dksn|dkrrgS|g|S)N samples_seccSs g|]}t|dk rt|qSrrr5rrrrrs z%_reaction_samples..mean_secavgnr)rr9r;r:rer)rsamplesrrZ avg_numberrrrrs   rcCsXtdddd|pgDD}|s.dgdSt|t|t|t|d|d|dS) Ncss|]}|dk r|VqdSrrrrrrrsz&_summarize_reaction..css|]}t|VqdSrr)r5rrrrrsr)rr_r)rrp95_secmax_secr)rTrrr _percentile)rrrrrrs  rcCsr|sdSt|dkr|dSt|d|d}t|}t|dt|d}||}|||||||S)Nr rd)rrrer)Z sorted_valuesZpctposlowhighfracrrrrs r)N)r)N)+ collectionsrrrrpymongorpymongo.errorsrAnalytic.centro_idsrSCAPP_DATABASE_NAMErZr< frozensetrrrrrrrr&r3rCrJrYr\rnryrfrgrurrrrrrrrrrrrrrsL       ($ J