o 7uh@s"UddlZddlZddlZddlmZddlmZmZddlm Z ddl m Z m Z ddl m Z ddlmZddlmZmZdd lmZdd lmZmZeed d d kZdae ed<iZddZddZ defddZ!defddZ"defddZ#defddZ$defddZ%ddZ&dS) N) load_dotenv) WebSocketWebSocketDisconnect)AIOKafkaProducer)datetime timedelta)connected_clients)database)ObjectIderrors)"list_nearby_devices_service_socket)NearbyDevicesRequestIotDeviceWithMappingList USE_KAFKAfalsetrueproducercstddatIdHdS)Nzlocalhost:9092)bootstrap_servers)rrstartrr8/var/www/html/moveengine/app/v1/sockets/tracking_data.pystart_kafka_producers rcstr tIdHdSdS)N)rstoprrrrstop_kafka_producersr websocketc s|IdHt}t|}|d}|d}|d}|d}|d}|d}|d} tdz |IdH} t| } t | d <td | | d } zt | } Wnt j yg|d IdHYq2w|d| i}|sz|dIdHq2i| |ddd}tj| td}t|iddd}d}|D]}|d|vr|}nqtdtd|td|td|r|d|di}||d|dd|dd d| |d|d|dd|d| | d dd }tr tr td!tj|tdd"IdHn||tr*tr*td#tj|tdd"IdHn||z~| d$}|r|| d%| d&| d'| d(d)t d*}| d+}| d,}|dt | id-|i|d.kr||r||dt |id-|in)|dkr|r| dt |id-|in|d/| id-|i| d/| id-|it| IdHWnty}ztd0| d1|WYd}~nd}~ww| d2r|d3|dd4IdHn|d5IdHq3tytd6YdSw)7N tracking_logs alert_logs iot_devicesalert_categoryalert_keywordsfleets workforcezwebsocket_tracking_data STARTEDT received_atzIncoming WebSocket Data: device_idu❌ Invalid device_id format_idzUnknown device account_idF)r%is_alertdefault)keyword category_idr*z=== ALERT DEBUG START ===z Payload text:zMatched keyword:z=== ALERT DEBUG END ===r+ category_namezAlert detected: )r+r,messager&unread)r#r%r+r,r- raw_payloadr"statusalertszutf-8vehicle_trackinglocationspeedheadingfuel timestamp) coordinatesr4r5r6r7) last_location last_updatedassociated_entity_typeassociated_entity_idz$setvehicledevicesu*⚠️ Failed to update last location for z: r&u*✅ Stored in tracking_logs + alert_logs ()u✅ Stored in tracking_logsu⚠️ WebSocket disconnected) acceptr get_mongo_dbnextprint receive_textjsonloadsrutcnowgetr bson_errors InvalidId send_textfind_onedumpsstrlowerlistfindupdaterrsendencode insert_one update_one update_manynotify_nearby_subscribers Exceptionr)r mongo_genmongorrr> categorieskeywordsr r!r-datar# device_obj_iddevicetracking_entry payload_text all_keywordsmatched_keywordkwcategory alert_docr8last_location_updateassociated_type associated_iderrrwebsocket_tracking_datas                      rlcsj|IdHt|tdttz tdIdHqty4t|tdttYdSw)Nu✅ Client connected. Total:Tr)u#❌ Client disconnected. Remaining:) r@raddrClenasynciosleeprremove)rrrrwebsocket_realtime_updates4s   rrcs|IdHz|IdH}t|}|d}|d}|r$|s6|dIdH|IdHWdSz t|d}t }Wnt y\|dIdH|IdHYWdSwt }t |}|d} | |||ddd d d d d d d d d d d d d d d d } td t| td | D]^} t| d| d| d| d| d| d| d| d| d| d| d| dd| d r| d ndd } td| |tj| tdIdHtdIdHq|dIdH|IdHzt |WWdStyYWdSwty&tdYdStyV} z$td t| |d!t| IdH|IdHWYd} ~ dSd} ~ ww)"Nr#datezMissing device_id or date.z%Y-%m-%d %H:%M:%Sz,Invalid date format. Use YYYY-MM-DD HH:MM:SSr)z$gtez$lt)r#r"r)) r$r#r3r4r5r6r"r%r&r+r,r-UUIDr"cursorcursorcursorcursorz************************r$r3r4r5r6rtr%r+r,r-r&F) idr#r3r4r5r6rtr%r+r,r-r&r"z$Historical record is alert checking:r'g?doneu⚠️ Client disconnected.u ❌ Error: zError: )r@rDrErFrHrKcloserstrptimerG ValueErrorr rArBrQsortrCrN isoformatrMrorp StopIterationrrY)rquery query_datar# datetime_str start_dateend_daterZr[ collectioncursordocresponserkrrrwebsocket_historical_routeFs         rc s|IdHz?|IdH}t|}|d}|s.|dIdH|IdHWdS|t|<td|dt t t dIdHq?t yg|tvrdt|=td|dt tYdSYdSw) N workforce_idzMissing workforce_id.u✅ Workforce z connected. Total clients: Tr)u❌ Workforce z disconnected. Remaining: ) r@rDrErFrHrKrxrrCrnrorpr)rmsgr^rrrrwebsocket_workforce_scheduless*   rcs|IdHt}t|}zH|IdH}t|}tdi|}|j|j |j |dt |<t |j|j |j |}t di|}|tj|tdIdH |IdHqQtyptd|t vrmt |=YdSYdSw)N)latitude longitude radius_kmdbr'Tu"❌ Nearby devices WS disconnectedr)r@r rArBrDrErFr rrractive_nearby_subscribersr rrKrMdictrNrrC)rrZr init_messager^payloadresultrrrrwebsocket_nearby_devicess6    rc sttD]D\}}z't|d|d|d|d}tdi|}|tj|t dIdHWqt yK}z t d|WYd}~qd}~wwdS)Nrrrrr'u,⚠️ Failed sending update to subscriber: r) rPritemsr rrKrErMrrNrYrC)updated_devicewssubrrrkrrrrXs"rX)'rorEosdotenvrfastapirraiokafkarrrapp.utils.connectionsrapp.dbr bsonr r rI#app.v1.services.platform.iotdevicesr !app.v1.models.platform.iotdevicesr rgetenvrOrr__annotations__rrrrlrrrrrrXrrrrs:       }` %