eO9bddlmZmZddlZddlZddlZddlmZddl Z ddl Z ddl m Z ddl m Z mZmZmZmZmZddlmZddlmZddlmZmZdd lmZeGd d ZeGd d eZeGddeZeGddeZ eGddZ!dZ"GddZ#eGddZ$GddeZ%eGddZ&eGddZ'eGdd Z(eGd!d"Z)eGd#d$Z*eGd%d&Z+eGd'd(Z,eGd)d*Z-eGd+d,Z.dS)-) dataclassfieldN)Enum)time)DictListOptional NamedTupleUnionLiteral)Path)wcswidth)loaddump)colorizec8eZdZUdZeed<eed<eed<dZdS)Locationz Generic location namelatitude longitudec|jSN)rselfs 5/home/salvo/dev/vasttrafik-cli/vasttrafik/__init__.py__str__zLocation.__str__,s yN)__name__ __module__ __qualname____doc__str__annotations__floatrrrrr#sO IIIOOOrrc4eZdZUdZeded<eed<dS)Stopz5 The object represents a stop or a position. )stoparea metastation stoppoint locationTypegidN)rrr r!r r#r"r%rrr'r'0s7@AAAA HHHHHrr'c&eZdZUeded<dS)PointOfInterestpointofinterestr+Nrrr r r#r%rrr.r.8s$+,,,,,,rr.c&eZdZUeded<dS)Addressaddressr+Nr0r%rrr2r2<s#)$$$$$$rr2c0eZdZUeed<eed<defdZdS)Token expires_in access_tokenreturnc0|jtkSr)r6rrs rexpiredz Token.expiredFs''rN)rrr intr#r"boolr:r%rrr5r5AsFOOO(((((((rr5cl t||S#tdt|xYw)NzFAILED TO PARSE DATA)rprint)datatype_s r _reportloadrAJs=D%    $%%% d  s!3c eZdZdededdfdZdefdZdefdZded e eeffd Z d ede e e zezfd Zdde dedzdedejdzdedf dZdedededdejdzde df dZdS) Vasttrafikkey tokenfiler8Nc0||_||_d|_dS)z key is the API key that must be sent on every request to obtain a reply. you can obtain one at api.vasttrafik.se, but it will be activated the night after registration. N)rD _tokenfile_token)rrDrEs r__init__zVasttrafik.__init__Us #'+ rcL|jw|jr^|jd5}t t j|t |_dddn #1swxYwY|jr |jrd|_|jr||_|jd5}t j t|j|dddn #1swxYwY|jS)Nrtwt) rHrGexistsopenrjsonr5r: _renew_tokenr)rfs r _get_tokenzVasttrafik._get_token_sm ; 4?#9#9#;#; %%d++ 8q"49Q<<77  8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 ; 4;..00 DK ; ++--DK%%d++ 0q $t{++Q/// 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0{s#-A44A8;A8 (DDDcd}tj|}d|_d|jz|jd<tj|5}ttj|t}dddn #1swxYwY|xj ttz c_ |S)Nz#https://ext-api.vasttrafik.se/tokensgrant_type=client_credentialszBasic Authorization) urllibrequestRequestr?rDheadersurlopenrrOr5r6r;r)rurlreqrQrs rrPzVasttrafik._renew_tokenps3n$$S))3'/$(': O$ ^ # #C ( ( *ATYq\\5))A * * * * * * * * * * * * * * * DFF # s(BBBserviceparamscv|j}d|dtj|}tj|}d|z|jd<d|jd<d}tj|5} | }t|d krn||z }. dddn #1swxYwY| d rtd tj|d S) Nz$https://ext-api.vasttrafik.se/pr/v4/?zBearer rTz text/plainacceptrTrsInvalid authKeyzInvalid authKeyutf8)rRr7rUparse urlencoderVrWrXrYreadlenstrip startswith ExceptionrOloadsdecode) rr]r^tokenrZr[r\rQls r_requestzVasttrafik._requestzs\!!._W__v|?U?UV\?]?]__n$$S))'05'8 O$ , H  ^ # #C ( ( A FFHHq66Q;;Q                   7799   2 3 3 /-.. .z!((6**+++s0CCC user_inputc|d|dd}t|dtttzt zS)z>Returns a list of Stop objects, completing from the user inputzlocations/by-text10)qlimitresults)rnrAlistr'r.r2)rroas rlocationzVasttrafik.locationsA MM-Z/N/N O O1Y<d_.Dw.N)OPPPrstop time_span departures datetime_objGroupedBoardItemcdi}t||d<d|d<|r||d<|rt||d<|d|jd|}t |dt t }d |D}t|S) z-Returns a departure board for a given station maxDeparturesPerLineAndDirection30rs startDateTimetimeSpanInMinutesz stop-areas/z /departuresrtc g|] }|j | Sr%) cancelled.0is r z$Vasttrafik.board..s???A1;?a???r) r" isoformatrnr,rr BoardItemr}group)rryrzr{r|r^b boarditemss rboardzVasttrafik.boards"$58__12w  ?&2&<&<&>&>F? #  9*-i..F& ' MM====v F F!I,Y88 @???? %%j111rorigindestdatetime_relates)arrival departureTripci}d|d<d|d<d|d<d|d<t|dr |j|d<n.t|j|d <t|j|d <t|dr |j|d <n.t|j|d <t|j|d <|r||d<||d<|d|}t|dtt}d|DS)NrqrsfalseonlyDirectConnectionsincludeNearbyStopAreasincludeOccupancyr, originGidoriginLatitudeoriginLongitudedestinationGiddestinationLatitudedestinationLongitudedateTimeRelatesTodateTimejourneysrtc.g|]}|j |j|Sr%)rallLegsrs rrz#Vasttrafik.trip..s&BBBa B BBBBr) hasattrr,r"rrrrnrArur)rrrrr|r^rtripss rtripzVasttrafik.trips8w*1&'+2'(%,!" 65 ! ! >"(*F;  '*6?';';F# $(+F,<(=(=F$ % 4   A'+xF# $ $,/ ,>,>F( )-0-@-@F) *  :*:F& '!-!7!7!9!9F:  MM*f - -AiL$t*55CB5BBBBr)NrxN)rrr r"r rIr5rRrPdictrnrur'r.r2rwr;datetimerrrr rr%rrrCrCSs,C,D,T,,,,E"e,,T#s(^,,,,(Q3Q40F0P+QQQQQ 22$2#d(2C2YaYjmqYq2{AS|T2222,#C#C#C&&<= #C #+d2 #C f #C#C#C#C#C#CrrCc@eZdZUeed<eded<eed<dZdS)Notetype)unknownlownormalhighseveritytextc|jdkrt|jdS|jdkrt|jdS|jS)Nr)ansir)rrrrs rtoTermz Note.toTermsN =F " "DIA... . =H $ $DIA... .yrN)rrr r"r#r rr%rrrrsI III89999 IIIrrcReZdZdZdZdZdZdZdZdZ dZ d Z d Z e d efd Zd S) VehicleTypetrambustrainferrytaxiwalkrnonebikecarr8c|jd|jd|jd|jd|jd|jd|jd|jd|jd|j d i }||S) Nu🚅u🚋u🚌u⛴u🚕u 🚶‍♂️u🚴u🚗) TRAINTRAMBUSFERRYTAXIWALKUNKNOWNNONEBIKECAR)rss rsymbolzVehicleType.symbols\ J Iv Hf J Iv I L" Ir Iv Hf wrN)rrr rrrrrrrrrrpropertyr"rr%rrrrsk D C E E D DG D D C     X   rrcLeZdZUeed<eed<eed<eed<eed<eed<dS)Liner shortNamebackgroundColorforegroundColor borderColor transportModeN)rrr r"r#rr%rrrrsT IIINNNrrc$eZdZUeed<eed<dS)Journey directionlineN)rrr r"r#rr%rrrrs"NNN JJJJJrrc2eZdZUeed<eed<dZeed<dS) StopPointrr,rplatformN)rrr r"r#rr%rrrr s4 III HHHHcrrcneZdZUdZejed<eed<eed<eed<eed<dZ ejdzed<dS) rzm This represents one item of the panel at a stop has a bunch of attributes to represent the stop plannedTime isCancelledisPartCancelledserviceJourney stopPointN estimatedTime) rrr r!rr#r<rrrr%rrrrsu"""".2M8$t+22222rrceZdZUeeed<edeededfdZede fdZ dZ d de fdZ de eee ffd Zdd Zd S)r}ritemsr8cp|dt|dgg}|D]w}d}|D]K}|jd}|j|jkr,|j|jkrd}|j|L|s#|t|gx|S)Nc|jjSr)rrrs rz(GroupedBoardItem.group..&s !5rrDrFT)sortr}poprrrappend)rr\raddedr firstitems rrzGroupedBoardItem.group"s  55 666 uyy||n - - . 0 0AE / /!,Q/ ;)"555!:JiNf:f:f E$++A... 0)1#..///rc0|jdjjS)Nr)rrrrs rtrackzGroupedBoardItem.track8sq!+44rc.|dS)NF)colorrrs rtoTxtzGroupedBoardItem.toTxt<s{{{'''rTc ||}|}d|d|dd|dd|jdjjdt d|fzS)a Returns a string representing the BoardItem colorized using terminal escape codes. Servertime must be retrieved from the Vasttrafik class, and indicates the time on the server, it will be used to show the difference in minutes before the arrival. z%s %0*d%s -> %s # %srxrr,c>t|d|dzS)Nrr)r"rs rrz)GroupedBoardItem.toTerm..IsJPSTUVWTXPYPY\]^_\`P`r)getNamer{rrrjoinmap)rrrdeltas rrzGroupedBoardItem.toTerm?sll5!!!!%aq!eAhqk4?[\K]KlKvx{yAyABEF`F`bgBhBhyiyi)jj jrcPd|jD}||S)Ncg|]Y}|jr|jn|jtjtjjz jdz|jrdndfZS)<👻r)rrrnowtimezoneutcsecondsrrs rrz/GroupedBoardItem.departures..LsMMMxyJQ__Q]hN_NcNcdldudyNzNzzDHJJXYXiNqNTNToqsMMMr)rr)rrs rr{zGroupedBoardItem.departuresKsBMM}A}LMMM  rFc|jdjjjdz}||jdjjjjz }t |dkrd|z}t |dk|s|Std|jdjjjddzd}td|jdjjj ddzd}t|||S) zReturns a nice version of the name If color is true, then 256-color escapes will be added to give the name the color of the liner 0xrNbg) rrrrrrulenr;rrr)rrrfgcolorbgcolors rrzGroupedBoardItem.getNamePsq!05?#E "16DKK4jj2oo:D4jj2oo KdT_Q/>CSTUTVTVWWY[\\dT_Q/>CSTUTVTVWWY[\\g'2222rN)T)F)rrr rurr# staticmethodrrr"rrrrtupler;r{rr%rrr}r}sYT)_.@)A\*5s555X5((( j jC j j j jDsCx1 333333rr}ceZdZUeed<dS)LegHalfrN)rrr rr#r%rrrrcsrrceZdZUeed<eed<eed<ejed<ejed<eed<dZejdzed<dZ ejdzed <dZ edzed <e d e fd Z d efd ZdS) ConnectionLegr destinationrplannedDepartureTimeplannedArrivalTimeplannedDurationInMinutesNestimatedDepartureTimeestimatedArrivalTimeestimatedDurationInMinutesr8ct|jjdo=t|jjdo#|jjj|jjjkS)Nr,)rrrrrrs rhiddenzConnectionLeg.hiddenyslt{,e44fAQA[]b9c9cfhlhsh}iCGKGWGaGfif frc t|jr|jn|jdd}t|jr|jn|jdd}d|jjz}t|dkrd|z}t|dk|d|d|d|j j j d|j j j }|S)Nzgo !r   -> ) r"rrrrrrrrrrrr)rrrrr\s rrzConnectionLeg.toTerm}s8Sr44Y]Yrxxzz{{|~}~|~ D4Mjt00SWSjpprrsstvuvtvwt)004jj2oo:D4jj2oo o oi o o7 o odk.C.H o odN^NhNm o or)rrr rr#rrr;rrrrr<r r"rr%rrrrms OOO"++++ ))))!!!!7;H-4;;;59(+d2999-1d 111 ffffXf       rrc&eZdZUeed<eed<eed<eed<eed<ejed<ejed<eed<d Z ejd zed <d Z ejd zed <d Z ed zed <e Z e ed fed<edefdZdefdZdefdZd S)LegrrrrrrrrNrrr.notesr8cdS)NFr%rs rr z Leg.hiddensurc |}t|jr|jn|jdd}t|jr|jn|jdd}|d|d|d|jjj d|j jj }|j r*|dd d|j Dzz }|S)Nr"r r$r% c3>K|]}|VdSrrrs r zLeg.toTerm..s*!A!A!((**!A!A!A!A!A!Ar) rr"rrrrrrrrrr(r)rrrrr\s rrz Leg.toTerms||~~8Sr44Y]Yrxxzz{{|~}~|~ D4Mjt00SWSjpprrsstvuvtvw o oi o o7 o odk.C.H o odN^NhNm o o : B  !A!Adj!A!A!AAAA AArc|jjjdz}||jjdzz }|jrd|z}t |dkr |dddz}||jjjjz }t|dkrd|z}t|dktd|jjj ddzd}td|jjj ddzd}t|||S) Nr rr#r rr r ) rrrrrrfrrrr;rrr)rrrrs rrz Leg.getNames"'1C7 #-33   !D=D t99r>>9s?D #(6==4jj2oo:D4jj2oodT05EabbII2NNdT05EabbII2NNg'2222r)rrr rr#rr<rr;rrrrr(rrr r"rrr%rrr'r's: OOO"++++ ))))!!!!7;H-4;;;59(+d2999-1d 111#eggE5s %%% X3333333rr'ceZdZUeeZeeed<dZe ed<eeZ ee ed<e de fdZ e deee zfdZdefd Ze defd Zd S) r)default_factorytripLegsF isDepartedconnectionLinksr8c.|jD] }|jrdS dS)NTF)r2r)rrs rrzTrip.cancelleds.  A} tt urcR|j|jz}|d|S)Nc|jSr)rrs rrzTrip.allLegs..s Q3rr)r2r4r)rr\s rrz Trip.allLegss.'+}t7K'K 33444rcJdd|jDS)Nr+c3TK|]#}|jdk |V$dS)FN)r rrs rr-zTrip.toTerm..s6OOQX=N=N=N=N=N=NOOr)rrrs rrz Trip.toTerms%yyOOT\OOOOOOrc |jsdS|jdjr|jdjn|jdj}|jdjr|jdjn|jdj}||z }|jdzS)Nrr)r2rrrrr)rstartendrs rdurationz Trip.durations} 1;?=;K;bN a 77hlhuvwhxiN8< b8I8^IdmB44dhdqrtdueIe }""rN)rrr rrur2r'r#r3r<r4rrrrr"rr;r>r%rrrrs%555Hd3i555J+05+F+F+FOT-(FFF 4X cM12X PPPPP#####X###rr)/ dataclassesrrurllib.requestrU urllib.parserenumrrOrertypingrrr r r r pathlibr wcwidthrr typedloadrr xtermcolorrrr'r.r2r5rArCrrrrrrr}rrr'rr%rrrIs&)((((((( CCCCCCCCCCCCCCCC$$$$$$                 8     -----h-- - %%%%%h%% % ((((((( (zCzCzCzCzCzCzCzCz          $:         3 3 3 3 3 3 3  3 A3A3A3A3A3A3A3 A3H    4 -3-3-3-3-3-3-3 -3` ####### ###r