o
    i)                     @   s   d dl Z d dlZd dlZd dlm  mZ d dlZd dlZd dlT d dl	T d dl
T d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT dd Zdd Zdd Zze e_W n   Y d	Zd
ZdS )    N)*c                 C   sh  zt j|   W n	   i t j| < Y z	t j|  |  W n   |i it j| < Y | dkrW|dkr2t|S |dkr;t||S |dkrDt||S |dkrMt||S |dkrVt||S n| dkre|dkrdt||S n| d	kr|d
krrt||S |dkr{t	||S |dkrt
||S |dkrt||S |dkrt||S |dkrt||S |dkrt||S |dkrt	||S |dkrt||S nn| dkr|dkrt||S n`| dkrt| ||S | dkr|dkrt||ddS nF| dkr|dkrt||ddS |dkrt||ddS n+| dkrt||S | dkr|dkrt||S n| dkr)|dkr)t||ddS td| t| ||S ) Nsevtsjfesl
norrbottenatwldedbkvvrmv	db_trainsdb2drbevrruklononlns)countrydkkbchfrpafihslzTRYING GTFS: )vsitesZ	vt_lookupZ	sj_lookupZferries_lookupZ	sl_lookupZresrobot_lookupZ	wl_lookupZ	db_lookupZ
kvv_lookupZ
rmv_lookupZdb_trains_lookupZ
db2_lookupZ	dr_lookupZ	be_lookupZ
vrr_lookupZ	lo_lookupZ	no_lookupZ	ns_lookupZ	kb_lookupZch_zu_lookupZ	pa_lookupZ
hsl_lookupprintgtfs_lookup)r   operatorstation r'   `C:\Users\computer\Desktop\notes\tskylt4.0\server\20260128 - new GTFS loader\traffic_operators.pysearch   sN   




r)   c              
   C   s   zt j|   W n	   i t j| < Y z	t j|  |  W n   ddit j|  |< t| | Y z	t|| |}W n tyQ } zt| ddiW  Y d }~S d }~ww t|d |dkrl|D ]}d||  ||< t| q]|S )Nz	Try again zDATA!!!ZsncbS)r!   cache
gtfsloaderload_static_gtfsr)   	Exceptionr#   lowerstrr"   Zsort_dict_by_keys)r   r%   r&   dataeresultcZsearch_listZstopsnameidr'   r'   r(   r$   >   s$   

r$   c              
   C   s  t d ztj|   W n	   i tj| < Y z	tj|  |  W n   |g dtt ditj|  |< t| | Y t	|| |}|dkri|D ] }t | |d d dkr\d|d d< |d d	 
 |d d	< qHt|r|tt d
tj|  | |< |S d S )NzLoading GTFS...z
Loading...)
departuresmessage	timestamprolinetransport_mode   METROr7   )r8   r:   )r#   r!   r,   timemktime	localtime	operatorsr-   r.   	departureupperlenZ	gtfs_mainZ
start_gtfsr/   replacedatetimenowr1   date	timedeltaappend)r   r&   r%   resultsrD   r2   r3   keyZcounterr8   Z_typedestinationr<   _timeZ_dateZ
new_recordr'   r'   r(   gtfs_load_departuresx   s"   $"rQ   ZMSoM0Zr2or225Nl2NFw7xL3zMToaZiHYI8_4fXoyd7VHQf2L28QcGX9Ma)r@   rH   requestsxml.etree.ElementTreeetreeElementTreeETr!   r-   Zcountries.swedenZcountries.netherlandsZcountries.denmarkZcountries.finlandZcountries.germanyZcountries.norwayZcountries.switzerlandZcountries.franceZcountries.ukZcountries.austriar)   r$   rQ   Zsj_load_stationsZsj_stationsZ	vt_clientZ	vt_secretr'   r'   r'   r(   <module>   s,   ,:b