o ïA¸h^ã@s®UddlZddlZddlmZmZddlmZddlmZmZddl m Z ddl m Z da eed<dd „Zd d „Zd efd d„Zd efdd„Zd efdd„Zd efdd„ZdS)éN)Ú WebSocketÚWebSocketDisconnect)ÚAIOKafkaProducer)ÚdatetimeÚ timedelta)Úconnected_clients)ÚdatabaseÚproducercÃstddat ¡IdHdS)Nzlocalhost:9092)Úbootstrap_servers)rr Ústart©r r ú8/var/www/html/moveengine/app/v1/sockets/tracking_data.pyÚstart_kafka_producer s€ rcÃstr t ¡IdHdSdS)N)r Ústopr r r r Ústop_kafka_producers€ÿrÚ websocketcÃs„| ¡IdHz/ | ¡IdH}t |¡}t ¡ ¡|d<tr/t dt  |¡  d¡¡IdH|  d¡IdHq t yAYdSw)NTÚ received_atÚvehicle_trackingzutf-8zMessage queued successfully.) ÚacceptÚ receive_textÚjsonÚloadsrÚutcnowÚ isoformatr ÚsendÚdumpsÚencodeÚ send_textr)rÚmessageÚdatar r r Úwebsocket_tracking_datas€ ú ÿr cÃsj| ¡IdHt |¡tdttƒƒz t d¡IdHqty4t |¡tdttƒƒYdSw)Nu✅ Client connected. Total:Téu#⌠Client disconnected. Remaining:) rrÚaddÚprintÚlenÚasyncioÚsleeprÚremove)rr r r Úwebsocket_realtime_updates"s€ ÿ  þr(c Ãs | ¡IdHz¾| ¡IdH}t |¡}| d¡}| d¡}|r#|s5| d¡IdH| ¡IdHWdSzt |d¡}|t dd}Wnt y^| d¡IdH| ¡IdHYWdSwt   ¡}t |ƒ}|d} |  |||d œd œddddd œ¡} td | ƒ| D] } ttj| td ƒ| tj| td ¡IdHt d¡IdHqƒ| d¡IdH| ¡IdHzt |ƒWWdStyÆYWdSwtyÔtdƒYdSty} z$tdt| ƒ›ƒ| dt| ƒ›¡IdH| ¡IdHWYd} ~ dSd} ~ ww)NÚ device_idÚdatezMissing device_id or date.z%Y-%m-%d %H:%M:%Sr!)Úhoursz,Invalid date format. Use YYYY-MM-DD HH:MM:SSÚ tracking_logs)z$gtez$lt)r)r)ÚlocationÚspeedÚheadingrzQuery returned:)Údefaultgš™™™™™É?ÚdonezClient disconnected.zError: )rrrrÚgetrÚcloserÚstrptimerÚ ValueErrorrÚ get_mongo_dbÚnextÚfindr#rÚstrr%r&Ú StopIterationrÚ Exception) rÚqueryÚ query_datar)Ú datetime_strÚ start_dateÚend_dateÚ mongo_genÚmongoÚ collectionÚcursorÚdocÚer r r Úwebsocket_historical_route.sd€     ýþþ ú  ÿ €ýrGc ÃsÐ| ¡IdHz?| ¡IdH}t |¡}| d¡}|s.| d¡IdH| ¡IdHWdS|t|<td|›dt tƒ›ƒ t   d¡IdHq?t yg|tvrdt|=td|›dt tƒ›ƒYdSYdSw) NÚ workforce_idzMissing workforce_id.u✅ Workforce z connected. Total clients: Tr!u⌠Workforce z disconnected. Remaining: ) rrrrr2rr3rr#r$r%r&r)rÚmsgrrHr r r Úwebsocket_workforce_schedulesks*€  ÿ þÿrJ)r%rÚfastapirrÚaiokafkarrrÚapp.utils.connectionsrÚapp.dbrr Ú__annotations__rrr r(rGrJr r r r Ús       =