blob: a977c1ad315ece55b23a735deeb7f261a90fbb7c (
plain)
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
|
module STIF
class NetexFile
CALENDAR_FILE_NAME = 'calendriers.xml'
LINE_FILE_FORMAT = /^offre_.*\.xml$/
XML_NAME_SPACE = "http://www.netex.org.uk/netex"
def initialize(file_name)
@file_name = file_name
end
def frames
frames = Hash.new { |h,k| h[k] = NetexFile::Frame.new(k) }
Zip::File.open(@file_name) do |zipfile|
zipfile.each do |entry|
next unless entry.ftype == :file
entry_dir_name, entry_file_name = File.split(entry.name)
case entry_file_name
when CALENDAR_FILE_NAME
entry.get_input_stream do |stream|
frames[entry_dir_name].parse_calendars(stream.read)
end
when LINE_FILE_FORMAT
frames[entry_dir_name].add_offer_file(entry_file_name)
end
end
end
frames.values
end
end
class NetexFile::Frame
attr_accessor :name
def initialize(name)
@name = name
end
def parse_calendars(calendars)
# <netex:ValidBetween>
# <netex:FromDate>2017-03-01</netex:FromDate>
# <netex:ToDate>2017-03-31</netex:ToDate>
# </netex:ValidBetween>
xml = Nokogiri::XML(calendars)
xml.xpath("//netex:ValidBetween", "netex" => NetexFile::XML_NAME_SPACE).each do |valid_between|
from_date = valid_between.xpath("netex:FromDate").try :text
to_date = valid_between.xpath("netex:ToDate").try :text
periods << Range.new(Date.parse(from_date), Date.parse(to_date))
end
end
def add_offer_file(file_name)
if file_name =~ /^offre_([^_]*)_/
line_refs << $1
end
end
def periods
@periods ||= []
end
def line_refs
@line_refs ||= []
end
end
end
|
