Serge NOEL
284 lines
11 KiB
Python
284 lines
11 KiB
Python
import re
|
|
import jmri
|
|
from java.awt import BorderLayout, Dimension, FlowLayout
|
|
from javax.swing import JEditorPane, JFrame, JPanel, JScrollPane
|
|
from java.awt.event import WindowAdapter, WindowEvent
|
|
from ast import literal_eval
|
|
import javax.swing.text.html.HTMLEditorKit as HTMLEditorKit
|
|
import java.net as net
|
|
|
|
|
|
class Timetable:
|
|
|
|
def __init__(self, station):
|
|
# print "start timetable"
|
|
self.html_content = "unset"
|
|
# print "html_content", self.html_content
|
|
self.frame = None
|
|
# print "creating window"
|
|
self.createWindow(station)
|
|
|
|
|
|
|
|
def update_timetable(self, station_name, station_names_list, time, timetable):
|
|
|
|
# print "required_station, time, timetable", required_station, time, timetable
|
|
# print "required_station", required_station
|
|
|
|
# print "update timetable"
|
|
html_table = None
|
|
# for station in required_stations:
|
|
# html_table += self.get_html_table(time, station, timetable)
|
|
html_table = self.get_html_table(time, station_name, station_names_list, timetable)
|
|
|
|
# print "html_table", html_table
|
|
html_content2 = self.html_content.replace("$time$", time).replace("$table$", html_table)
|
|
|
|
# print "html_content2", html_content2
|
|
# print "about to load html", html_content2
|
|
self.load_html(html_content2)
|
|
# print "end update timetable"
|
|
|
|
def get_html_table(self, time, station_name, station_names_list, timetable):
|
|
|
|
# a lot of heavy work here because the arrival and departure times have spaces
|
|
# and to overcome this and sort by time we need to sort by the max of the
|
|
# arrival and departure times
|
|
|
|
# print "timetable", timetable
|
|
|
|
if timetable == None or timetable == []:
|
|
return ""
|
|
|
|
string = '''<tr>
|
|
<th>$station_arrival_time$</th>
|
|
<th>$station_departure_time$</th>
|
|
<th>$platform$</th>
|
|
<th>$first_station$</th>
|
|
<th>$last_station$</th>
|
|
<th>$via$</th$>
|
|
</tr>
|
|
'''
|
|
html_table1 = '''
|
|
<table>
|
|
<tr>
|
|
<th colspan="2">Departures: $station$</th>
|
|
<th>$time$</th>
|
|
</tr>
|
|
<tr>
|
|
<th class="it">Arrival Time</th>
|
|
<th class="it">Departure Time</th>
|
|
<th class="it">Platform</th>
|
|
<th class="it">From</th>
|
|
<th class="it">Destination</th>
|
|
<th class="it">Via</th>
|
|
</tr>
|
|
'''
|
|
|
|
try:
|
|
html_table = html_table1.replace("$station$", station_name).replace("$time$", time)
|
|
# html_table = html_table1.replace("$station$", "fred").replace("$time$", time)
|
|
except:
|
|
html_table = html_table1.replace("$station$", "station not set 1").replace("$time$", time)
|
|
timetable_list = []
|
|
for timetable_entry in timetable:
|
|
[train_name, station_name1, platform_name, station_arrival_time, station_departure_time, first_station, last_station, via] \
|
|
= timetable_entry
|
|
if str(station_name1) in station_names_list:
|
|
station_time = 0
|
|
sat = 0
|
|
sdt = 0
|
|
if station_arrival_time != "":
|
|
[h, m] = station_arrival_time.split(":")
|
|
sat = int(h)*60 + int(m)
|
|
|
|
if station_departure_time != "":
|
|
[h, m] = station_departure_time.split(":")
|
|
sdt = str(int(h)*60 + int(m))
|
|
station_time = str(max(sat, sdt))
|
|
html_table_entry = string.replace("$station_name$", station_name1) \
|
|
.replace("$station_arrival_time$", station_arrival_time) \
|
|
.replace("$station_departure_time$", station_departure_time) \
|
|
.replace("$platform$", platform_name) \
|
|
.replace("$first_station$", first_station) \
|
|
.replace("$last_station$", last_station) \
|
|
.replace("$via$", via)
|
|
# print "html_table_entry", html_table_entry
|
|
# timetable_list += {station_time: "station_time", str(html_table_entry): "html_table_entry"}
|
|
timetable_list.append({"station_time": station_time, "html_table_entry": html_table_entry})
|
|
# print "#######################################################"
|
|
# print "html_table", html_table_entry
|
|
# print "#######################################################"
|
|
# sort timetable_list
|
|
# print "timetable_list_before sort", timetable_list
|
|
timetable_list.sort(key=self.myFunc)
|
|
# print "timetable_list", timetable_list
|
|
for t in timetable_list:
|
|
# html_table += literal_eval(t['html_table_entry'])
|
|
html_table += t['html_table_entry']
|
|
html_table += "\n</table>"
|
|
# print "html_table", html_table
|
|
|
|
return html_table
|
|
|
|
def myFunc(self, e):
|
|
station_time = e['station_time']
|
|
return int(station_time)
|
|
|
|
def createWindow(self, station):
|
|
global timetable_frame_gbl
|
|
# print "create window start"
|
|
timetable_frame_gbl = MyFrame("Timetable") #based on JFrame see definition below
|
|
self.frame = timetable_frame_gbl
|
|
|
|
self.createUI(station)
|
|
self.frame.setSize(900, 450)
|
|
self.frame.setLocationRelativeTo(None)
|
|
self.frame.setVisible(True)
|
|
# print "created window"
|
|
|
|
# Define a custom window listener for closing events
|
|
class CustomWindowListener(WindowAdapter):
|
|
def windowClosing(inner_self, event):
|
|
global run_local_timetable_gbl
|
|
# hide the window
|
|
# can be shown again by clicking show timetable in main menu and selecting show timetable option
|
|
run_local_timetable_gbl = False
|
|
|
|
# Add the custom window listener to the frame.
|
|
self.frame.addWindowListener(CustomWindowListener())
|
|
# Prevent the default close operation so that our custom behavior is applied.
|
|
self.frame.setDefaultCloseOperation(JFrame.DO_NOTHING_ON_CLOSE)
|
|
|
|
def hideWindow(self):
|
|
if self.frame is not None:
|
|
self.frame.setVisible(False)
|
|
|
|
def showWindow(self):
|
|
if self.frame != None:
|
|
self.frame.setVisible(True)
|
|
|
|
def createUI(self, station):
|
|
# print "in createUI"
|
|
self.panel = JPanel()
|
|
# self.panel.setLayout(BorderLayout()) #allow resizing of window by using BorderLayout
|
|
layout = BorderLayout()
|
|
self.panel.setLayout(layout)
|
|
|
|
self.jEditorPane = JEditorPane()
|
|
|
|
self.jEditorPane.setEditable(False)
|
|
# print "about to get_html", "html_content", self.html_content
|
|
# if self.html_content == "unset":
|
|
# print "about to get_html2"
|
|
self.html_content = self.get_html()
|
|
self.html_content = self.edit_html_station(station)
|
|
# print "about to load html"
|
|
self.load_html(self.html_content)
|
|
# print "loaded html"
|
|
jScrollPane = JScrollPane(self.jEditorPane)
|
|
jScrollPane.setPreferredSize(Dimension(540, 400))
|
|
|
|
self.panel.add(jScrollPane)
|
|
self.frame.getContentPane().add(self.panel, BorderLayout.CENTER)
|
|
# print "finished create UI"
|
|
|
|
def get_html(self):
|
|
# print "in get_html"
|
|
file = jmri.util.FileUtil.getExternalFilename('program:jython/DispatcherSystem/Timetable.html')
|
|
# print "file", file
|
|
with open(file, 'r') as file:
|
|
html_content = file.read()
|
|
# print "html_content read"
|
|
# html_content = html_content.replace ("$station$", station)
|
|
|
|
return html_content
|
|
|
|
def load_html(self, html_content):
|
|
# print "in load_html"
|
|
# print "type html_edit", type(html_content)
|
|
# print "html_content", html_content
|
|
# Set the content type to HTML
|
|
self.jEditorPane.setContentType("text/html")
|
|
self.jEditorPane.setText(html_content)
|
|
# try:
|
|
# self.jEditorPane.setText(html_content)
|
|
# except IndexError:
|
|
# "********************************************error in loading html"
|
|
# except:
|
|
# self.jEditorPane.setText("<html>Page not found.</html>")
|
|
|
|
def edit_html_station(self, station):
|
|
|
|
html_edit = self.html_content
|
|
# print "html_edit type", type(html_edit)
|
|
# print "*********************station*****************************", station
|
|
try:
|
|
html_edit = html_edit.replace("$station$", station)
|
|
except:
|
|
html_edit = html_edit.replace("$station$", "station not set")
|
|
# html_edit = html_edit.replace ("$time$", time)
|
|
# print "EDITED html time"
|
|
return html_edit
|
|
|
|
def edit_parameters_html(self, parameters):
|
|
# print "in edit_html"
|
|
[param1, param2] = parameters
|
|
# print "parameters", parameters
|
|
# if "$param1$" in self.html_content:
|
|
# print '$param1$ in self.html_content'
|
|
html_edit = self.html_content
|
|
# print "html_edit assigned"
|
|
# if "$param1$" in html_edit:
|
|
# print '$param1$ in html_edit'
|
|
# print "type html_edit", type(html_edit)
|
|
# replacements = {"$param1$": "fred", "$param2$": "jim"}
|
|
# print "replacements", replacements
|
|
# html_edit = "param1"
|
|
# print "html_edit", html_edit
|
|
# self.custom_replace(html_edit, replacements)
|
|
html_edit = html_edit.replace("$param1$", "fred")
|
|
# print "html_edit after", html_edit
|
|
html_edit = html_edit.replace("[param2]", param2)
|
|
# if "$param1$" in html_edit:
|
|
# print '$param1$ in html_edit after edit'
|
|
# else:
|
|
# print '$param1$ not in html_edit after edit'
|
|
# if param1 in html_edit:
|
|
# print 'param1 in html_edit after edit', "param1", param1
|
|
# print "EDITED html"
|
|
return html_edit
|
|
|
|
def edit_html_time(self, time):
|
|
|
|
html_edit = self.html_content
|
|
html_edit = html_edit.replace("$time$", time)
|
|
# print "EDITED html time"
|
|
return html_edit
|
|
|
|
|
|
class MyFrame(JFrame):
|
|
|
|
def __init__(self, title):
|
|
|
|
self.setTitle(title)
|
|
|
|
# Set the default close operation
|
|
self.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE)
|
|
|
|
# Add a window listener to handle the close event
|
|
self.addWindowListener(MyWindowAdapter())
|
|
|
|
|
|
class MyWindowAdapter(WindowAdapter):
|
|
|
|
def windowClosing(self, e):
|
|
global run_timetable_gbl
|
|
# print("Window is closing")
|
|
# we need to hide the frame
|
|
frame = e.getSource()
|
|
frame.setVisible(False)
|
|
|
|
#make sure the frame is not created again
|
|
run_timetable_gbl = False
|