>^j>4TdZddlmZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZdd lmZmZmZmZdd lmZmZmZmZmZdd lmZdd lmZmZdd lm Z ddl!m"Z"ddl#m$Z$m%Z%ddl&m'Z'e Z(ejRe*Z+ejXjZejXj\ejXj^ejXj`ejXjbejXjdfZ3ddZ4ddZ5dd ddZ6ddZ7ddZ8dddZ9y)zE Bidding Rate Distribution and Sequential/Broadcast Dispatch Engine. ) annotationsN)Decimal)settings)get_user_model)models transaction)timezone)PlatformSettingsTransporterProfileUserRoleVehicle)Bid RateRequestShipmentTripZone)vehicle_matches_required_type)normalize_country_code!transporter_load_search_radius_km)country_code_for_point) notify_users)get_latest_device_positionget_latest_positions_map)record_trip_statuscddl}d}|j||z }|j||z }|j|dz dz|j|j||j|j|z|j|dz dzzz}d|j |j ||j d|z z} || zS)Nrg@)mathradianssincosatan2sqrt) lat1lon1lat2lon2r radius_kmdlatdlonacs 0/var/www/html/truck-me/core/rate_distribution.py _haversine_kmr.'sI <<t $D <<t $D a ((4<<% &$,,t2D)E EQUXYQYHZ^_H_ _ ` DJJtyy|TYYq1u%5 66A q=c |j |jgS t|j}t|j}t |j xst||xsd} t}ttjjtjddttjjtdjd dz}g}t jjdt"j$j&d j)| j+d d }|D]}|j,} t/|j0xsdj3} | s:|j5| } | s t7| } | s\ t| d} t| d} t| | }|s|j:r|j:r||k7r,|j<r|j<s |j:s||k7rt?|\}}tA||| | }||kDrtBjjtEjF| tEjF| zdd}d}|D]}tI||jJs|}n|sg|jM| ||tO|d| | d|jQd|S#ttf$rgcYSwxYw#t$ri}YwxYw#t$rd} YwxYw#ttt8f$rYwxYw)z{ Find all available, verified drivers compatible with this shipment, ranked by ascending distance from pickup. N) status__intransporter_idTflatF)r2assigned_driver__isnullassigned_driver_id)user__is_activeuser__role__roledocuments_verified) user_id__inuser user__rolelatitude longitude)owner)assigned_driver) is_verified is_activer)driverprofilevehicle distance_kmlatlonc |dS)NrG)xs r-z4find_eligible_drivers_for_shipment..s 1]+;r/)key)) pickup_lat pickup_lonfloat TypeError ValueErrorrpickup_country_coderr ExceptionsetrobjectsfilterACTIVE_TRIP_STATUSES values_listr r Role TRANSPORTERexcludeselect_relatedr<strtc_idstripgetrKeyErrorlocalcountry_to_countryrr.r rQrvehicle_type_requiredappendroundsort)shipmentrOrPshipment_country positions_mapbusy_driver_idseligible_candidatesprofilesrEr<r`pos driver_lat driver_londriver_countryr(_dist_km user_vehiclesmatching_vehiclevs r-"find_eligible_drivers_for_shipmentrz5s] "h&9&9&A 8../ 8../ .$$\(>z:(V\Z\02  ';< %D 1  ';UZ[ ) 5 O "")) %]]66# * _ -  - ? ||W]](b)002 & 07  s:/Js;/0J0 JG  ==>>^7G%G  ' '..x~~!119H 1 J JO Y   .. HH4 688D#A A/   A,Q0N0NO#$    ""' !, $  q? B!;< G z "   D  :x0   sG*K5. L " L1L05L L  LL L-,L-0MM batch_numberc|jtjjk7r-tj d|j |jgStj}tdtt|ddxsd}tjj|}t|j!dd}|0|j#t%j&d d }|xsd dz}t)|}|Dcgc]}|d j*|vs|} }| s"tj d |j gS| d|} g} t-j.5| D]} tjj1|| d |tjj2t5t7| d|j8|j:xsdd\} }|sy| j=|  ddd| D]} t?| j@gdt7| j*t7|j*|jBxst7|j*t7| jDxsd| j:t7| jF|jH|jJ|jLd dd|jLd| jDxsdd| j:dtj d|tS| |j | Scc}w#1swY.xYw#tN$r/tjQd| j@| j*YfwxYw)z Dispatch rate requests to the next batch of eligible drivers based on PlatformSettings.bidding_rate_distribution_count. z8Cannot dispatch rate requests: shipment #%s status is %srbidding_rate_distribution_countrk driver_idTr4Nr|batch_number__maxrrDz1No remaining uncontacted drivers for shipment #%srGUSD)r|statusrG offered_pricecurrency)rkrDdefaults RATE_REQUESTr1) rate_request_id shipment_id unique_idrrr|pickup_addressdelivery_address cargo_typezNew Load Rate Requestz A new load (z") is requested for you with offer  .titlebodyz.Failed to notify driver %s for rate request %sz9Dispatched batch #%s of %s rate requests for shipment #%s)*rrStatus PUBLISHEDloggerinfopkr loadmaxintgetattrrrWrXrVrZ aggregaterMaxrzidratomic get_or_createPENDINGrr_suggested_pricerrhrrrrr|rrrrU exceptionlen)rkr|ps batch_sizeexisting_requestscontacted_driver_ids max_batcheligibler, uncontacted to_dispatchcreated_requestsitemrrcreateds r-#dispatch_rate_requests_for_shipmentrs)(//333 NPXP[P[]e]l]lm    BQGB(I1MRQRSTJ#++22H2E0<<[t<TU%// >0JKL_` !Q!+ 1(;H&U!H+..@T*T1UKU  GU kz*K*,    , ,D%--;;!H~$0)0088#*3tM/B+C#D%-%=%= ( 1 1 :U < KB ''+ ,, d d '*255z#&x{{#3!)!3!3!Gs8;;7G%()9)9)?R%@ " $'$8&.&=&=(0(A(A"*"5"5 .#H$7$7#88Z[][k[k[qoqZrrstvtttAABC d, KKC   kV,,F d   Mr||]_]b]b c ds2 L!L&A;L""L"C*L/"L,/4M'&M'c|jtjjk7rgStj j |tjjjrgStj j |jtjdd}| t|Stj j ||tjjj}|s1tjd||j t||dzSgS) z Check if the current batch is fully exhausted (all rejected/cancelled/expired). If so, automatically dispatch the next batch. rkrrr|r)rkr|rzVBatch #%s for shipment #%s fully resolved with no acceptance; advancing to next batch.rr{)rrrrrrWrXACCEPTEDexistsrrrrrrrr)rk latest_batchpending_in_batchs r-%advance_rate_requests_batch_if_neededrs  (//333 !!8K[,BUBUB[B[\\lmummnODLDODOmPPQR   K2727D QOP  Qs !K P23B;P>2P;>Q"!Q"c|jtjjk7r't d|jj dt j5tjj|_|xsdj|_ tj|_ |jgddddt|j |S#1swY xYw)zh Reject a rate request and advance to the next batch if all requests in the batch are resolved. rrr1)rrejection_reasonrrrN)rrrrrSrrrREJECTEDrarr rrrrrk)rreasons r-handle_rate_request_rejectrnsk008883L4G4G4M4M4O3PPQRSS    f)0099 )/2(<(<(> %$,LLN !(de f *,*?*?@ ffs #A&C((C1) r$rQr%rQr&rQr'rQreturnrQ)rkrrz list[dict])rkrr|z int | Nonerlist[RateRequest])rkrrr)rrrztuple[Trip, RateRequest])r1)rrrr_rr):__doc__ __future__rloggingdecimalr django.confrdjango.contrib.authr django.dbrr django.utilsr accounts.modelsr r r r core.modelsrrrrrcore.load_discoveryrcore.countriesrrcore.pricing.servicer api.notifyr api.traccarrrapi.trip_statusrUser getLogger__name__rrrEN_ROUTEARRIVED_PICKUPLOADED IN_TRANSITARRIVED_DELIVERYrYr.rzrrrrrKr/r-rs# .)!SS>>=8#L.   8 $ KKKKKKKKKKKK   nh $MMM M`BTnr/