U jT@s`ddlmZmZmZddlmZddlmZdZdZddZ d d Z d d Z d dZ ddZ dS))datetime timedeltatimezone)CollectionInvalid)_to_utc_datetimegenerador_comap_timeseriescCs|d\}}}|dtS)u>Construye el nombre de la colección COMAP dedicada al centro.-_)splitCOMAP_COLLECTION_SUFFIX) centro_idr centror3/var/www/itgmanager/itgmanager/nanoox/comap_data.pyget_comap_collection_namesrc Cs||krzr|j|ddddd}z|jddgdd Wn@tk rz}z"td |d |jjd |W5d }~XYnXWntk rYnX||S)u0Crea la time-series COMAP si todavía no existe. timestampcomap_idminutes) timeField metaField granularity) timeseries)rr)rZidx_comap_latest)nameu&No fue posible crear índice COMAP en  linea : N)list_collection_namescreate_collection create_index Exceptionprint __traceback__ tb_linenor)dbcollection_name collectionerrr_ensure_comap_collections*  r)c Csz|rt|ts&td|WdS|d}t|tr>|sRtd|WdSt|}g}|D]\}}|r|t|tstd|d|qf|d}|std|d |qfz t|} WnPtk r} z0td |d |d | jj d | WYqfW5d} ~ XYnXt |dr(| nt |} t|} | | | |||d|d|| qf|spWdSt||} | j|ddtdt|d|d|t|WStk r} z"td| jj d | WYdSd} ~ XYnXdS)u- Guarda un documento por generador y recepción en la time-series del centro. No deduplica: si el centro reenvía una medición, se conserva nuevamente. La fecha visible se mantiene en ``timestamp_str`` y ``timestamp`` contiene el BSON Date UTC requerido por MongoDB time-series. uSin bloque COMAP válido para r generatorszSin generadores COMAP para u#Generador COMAP vacío o inválido z para rzGenerador COMAP z sin timestamp para uTimestamp COMAP inválido para /rrN isoformatr )r timestamp_strrgenerador_namerF)orderedz Guardados z documentos COMAP en zError en save_comap_data linea ) isinstancedictr"getritemsrr!r#r$hasattrr,strupdateappendr) insert_manylen)r%rgenerators_comapr*r& documentsgenerator_namegenerator_datareceived_timestampZ timestamp_utcr(r-documentr'rrrsave_comap_data4sl        r@cCsHt|tsdS|d}t|ts&dSdd|D}|rDt|SdS)zDevuelve el timestamp mas reciente del bloque COMAP, o None. Un centro COMAP puro no tiene modulos ni bombas, asi que la hora del endpoint no puede salir de ahi: se toma del ultimo snapshot del generador. Nr*cSs0g|](}t|trt|dtr|dqS)r)r0r1r2r5).0datarrr s z.get_latest_comap_timestamp..)r0r1r2valuesmax)r:r*Z timestampsrrrget_latest_comap_timestamps   rFc CsXdii}z t|}||kr&|WSttjttd}ddd|iiidddd id d d d idig}||j|ddD]}| d}| d}|r|t |t sq|t |} | dd} | dd| dd| dd| dd| dd| dk r| | d<| |d|<q|Wn<t k rR} ztd| jjd| W5d} ~ XYnX|S)uJDevuelve el último snapshot de cada generador con el contrato del emisor.r*)daysz$matchrz$gtez$sortrr)rrz$groupz$generador_namez$firstz$$ROOT)_idsnapshotT) allowDiskUserHrIr-Nrr.rz%Error en get_latest_comap_data linea r)rrrnowrutcrCOMAP_LATEST_WINDOW_DAYS aggregater2r0r1popr!r"r#r$) r%rresponser&Z fecha_limitepipelineresultr<rIr=Zoriginal_timestampr(rrrget_latest_comap_datasD             *rSN)rrrpymongo.errorsr save_datarr rMrr)r@rFrSrrrrs  O