import requests, urllib3
urllib3.disable_warnings()

URL = "https://mtr.ciapongi.szablix.pl/mtr/api/map/stations-and-routes?dimension=0"

def main():
    print("=== SKANOWANIE TYPÓW POJAZDÓW W MTR ===")
    try:
        r = requests.get(URL, verify=False, timeout=10)
        data = r.json().get('data', {})
        routes = data.get('routes', [])
        
        types_count = {}
        
        for route in routes:
            route_type = route.get('type', 'BRAK_TYPU')
            
            # Niektóre buildy MTR mają też specjalne nadpisania typów
            jme_type = route.get('jmeType', '')
            if jme_type:
                route_type = f"{route_type} (jmeType: {jme_type})"
                
            route_name = route.get('name', 'Nieznana').split('||')[0]
            if not route_name:
                route_name = route.get('number', 'Nieznana')
            
            if route_type not in types_count:
                types_count[route_type] = []
            
            types_count[route_type].append(route_name)
            
        print(f"Znaleziono {len(routes)} tras. Oto podział na typy:\n")
        
        for t, lines in types_count.items():
            print(f"🚆 Typ: {t} (Ilość tras: {len(lines)})")
            # Wyświetlamy do 7 przykładowych linii dla podglądu
            sample_lines = ", ".join(lines[:7])
            if len(lines) > 7:
                sample_lines += "..."
            print(f"   Przykładowe linie: {sample_lines}")
            print("-" * 50)
            
    except Exception as e:
        print(f"Błąd pobierania danych: {e}")

if __name__ == "__main__":
    main()