chore: import bvg
This commit is contained in:
50
bvg/bvg.rb
Executable file
50
bvg/bvg.rb
Executable file
@@ -0,0 +1,50 @@
|
||||
#!/usr/bin/env ruby
|
||||
|
||||
require "open-uri"
|
||||
require "nokogiri"
|
||||
require "rainbow"
|
||||
|
||||
Departure = Struct.new(:time, :line, :terminus) do
|
||||
def self.from_row(row)
|
||||
Departure.new(row[0].text.strip, row[1].text.strip.gsub(/\s+/, " "), row[2].text.strip)
|
||||
end
|
||||
|
||||
def log
|
||||
time_string = if time[-1] == "*" then Rainbow(time[0..4]).inverse else time end
|
||||
line_string = {
|
||||
'B' => Rainbow(line).magenta,
|
||||
'R' => Rainbow(line).white,
|
||||
'U' => Rainbow(line).blue,
|
||||
'S' => Rainbow(line).green,
|
||||
'T' => Rainbow(line).red,
|
||||
}[line[0]] || line
|
||||
|
||||
printf("%s %-20s %s\n", time_string, line_string, terminus)
|
||||
end
|
||||
end
|
||||
|
||||
def request_url(input)
|
||||
"http://mobil.bvg.de/Fahrinfo/bin/stboard.bin/dox?" + URI.encode_www_form(
|
||||
"input" => URI.encode_www_form_component(input),
|
||||
"start" => "yes",
|
||||
"boardType" => "depRT",
|
||||
)
|
||||
end
|
||||
|
||||
url = request_url(ARGV[0])
|
||||
|
||||
document = Nokogiri::HTML(open(url))
|
||||
|
||||
title = document.xpath("//div[@id=\"ivu_overview_input\"]/strong")
|
||||
puts Rainbow(title.text).bright if title
|
||||
|
||||
document.xpath("//span[@class=\"select\"]/a").each do |option|
|
||||
station_name = option.text.strip
|
||||
station_code = option.xpath("@href").text[/input=(\d+)/, 1]
|
||||
print station_code, " ", Rainbow(station_name).yellow, "\n"
|
||||
end
|
||||
|
||||
document.xpath("//table/tbody/tr").each do |row|
|
||||
columns = row.xpath("td")
|
||||
Departure.from_row(columns).log unless columns.length < 3
|
||||
end
|
||||
Reference in New Issue
Block a user