U ^jOf@sDddlZejddddlmZddlmZmZddlmZddl Z ddl m Z ddl m Z mZmZmZmZddl m Z mZdd lmZdd lmZddlZdd lmZddlZed ddlmZddlm Z!Gd d d Z"e"Z#ddZ$d(ddddZ%d)ddZ&d*ddZ'd+ddZ(ddZ)d,d"d#Z*d$d%Z+d&d'Z,dS)-Nz./) defaultdict)datetime timedelta)Path)get_column_letter)BorderSide PatternFillFont Alignment)rcolumn_index_from_string)Image)BytesIOAggc@s,eZdZddZddZddZddZd S) SimpleLoggercCstd|dS)Nz[INFO] printselfmsgr6/var/www/itgmanager/itgmanager/nanoox/reports_utils.pyinfo szSimpleLogger.infocCstd|dS)Nz [WARNING] rrrrrwarning#szSimpleLogger.warningcCstd|dS)Nz[ERROR] rrrrrerror&szSimpleLogger.errorcCstd|dS)Nz[DEBUG] rrrrrdebug)szSimpleLogger.debugN)__name__ __module__ __qualname__rrrrrrrrrsrcCst}|d}|S)z Retorna la ruta al escritorio del usuario independientemente del sistema operativo. Usa Path.home() y asume que la carpeta se llama 'Desktop'. Desktop)rhome)r desktoprrrget_escritorio_path1s r"sensores_timeseries)returncCsntd|||}|}dgdgd}|D]2\}}||kr6|j||dtd|d|q6dS)Nz4En 'asegurar_indices_sensores_timeseries' coll_name=)bomba_id) centro_idr&)Z bomba_id_1Z centro_id_1)nameu Índice 'z' creado sobre )loggerrindex_informationitems create_indexr)db coll_namets_colZexisting_indexesZdesired_indexesr(specrrr$asegurar_indices_sensores_timeseries?sr1metadatac Cs,ttd|td|td|||}t|d}|tdd}d||di}t|d |id |d d d diddid ddiiddddddddigigiddddddddddddddddiddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/id0ig}|j|d1d2} td3d4| DS)5u Obtiene todas las lecturas de un día, agrupadas por alias de bomba, usando un aggregation pipeline para offload al servidor Mongo. Args: db: instancia de la base de datos. fecha_str: fecha en 'YYYY-MM-DD'. centro_id: opcional, filtra por centro. Returns: Dict donde la clave es el alias y el valor es la lista de documentos de timeseries. 9En 'obtener_datos_por_fecha_por_alias' sensors_coll_name=:En 'obtener_datos_por_fecha_por_alias' metadata_coll_name=z fecha_str=%Y-%m-%dr&days timestampz$gtez$ltz$matchz$lookupr%_idmeta)fromZ localFieldZ foreignFieldasz$unwindz$metaz meta.tipobombaz$projectrz$ifNullz $meta.aliasz$concatbomba_z $toStringz $bomba_id)r:aliasr8O2SO2TEMPSetPoint SetPointOFFRLYInStartInSelControlPresionIoVoalarmsStatusz$groupz$aliasz$pushz $timestampz$O2z$SO2z $SENSOR_SO2z$TEMPz$RLYz$InStartz$InSelz$Controlz$Presionz$Ioz$Voz $SetPointz $SetPointOFFz$alarmsz$Status)r8rArBZ SENSOR_SO2rCrFrGrHrIrJrKrLrDrErMrNr:ZlecturasT) allowDiskUsecSsi|]}|d|dqSrOr).0docrrr sz5obtener_datos_por_fecha_por_alias..)rr)rrstrptimer aggregate) r- fecha_strsensors_coll_namemetadata_coll_namesensores_ts_colfecha siguiente_diaZmatch_tspipelinecursorrrr!obtener_datos_por_fecha_por_alias[s   3 6r^c Cs(ttd|td|||}||}td|t|d}|tdd}ddi}i} ||D]"} | d d | d | | d <qvd t| i||d d} ddddddddddd } t t} || | D]4} | | dt | d}| d| | | qtt| S)Nr3r4z metadata_col=r5r&r6tipor>r@r?r:z$inr9)r%r8r) r:r%r8rArBrCrFrGrIrJr%)rr)rrrTrfindgetlistkeysrstrpopappenddict)r-rVrWrX metadata_colrYrZr[Zfiltro_metadataZ alias_maprRZ filtro_tsZcampos resultadosr@rrr"_obtener_datos_por_fecha_por_aliassB     rjhistorial_controlc CszD|d}|i}i}|D]"}|d}|dg}|||<q|WStk r}z td|d|jjW5d}~XYnXdS)u% Obtiene todos los logs de la colección 'historial_control' y los organiza en un diccionario. Las llaves del diccionario son las fechas (en formato 'YYYY-MM-DD') y los valores son listas de logs. :return: Diccionario con las fechas como llaves y listas de logs como valores. rkdatelogsz6Error al obtener el historial de 'historial_control':  N)r`ra Exceptionr)r __traceback__ tb_lineno) r-r. collection documentshistorial_dictrRrlrmerrrget_historial_control_as_dict(s    rvc Csztddd}tdddd}d}d}d }d }d }td||d}td||d}td||d} td||d} ttd d dtd d dtd d dtd d dd} Wn:tk r} ztd| jjd| W5d} ~ XYnX||||| | | fS)NTFFFFFF)bcolorsolide84e0e) fill_type start_color end_colorZ92CDDCZ92DC97ZFFFF00C8C8C8ZEE6b6Ethin000000) border_stylery)leftrighttopbottomzError obteniendo estilos (): )r r rrrorprq) fontfillZcelesteZverdeZamarilloZgrisZrojofill_iny fill_iny_manfill_vdf fill_alarmborderrurrr get_stylesFsF     *rCc Cszt|}t|D]}|ddkr$q|||}|dkrNtd|dq|dkrjtd|dqt|} | |} t|} d| _d | _| | | t d | qt d Wn>t k r} z td | j jd | dW5d} ~ XYnXdS)u Inserta repetidamente la imagen en la hoja, avanzando horizontalmente columnas que pueden exceder la 'Z' (soporta AA, AB, etc.). :param worksheet: Hoja de cálculo (openpyxl worksheet) :param img_path: Ruta al archivo de imagen :param start_column: Columna inicial (string), p.ej "C" :param start_row: Fila inicial (int) :param repeats: Cuántas veces quieres repetir la imagen :param step: Cuántas columnas saltar cada vez (int) r&i@zLa columna calculada (u/) excede el límite de Excel (XFD). Deteniendo.uÍndice de columna no válido (z ). Se ignora.iXizImagen insertada en uGMarca de agua agregada (con expansión de columnas) satisfactoriamente.z!Error insertando marcas de agua (r.N)r ranger)rrrExcelImgwidthheight add_imagerrrorprq) worksheetimg_path start_column start_rowrepeatsstep start_col_numi col_index new_col_str new_positionimg_copyrurrradd_markwater_imagests,      rc( s\ztjdddddddt|D]\}zt|d}|d}|d} t|dkst|dkst| dkrtd Wq&fd d |D} |} d | rd nd| rd nd} d } d}d\}}tj ||fdd}d\}}}||}||}| ||||g}|j ||ddd d| t |t||dd|jddd|jdddd|jddd|\}}| rt| }t|d krt tt|nd}|j||gt|||dddd |||| rXt| }t|d kr&t tt|nd}|j||gt|||d!dd"d ||||}|j || d#d$d%d d&| t |t||dt| d'|jd#d$dd|jdd$d|jtjd(d)|jtd*||jd+dd| d,|j!|d-|"D]}|#d.|$d/q|%\}}|%\} }!|| }"||!}#|j&|"|#| d0d1|j'd2t(}$|j)|$d3dd4t*||$+dt,|$}%d5\|%_-|%_.|d6d7}&|/|%|&Wq&t0k r}'zt1d8|'j2j3d9|'W5d}'~'XYq&Xq&Wn>t0k rV}'zt1d:|'j2j3d;|'W5d}'~'XYnXdS)s z)insert_matplot_graphs..rr&g333333?)g333333@d)figsizedpi)g?ffffff?rzNivel de O2 (mg/L)z#1e4882)labelry linewidth u Hora del día)fontsize)ryry)axis labelcolorgllF?z#92CDDCuInyección O2 - Automática)rrryalpharz#92DC97uInyección O2 Manualu Saturación %z#ab0202z--)rry linestyler <)intervalz%H:%MzSensor T)r-rz upper center)g?gɿ)ncollocbbox_to_anchorbbox_transformpng)formatr)ii$L_Graph2z!Error generando reporte en linea rnuError insertando gráficos (r)4pltrcParamsupdate enumeratemdatesdate2numlenr)rfigureadd_axesplotset_xlimminmaxset_ylim set_xlabel set_ylabel tick_paramsget_ylimnpdiffsortbartwinxxaxisset_major_locator MinuteLocatorset_major_formatter DateFormatter autofmt_xdate set_titlegridsubplots_adjustget_xticklabels set_rotationset_haget_legend_handles_labelslegend transAxesrsavefigcloseseekrrrrrorrprq)(rvalues_for_chartsrangos_iny_on_tsrrangos_iny_on_manual_tsRangosidxxo2so2 ts_inyecting ts_iny_manual total_graphsr bottom_marginZ ancho_figZalto_figfigrrrrrax1y_miny_maxx_injdxax2lbllines_1labels_1lines_2labels_2handleslabelsimg_dataimgZceldarurrrinsert_matplot_graphss   $                         rc. Cszt|D]r\}}t|||d}t|||d}t|||d}t|||d}t|||d}t|||d}t||||||||t|d}| |_| |_||t|d}||t|d}||t|d}||t|d}||t|d}||t|d} | |_| |_| |_| |_| |_| | _| |_| |_| |_| |_| |_| | _q tt|D]8}!|j|d|!||d|d|!|||dq|D]}|||d||d }"t|"D]\}#}$t|$D]r\}%}&| |&_ |$d j ||k}'|$d j | |ko<|'}(|$d j | |k})|(r^||&_n |'rj||&_|)r||&_q|$d j }*t|$d j d |#d |%d|dd|*rd|*kr|#dkr||&_qq|j d||ddD]$}$|$D]}&t ddddd|&_ qq|jD]v}!d }+|!D]B}&z.|&j dk rTtt|&j |+krTt|&j }+Wn YnXq$|+d},|,dkr|,|j|!d j_qWn>tk r}-ztd|-jjd|-W5d}-~-XYnXdS)Nr&rr)rrend_row end_column celdaInicial celdaFinalrz row: z col: z offset: rnA)min_rowmax_rowcenterTF) horizontalvertical shrink_to_fit wrap_textz"Error aplicando estilos a celdas (r)rrrrdrrrr merge_cellsrvalue iter_rowsr alignmentcolumnscolumn_dimensions column_letterrror)rrprq).r bomb_namesrcant_keys_tabla tablas_disth_offsetv_offset max_columnrangos_bomba_on rangos_iny_onrangos_iny_on_manualrrrrrrrrchannelLetra1Letra2Letra3ZLetra4ZLetra5ZLetra6 cell_mergecell_1cell_2cell_3Zcell_4Zcell_5Zcell_6column cell_rangerrowjcell bomb_activeZactive_inyectionZ iny_manualZalarm_cell_value max_lengthadjusted_widthrurrrapply_cell_stylesTs      *    r@)r#)r#r2)r#r2)rk)rrrr)-syspathinsert collectionsrrrpathlibropenpyxlopenpyxl.utilsropenpyxl.stylesrrr r r r openpyxl.drawing.imager rnumpyrior matplotlibusematplotlib.pyplotpyplotrmatplotlib.datesdatesrrr)r"r1r^rjrvrrrr@rrrrs:           @ . 4-