U ^j9@sddlmZmZddlmZmZmZddlmZddl m Z dZ dZ dZ dZd d d d gZd dZeeeedddZeedddZeedddZeedddZeedddZeeeeddd Zed!d"d#Zed$d%d&Zeedd'd(Zeed)d*d+Zd,d-Zd.d/Z eed0d1d2Z!eed3d4d5Z"eeed6d7d8Z#eeeed9d:d;Z$eeeed<d=d>Z%eeed?d@dAZ&dBS)C)Flaskjsonify) JWTManager jwt_requiredget_jwt_identity) MongoClient) defaultdictusersENVIRO NANOOX__TSScappDBenvironanooxpssscappcCsdtkrtdt_tjS)u-Obtiene una conexión de MongoDB por request. db_clientzmongodb://localhost:27017/)grrrr=/var/www/itgmanager/itgmanager/modules/consolidado_estados.pyget_db s r) ciclo_info user_statusservicioreturncCst|o||iddS)z_ Intersecta el estado de permiso del usuario con el estado del ciclo para un servicio. statusF)boolget)rrrrrrintersect_service_statussr)usernamercCs6|d|i}|s*tjd|diS|diS)uT Obtiene los permisos crudos (raw) del usuario desde la colección 'users'. rz Usuario 'z' no encontrado.permisos)find_oneapploggerwarningr)users_collectionrusuariorrrget_accesos_usuarios r&)permisos_usuariosrc Cs|tj}i}|diD]\}}i||<|D]\}}i|||<|D]}|d|i} | szg||||<qNi} tD]<} |d|||| idd} dt| | | i| | <q| ||||<qNq6q|S)un Cruza los permisos del usuario con los datos de la colección 'ciclos', retornando permisos activos. r centro_idrF)user_database_nameciclosritemsr lower SERVICE_KEYSr) clientr'r* resultadoregionareasareacentroscentroinfonodosvcrrrrprocesar_permisos&s.  r8) permisos_procrcCs0i}|D]\}}dd|D||<q |S)zw Transforma {R:{A:{C_ids: [...]}}} en {R:{A:{clave:{'_id':orig}}}}. clave = texto antes de '-' en cada ID. cSs i|]\}}|dd|DqS)cSs"i|]}|dddd|iqS)-r_id)split).0r(rrr Isz6transformar_diccionario...r)r>r2r3rrrr?Hs z+transformar_diccionario..)r+)r9outr0r1rrrtransformar_diccionarioAs  rA)datarcCsJg}|D]8}|D]*}|D]}|d}|r$||q$qq |S)zP Extrae lista de todos los valores '_id' en la estructura transformada. r<)valuesrappend)rBidsr1r3r6r<rrr obtener_idsRs    rF)rr(rrcCs|tj}dd|iiddddiiiddidddd iiidd idd dd iiidd iddddididdid||diddddig }t||}|r|ddSdS)u\ Consulta la colección 'users' con pipeline para extraer el status de un servicio. z$matchrz$projectrz$objectToArrayz $permisosz$unwindciudadesz $permisos.vz $ciudadesr3z $ciudades.vz$centrosz $centros.kz $centros.v)r( serviciosz $servicios)r(z servicios.krz$servicios.v.status)r<rrF)r)r list aggregate)r.rr(rr pipelineresrrrstatus_servicio_permiso_usuario`s     rM)r(cCs0|tj}|d|iddi}|r&|SdddS)uQ Retorna el estado de 'enviro' y 'nanoox' desde la colección 'estados'. r(r<rB)r r)r)estadosr )r.r(coldocrrrobtener_estado_servicio_centrovs rR)rcCs8||g}tdd|DrdStdd|Dr4dSdS)uh Obtiene el estado resultante del centro según el estado de Enviro y Nanoox ('C' > 'A' > 'B'). css|]}|dkVqdS)CNrr>errr sz estado_centro..rScss|]}|dkVqdS)ANrrTrrrrVsrWrN)any) estado_enviro estado_nanooxrOrrr estado_centros r[cCs`tt}|D]F\}}|D]4\}}|d|}|D]}d|||d<q>q qt|S)zA Agrupa centros bajo clave 'region-area' con valor None. r:Nr<)rdictr+rC)rBr@r0r1r2r3keyr6rrragrupar_por_region_areas r^)r.r(cCsn|tj}|d|iddd}|r0d|kr:dddS|d}|d}||i}|d |d dS) z Retorna {'lat': ..., 'lon': ...} desde db users coleccion ciclos inspeccionando gps_coordinates.active ('manual' o 'auto'). r(rr;)r<gps_coordinatesr_N)latlonactiver`ra)r)r*r r,r)r.r(r*rQgpsZactivosubrrrobtener_coordenadass      recCst|tr\g}|D],\}}t|}|dkr:||q|||<q|D] }||=qH|pZdSt|trdd|D}dd|D}|pdS|S)u~ Elimina recursivamente claves con valor None o listas vacías. Devuelve None si el objeto completo queda vacío. NcSsg|] }t|qSr) clean_noner>xrrr szclean_none..cSsg|]}|dk r|qS)Nrrgrrrris) isinstancer\r+rfrDrI)rBZto_delkvcvcleanedrrrrfs    rfcCs d|kr dSd|krdSdSdS)u A partir de una lista de estados retorna el valor predominante(C>A>B). Si existe al menos una C, retornará C; de existir al menos una A sin existencia de C, retornará A. De no existir, retornará B rSrWrNNr)valorrrractualizar_valors rp)consolidado_por_centrorcCsi}|p iD]r\}}g}|D]F}t|trJd|krJ||dq$t|tr`||q$|dq$d|}t|||<q|S)z Concatena y colapsa por 'R-A', luego aplica actualizar_valor. Extrae el campo 'estado' cuando el valor es un dict. estadorN)r+rCrjr\rDstrjoinrp)rqr@rar3rOitemZ concatenatedrrrestado_resultante_area_region s     rx)area_region_statesrcCsTtt}|piD],\}}|ddd}|||p:d7<qdd|DS)u Agrupa por región (parte antes del guión), concatena los valores resultantes de estado por área y aplica actualizar_valor. r:r;rrNcSsi|]\}}|t|qSr)rp)r>rrlrrrr?(sz,estado_resultante_region..)rrtr+r=)ryZaggrvvalr0rrrestado_resultante_regions r|)rpagercCs(t||}|dkr|St|d|iS)uX Retorna permisos de usuario procesados según PAGE: raw o cruzados con ciclos. HOMEr)r&upperr8)r. users_collrr}rawrrrfetch_effective_permissions6s  r)r}rBrrc Cs|tj}i}t|D]r}|dkp6t||||}|s>qd}d} |dkr`t||d}|dkrzt||d} t|| ||<q|S)uF Calcula estado actual de cada centro según PAGE y permisos. r~N)r~r r )r~NANOOXr)r)r*rFrrMr,rRr[) r.r}rBrr*nuevosr(Z permitidoZ state_envZ state_nanrrrcompute_new_values@s      r)grupo dict_estadosr.rcCsD|D]6\}}t|D]$}||kr||t||d||<qq|S)zT Inserta estado y coordenadas de cada centro en el diccionario de permisos. )rrZ coordenadas)r+rIre)rrr.rvr3r(rrr$agregar_estado_centros_a_consolidado_s r)r.rr}c Csztd|tj}t||||}t|}t||||}ttt|||}td|dt |pdi}td|dt |pi} td| d||| dWSt k r} z"td| j j d| WYd Sd } ~ XYnXd S) u Orquesta todo el flujo del consolidado: 1. Permisos → 2. Transformar → 3. Valores → 4. Merge+coords → 5. Jerarquías (áreas y regiones) zEN CONSOLIDADO EMPRESASzcentros=z zareas=z regiones=)r3r1regionesu/Error obteniendo consolidado empresa en línea z: N)printr)r rrArrfrr^rxr| Exception __traceback__ tb_lineno) r.rr}rrrBrr3r1rrUrrrconsolidado_empresa|s,   rN)'flaskrrflask_jwt_extendedrrrpymongor collectionsrr)Zenviro_database_nameZnanoox_database_nameZscapp_database_namer-rr\rrtrr&r8rArIrFrMrRr[r^rerfrprxr|rrrrrrrrs6       1(