parent
							
								
									84ea916747
								
							
						
					
					
						commit
						efbbf3e350
					
				| 
						 | 
					@ -38,6 +38,25 @@ def get(
 | 
				
			||||||
    ]
 | 
					    ]
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def get_trains(from_date: date, filepath: str) -> list[Event]:
 | 
				
			||||||
 | 
					    """Get train events."""
 | 
				
			||||||
 | 
					    events: list[Event] = []
 | 
				
			||||||
 | 
					    for item in yaml.safe_load(open(filepath)):
 | 
				
			||||||
 | 
					        if item["depart"].date() < from_date:
 | 
				
			||||||
 | 
					            continue
 | 
				
			||||||
 | 
					        events += [
 | 
				
			||||||
 | 
					            Event(
 | 
				
			||||||
 | 
					                date=leg["depart"],
 | 
				
			||||||
 | 
					                end_date=leg["arrive"],
 | 
				
			||||||
 | 
					                name="transport",
 | 
				
			||||||
 | 
					                title=f'train from {leg["from"]} to {leg["to"]}',
 | 
				
			||||||
 | 
					                url=item.get("url"),
 | 
				
			||||||
 | 
					            )
 | 
				
			||||||
 | 
					            for leg in item["legs"]
 | 
				
			||||||
 | 
					        ]
 | 
				
			||||||
 | 
					    return events
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def flight_number(flight: Leg) -> str:
 | 
					def flight_number(flight: Leg) -> str:
 | 
				
			||||||
    """Flight number."""
 | 
					    """Flight number."""
 | 
				
			||||||
    airline_code = flight["airline"]
 | 
					    airline_code = flight["airline"]
 | 
				
			||||||
| 
						 | 
					@ -49,7 +68,7 @@ def flight_number(flight: Leg) -> str:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def all_events(from_date: date, data_dir: str) -> list[Event]:
 | 
					def all_events(from_date: date, data_dir: str) -> list[Event]:
 | 
				
			||||||
    """Get all flights and rail journeys."""
 | 
					    """Get all flights and rail journeys."""
 | 
				
			||||||
    trains = get(from_date, "train", os.path.join(data_dir, "trains.yaml"))
 | 
					    trains = get_trains(from_date, os.path.join(data_dir, "trains.yaml"))
 | 
				
			||||||
    flights = get(
 | 
					    flights = get(
 | 
				
			||||||
        from_date, "flight", os.path.join(data_dir, "flights.yaml"), extra=flight_number
 | 
					        from_date, "flight", os.path.join(data_dir, "flights.yaml"), extra=flight_number
 | 
				
			||||||
    )
 | 
					    )
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in a new issue