timetable 0.1.0

Construct timetables from General Transit Feed Specification (GTFS)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

use reqwest;
use std::error::Error;
use std::io::prelude::*;
use std::fs::File;
use std::path::Path;
use transitfeed::{GTFSIterator, Calendar, Route, Stop, StopTime, Trip};
use zip::ZipArchive;

pub fn unzip_to_path(archive: String, path: String) {
    let reader = File::open(&archive).unwrap();
    let mut zip = ZipArchive::new(reader).unwrap();

    for i in 0..zip.len() {
        let mut file = zip.by_index(i).unwrap();
        let filename = Path::new(&path).join(file.name());
        let mut f = File::create(&filename).unwrap();
        let mut content = Vec::new();
        if let Err(why) = file.read_to_end(&mut content) {
            panic!("Error: could not read '{}' in '{}': {}", file.name(), archive, why.description());
        }
        if let Err(why) = f.write_all(&content) {
            panic!("Error: could not write '{}': {}", filename.display(), why.description());
        }
    }
}

pub fn download_to_path(url: String, path: String) {
    let mut res = reqwest::get(&url).unwrap();
    if !res.status().is_success() {
        panic!("Error: could not download '{}': {:?}", url, res.status());
    }
    let mut buf = Vec::new();
    res.copy_to(&mut buf).unwrap();
    let filename = Path::new(&path).join("gtfs.zip");
    let mut f = File::create(&filename).unwrap();
    if let Err(why) = f.write_all(&buf) {
        panic!("Error: could not write '{}': {}", filename.display(), why.description());
    }
}

pub struct GTFS {
    path: String
}

impl GTFS {
    pub fn from_path(path: String) -> GTFS {
        GTFS { path }
    }

    pub fn calendar(&self) -> GTFSIterator<File, Calendar> {
        let path = Path::new(&self.path).join("calendar.txt");
        GTFSIterator::from_path(path.to_str().unwrap()).unwrap()
    }

    pub fn routes(&self) -> GTFSIterator<File, Route> {
        let path = Path::new(&self.path).join("routes.txt");
        GTFSIterator::from_path(path.to_str().unwrap()).unwrap()
    }

    pub fn stops(&self) -> GTFSIterator<File, Stop> {
        let path = Path::new(&self.path).join("stops.txt");
        GTFSIterator::from_path(path.to_str().unwrap()).unwrap()
    }

    pub fn stop_times(&self) -> GTFSIterator<File, StopTime> {
        let path = Path::new(&self.path).join("stop_times.txt");
        GTFSIterator::from_path(path.to_str().unwrap()).unwrap()
    }

    pub fn trips(&self) -> GTFSIterator<File, Trip> {
        let path = Path::new(&self.path).join("trips.txt");
        GTFSIterator::from_path(path.to_str().unwrap()).unwrap()
    }
}