a
    `                     @   s\   d dl mZ d dlmZ d dlmZ ddddZeee	e f eeef ddd	d
Z
dS )    )load_controls)same_first_constraints)CategoryN)returnc                 C   s   t  }t  }| D ]z}t| | }t|dkr.q|d }||vrHt ||< || | | j t|}||vrvt ||< || | | j qt||  t||  d S )Nr   )	dictr   lenlistappendnamestrsame_route_constraintsr   )
categoriescategories_by_firstcategories_by_routecategorycontrolsZfcZc_str r   L/home/drajnoha/PycharmProjects/pa026/source/data_parsing/oris_constraints.pyadd_oris_constraints   s     


r   )r   race_categoriesr   c                 C   sb   | D ]X}| | }|D ]F}||vr"q|| }|D ]*}||vr<q.||krFq.|| }| | q.qqd S )N)add_same_route_constraint)r   r   firstcategory_keyscategory_keyr   constr_category_keycategory_constrr   r   r   r      s    r   )source.data_parsing.load_orisr   *source.data_parsing.load_route_constraintsr   source.resourcesr   r   r   r   r   r   r   r   r   r   <module>   s   