class FTP:
"""docstring for FTP."""
def __init__(self, lowerBound, upperBound):
self.lowerBound = lowerBound
self.upperBound = upperBound
self.timeline = TimeLine(lowerBound, upperBound)
self.config = None
def __str__(self):
out = "Schedule from: {0} to: {1}; {2} tasks\n"
out = out.format(self.lowerBound, self.upperBound,
len(self.config.tasks))
out += self.timeline.__str__()
return out
def setConfig(self, config):
self.config = config
for task in self.config.tasks:
self.timeline.applyTask(task)
def schedule(self, soft, task):
res = True
counter = task.offset
cpuUsed = task.wcet
jobNb = 0
while res and counter < self.upperBound:
if self.timeline[counter] is None:
if not cpuUsed: # cpuUsed == 0
jobNb += 1
counter = task.offset + (jobNb * task.period)
cpuUsed = task.wcet
else:
if soft or (counter < task.offset +
(jobNb * task.period) + task.deadline):
self.timeline[counter] = (task.priority, jobNb)
counter += 1
cpuUsed -= 1
else:
if not soft:
self.timeline.addDeadlineMiss(
counter, task, jobNb)
res = False
else:
counter += 1
return res
def hardSchedule(self, task):
return self.schedule(False, task)
def __init__(self, simTopology='', loghandler=logging.NullHandler()):
# don't re-initialize an instance (singleton pattern)
if self._init:
return
self._init = True
# store params
self.loghandler = loghandler
# local variables
self.moteHandlers = []
self.timeline = TimeLine.TimeLine()
self.propagation = Propagation.Propagation(simTopology)
self.idmanager = IdManager.IdManager()
self.locationmanager = LocationManager.LocationManager()
self.pauseSem = threading.Lock()
self.isPaused = False
self.stopAfterSteps = None
self.delay = 0
self.stats = SimEngineStats()
# logging this module
self.log = logging.getLogger('SimEngine')
self.log.setLevel(logging.INFO)
self.log.addHandler(logging.NullHandler())
# logging core modules
for loggerName in [
'SimEngine',
'Timeline',
'Propagation',
'IdManager',
'LocationManager',
'SimCli',
]:
temp = logging.getLogger(loggerName)
temp.setLevel(logging.INFO)
temp.addHandler(loghandler)
def __init__(self, loghandler=NullLogHandler):
# store params
self.loghandler = loghandler
# local variables
self.moteHandlers = []
self.timeline = TimeLine.TimeLine(self)
self.propagation = Propagation.Propagation(self)
self.idmanager = IdManager.IdManager(self)
self.locationmanager = LocationManager.LocationManager(self)
self.pauseSem = threading.Lock()
self.isPaused = False
self.stopAfterSteps = None
self.delay = 0
self.stats = SimEngineStats()
# create daemon thread to handle connection of newly created motes
self.daemonThreadHandler = DaemonThread.DaemonThread(self)
# logging this module
self.log = logging.getLogger('SimEngine')
self.log.setLevel(logging.DEBUG)
self.log.addHandler(NullLogHandler())
# logging core modules
for loggerName in [
'SimEngine',
'Timeline',
'Propagation',
'IdManager',
'LocationManager',
'SimCli',
'DaemonThread',
]:
temp = logging.getLogger(loggerName)
temp.setLevel(logging.DEBUG)
temp.addHandler(loghandler)
def __init__(self, project, mainview):
"""
Creates a new instance of TimeLineBar
Parameters:
project -- reference to Project (Project.py).
mainview -- reference to MainApp (JokosherApp.py).
"""
gtk.HBox.__init__(self)
self.project = project
self.mainview = mainview
self.timeline = TimeLine.TimeLine(self.project, mainview)
self.Updating = False
# add click / bpm / signature box
self.clickbutton = gtk.VolumeButton()
self.clickbutton.set_value(0)
self.clickbutton.set_relief(gtk.RELIEF_NORMAL)
self.clickbutton.set_property("size", gtk.ICON_SIZE_BUTTON)
self.clickbutton_metronome_image = gtk.Image()
self.clickbutton_metronome_image.set_from_file(
os.path.join(Globals.IMAGE_PATH, "icon_click.png"))
# use get_child() not get_image here because GtkScaleButton
# uses gtk_container_add() in its internal implementation
self.clickbutton_volume_image = self.clickbutton.get_child()
self.clickbutton.set_image(self.clickbutton_metronome_image)
image_size_group = gtk.SizeGroup(gtk.SIZE_GROUP_BOTH)
image_size_group.add_widget(self.clickbutton_metronome_image)
image_size_group.add_widget(self.clickbutton_volume_image)
self.clickbutton.set_tooltip_text(_("Adjust volume of click track"))
self.bpmeventbox = gtk.EventBox()
self.bpmframe = gtk.Frame()
self.bpmeventbox.set_tooltip_text(_("Beats per minute"))
self.bpmframe.set_shadow_type(gtk.SHADOW_ETCHED_OUT)
self.bpmlabel = gtk.Label()
self.bpmlabel.set_use_markup(True)
self.bpmlabel.set_markup("<b>%s</b>" % (self.project.bpm))
self.bpmlabel.set_padding(5, 5)
self.bpmeventbox.add(self.bpmlabel)
self.bpmframe.add(self.bpmeventbox)
self.bpmedit = gtk.SpinButton()
self.bpmedit.set_range(1, 400)
self.bpmedit.set_increments(1, 5)
self.bpmedit.set_value(self.project.bpm)
self.bpmedit.connect("activate", self.OnAcceptEditBPM)
self.bpmedit.connect("key_press_event", self.OnEditBPMKey)
self.bpmedit.connect_after("button-release-event", self.OnEditBPMClick)
self.sigeventbox = gtk.EventBox()
self.sigeventbox.set_tooltip_text(_("Time signature"))
self.sigframe = gtk.Frame()
self.sigframe.set_shadow_type(gtk.SHADOW_ETCHED_OUT)
self.siglabel = gtk.Label()
self.siglabel.set_use_markup(True)
self.siglabel.set_markup(
"<b>%d/%d</b>" %
(self.project.meter_nom, self.project.meter_denom))
self.siglabel.set_padding(5, 5)
self.sigeventbox.add(self.siglabel)
self.sigframe.add(self.sigeventbox)
self.sigeditPacked = False
# set events
self.clickbutton.connect("value-changed", self.OnClickTrackVolume)
self.clickbutton.connect("enter_notify_event", self.OnClickButtonEnter)
self.clickbutton.connect("leave_notify_event", self.OnClickButtonEnter)
self.bpmeventbox.set_events(gtk.gdk.BUTTON_PRESS_MASK)
self.bpmeventbox.connect("button_release_event", self.OnEditBPM)
self.bpmeventbox.connect("enter_notify_event", self.OnMouseMoveBPM)
self.bpmeventbox.connect("leave_notify_event", self.OnMouseMoveBPM)
self.sigeventbox.set_events(gtk.gdk.BUTTON_PRESS_MASK)
self.sigeventbox.connect("button_press_event", self.OnEditSig)
self.sigeventbox.connect("enter_notify_event", self.OnMouseMoveTimeSig)
self.sigeventbox.connect("leave_notify_event", self.OnMouseMoveTimeSig)
self.project.connect("bpm", self.OnProjectBPMChange)
self.project.connect("time-signature", self.OnProjectSigChange)
self.project.connect("click-track", self.OnProjectClickTrackChange)
self.sigDialog = None
# ###########################################################
self.headerhbox = gtk.HBox()
self.headerhbox.set_border_width(2)
self.headerhbox.set_spacing(5)
self.headerhbox.pack_start(self.clickbutton, True, True)
self.headerhbox.pack_start(self.bpmframe, True, True)
self.headerhbox.pack_start(self.sigframe, True, True)
self.headerVBox = gtk.VBox()
self.headerVBox.pack_start(self.headerhbox, True, True)
self.headerVBox.pack_start(gtk.HSeparator(), False, False)
self.timelineVBox = gtk.VBox()
self.timelineVBox.pack_start(self.timeline, True, True)
separator = EventLaneHSeparator.EventLaneHSeparator(
self.project, self.project.transport)
self.timelineVBox.pack_start(separator, False, False)
self.pack_start(self.headerVBox, False, False)
self.pack_start(self.timelineVBox)
def __init__(self, lowerBound, upperBound):
self.lowerBound = lowerBound
self.upperBound = upperBound
self.timeline = TimeLine(lowerBound, upperBound)
self.config = None
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.centralWidget = QtGui.QWidget()
self.setCentralWidget(self.centralWidget)
self.buttonSave = QtGui.QPushButton("Save")
self.buttonBack = QtGui.QPushButton("<<")
self.buttonForward = QtGui.QPushButton(">>")
self.selectDriver = QtGui.QComboBox()
self.selectDriver.setInsertPolicy(6)
self.label = QtGui.QLabel()
self.textInput = QtGui.QTextEdit()
self.buttonClear = QtGui.QPushButton("Clear")
self.commentsBox = QtGui.QTextEdit()
self.dayView = QtGui.QTableWidget()
header = self.dayView.horizontalHeader()
header.setResizeMode(QtGui.QHeaderView.Stretch)
self.dayView.setColumnCount(3)
self.dayView.setHorizontalHeaderLabels(
['Activity Length', 'Activity', 'Infringement'])
self.buttonAdd = QtGui.QPushButton("Add")
self.workGraph = QtGui.QGraphicsView()
self.workGraph.setMinimumHeight(195)
self.calculator = Calculator(self)
self.TC = TimeConvert()
self.timeLine = TimeLine(self)
self.workGraph.setScene(self.timeLine)
self.yearView = YearView(self)
self.yearView.setMinimumHeight(330)
# the basic layout:
box = QtGui.QGridLayout()
# Top button bar
top_button_widget = QtGui.QWidget()
top_button_layout = QtGui.QHBoxLayout()
top_button_widget.setLayout(top_button_layout)
top_button_layout.addWidget(self.selectDriver)
top_button_layout.addWidget(self.buttonBack)
top_button_layout.addWidget(self.label)
top_button_layout.addWidget(self.buttonForward)
top_button_layout.addWidget(self.buttonSave)
self.label.setAlignment(Qt.Qt.AlignCenter)
# Calendar secn
top_box = QtGui.QWidget()
top_box_layout = QtGui.QVBoxLayout()
top_box.setLayout(top_box_layout)
top_box_layout.addWidget(top_button_widget)
top_box_layout.addWidget(self.yearView)
# Calc buttons
calc_box = QtGui.QWidget()
calc_box_layout = QtGui.QHBoxLayout()
calc_box.setLayout(calc_box_layout)
calc_box_layout.addWidget(self.buttonClear)
# Left mid top
textInput_box = QtGui.QWidget()
textInput_layout = QtGui.QVBoxLayout()
textInput_box.setLayout(textInput_layout)
textInput_layout.addWidget(self.textInput)
# Left mid box
left_mid_box = QtGui.QWidget()
left_mid_layout = QtGui.QVBoxLayout()
left_mid_box.setLayout(left_mid_layout)
left_mid_layout.addWidget(textInput_box)
left_mid_layout.addWidget(calc_box)
# Right mid bottom
add_box = QtGui.QWidget()
add_layout = QtGui.QVBoxLayout()
add_box.setLayout(add_layout)
add_layout.addWidget(self.commentsBox)
add_layout.addWidget(self.buttonAdd)
# Right mid top
dayView_box = QtGui.QWidget()
textInput_layout = QtGui.QVBoxLayout()
dayView_box.setLayout(textInput_layout)
textInput_layout.addWidget(self.dayView)
# Right mid
splitter_R_mid = QtGui.QSplitter(Qt.Qt.Vertical)
splitter_R_mid.addWidget(dayView_box)
splitter_R_mid.addWidget(add_box)
# Middle
splitter_central = QtGui.QSplitter(Qt.Qt.Horizontal)
splitter_central.addWidget(left_mid_box)
splitter_central.addWidget(splitter_R_mid)
# Top and middle
splitter_top = QtGui.QSplitter(Qt.Qt.Vertical)
splitter_top.addWidget(top_box)
splitter_top.addWidget(splitter_central)
# Bottom
splitter_bottom = QtGui.QSplitter(Qt.Qt.Vertical)
splitter_bottom.addWidget(splitter_top)
splitter_bottom.addWidget(self.workGraph)
box.addWidget(splitter_bottom)
self.centralWidget.setLayout(box)
stylesheet = "QSplitter::handle{background: gray; " \
"width: 1px; height: 1px;}"
splitter_bottom.setStyleSheet(stylesheet)
QtGui.QApplication.setStyle(Qt.QStyleFactory.create('cleanlooks'))
self.resize(1125, 702)
self.setWindowTitle("TachoAnalyser v 2.1")
self.setWindowIcon(QtGui.QIcon('./icons/icon.png'))
##################################################
# Signals
self.buttonBack.clicked.connect(self.year_back)
self.buttonForward.clicked.connect(self.year_forward)
self.buttonClear.clicked.connect(self.clear_input)
self.selectDriver.activated[str].connect(self.select_driver)
self.buttonAdd.clicked.connect(self.add)
self.buttonSave.clicked.connect(self.save)
self.textInput.textChanged.connect(self.read_input)
##################################################
# Initialise
self.drivers = [
'Driver...', 'Andy', 'Chris', 'Dan', 'Richard', 'DriverX'
]
self.months = {
'Jan': '01',
'Feb': '02',
'Mar': '03',
'Apr': '04',
'May': '05',
'Jun': '06',
'Jul': '07',
'Aug': '08',
'Sep': '09',
'Oct': '10',
'Nov': '11',
'Dec': '12'
}
self.selectDriver.addItems(self.drivers)
self.driver = ""
self.key_list = [] # temp store for model_dict keys
self.model_dict = {}
self.infringements = ""
today = datetime.date.today()
self.year = today.year
self.dirty = False
self.init_model()
class MainWindow(QtGui.QMainWindow):
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.centralWidget = QtGui.QWidget()
self.setCentralWidget(self.centralWidget)
self.buttonSave = QtGui.QPushButton("Save")
self.buttonBack = QtGui.QPushButton("<<")
self.buttonForward = QtGui.QPushButton(">>")
self.selectDriver = QtGui.QComboBox()
self.selectDriver.setInsertPolicy(6)
self.label = QtGui.QLabel()
self.textInput = QtGui.QTextEdit()
self.buttonClear = QtGui.QPushButton("Clear")
self.commentsBox = QtGui.QTextEdit()
self.dayView = QtGui.QTableWidget()
header = self.dayView.horizontalHeader()
header.setResizeMode(QtGui.QHeaderView.Stretch)
self.dayView.setColumnCount(3)
self.dayView.setHorizontalHeaderLabels(
['Activity Length', 'Activity', 'Infringement'])
self.buttonAdd = QtGui.QPushButton("Add")
self.workGraph = QtGui.QGraphicsView()
self.workGraph.setMinimumHeight(195)
self.calculator = Calculator(self)
self.TC = TimeConvert()
self.timeLine = TimeLine(self)
self.workGraph.setScene(self.timeLine)
self.yearView = YearView(self)
self.yearView.setMinimumHeight(330)
# the basic layout:
box = QtGui.QGridLayout()
# Top button bar
top_button_widget = QtGui.QWidget()
top_button_layout = QtGui.QHBoxLayout()
top_button_widget.setLayout(top_button_layout)
top_button_layout.addWidget(self.selectDriver)
top_button_layout.addWidget(self.buttonBack)
top_button_layout.addWidget(self.label)
top_button_layout.addWidget(self.buttonForward)
top_button_layout.addWidget(self.buttonSave)
self.label.setAlignment(Qt.Qt.AlignCenter)
# Calendar secn
top_box = QtGui.QWidget()
top_box_layout = QtGui.QVBoxLayout()
top_box.setLayout(top_box_layout)
top_box_layout.addWidget(top_button_widget)
top_box_layout.addWidget(self.yearView)
# Calc buttons
calc_box = QtGui.QWidget()
calc_box_layout = QtGui.QHBoxLayout()
calc_box.setLayout(calc_box_layout)
calc_box_layout.addWidget(self.buttonClear)
# Left mid top
textInput_box = QtGui.QWidget()
textInput_layout = QtGui.QVBoxLayout()
textInput_box.setLayout(textInput_layout)
textInput_layout.addWidget(self.textInput)
# Left mid box
left_mid_box = QtGui.QWidget()
left_mid_layout = QtGui.QVBoxLayout()
left_mid_box.setLayout(left_mid_layout)
left_mid_layout.addWidget(textInput_box)
left_mid_layout.addWidget(calc_box)
# Right mid bottom
add_box = QtGui.QWidget()
add_layout = QtGui.QVBoxLayout()
add_box.setLayout(add_layout)
add_layout.addWidget(self.commentsBox)
add_layout.addWidget(self.buttonAdd)
# Right mid top
dayView_box = QtGui.QWidget()
textInput_layout = QtGui.QVBoxLayout()
dayView_box.setLayout(textInput_layout)
textInput_layout.addWidget(self.dayView)
# Right mid
splitter_R_mid = QtGui.QSplitter(Qt.Qt.Vertical)
splitter_R_mid.addWidget(dayView_box)
splitter_R_mid.addWidget(add_box)
# Middle
splitter_central = QtGui.QSplitter(Qt.Qt.Horizontal)
splitter_central.addWidget(left_mid_box)
splitter_central.addWidget(splitter_R_mid)
# Top and middle
splitter_top = QtGui.QSplitter(Qt.Qt.Vertical)
splitter_top.addWidget(top_box)
splitter_top.addWidget(splitter_central)
# Bottom
splitter_bottom = QtGui.QSplitter(Qt.Qt.Vertical)
splitter_bottom.addWidget(splitter_top)
splitter_bottom.addWidget(self.workGraph)
box.addWidget(splitter_bottom)
self.centralWidget.setLayout(box)
stylesheet = "QSplitter::handle{background: gray; " \
"width: 1px; height: 1px;}"
splitter_bottom.setStyleSheet(stylesheet)
QtGui.QApplication.setStyle(Qt.QStyleFactory.create('cleanlooks'))
self.resize(1125, 702)
self.setWindowTitle("TachoAnalyser v 2.1")
self.setWindowIcon(QtGui.QIcon('./icons/icon.png'))
##################################################
# Signals
self.buttonBack.clicked.connect(self.year_back)
self.buttonForward.clicked.connect(self.year_forward)
self.buttonClear.clicked.connect(self.clear_input)
self.selectDriver.activated[str].connect(self.select_driver)
self.buttonAdd.clicked.connect(self.add)
self.buttonSave.clicked.connect(self.save)
self.textInput.textChanged.connect(self.read_input)
##################################################
# Initialise
self.drivers = [
'Driver...', 'Andy', 'Chris', 'Dan', 'Richard', 'DriverX'
]
self.months = {
'Jan': '01',
'Feb': '02',
'Mar': '03',
'Apr': '04',
'May': '05',
'Jun': '06',
'Jul': '07',
'Aug': '08',
'Sep': '09',
'Oct': '10',
'Nov': '11',
'Dec': '12'
}
self.selectDriver.addItems(self.drivers)
self.driver = ""
self.key_list = [] # temp store for model_dict keys
self.model_dict = {}
self.infringements = ""
today = datetime.date.today()
self.year = today.year
self.dirty = False
self.init_model()
def init_model(self):
"""Placeholder model - called at startup or if no driver selected so
that the calendar dates can be filled in, or if no model has been
initialised for driver.
"""
model = Model()
model.set_year(self.year)
self.yearView.setModel(model)
self.yearView.set_selection_model(model)
self.model_dict[self.year] = model
self.label.setText(str(self.year))
self.clear_input()
def year_back(self):
self.year -= 1
self.select_driver(self.driver)
if self.year not in self.model_dict:
self.init_model()
self.get_file()
else:
self.label.setText(str(self.year))
self.clear_input()
self.set_model()
def year_forward(self):
self.year += 1
self.select_driver(self.driver)
if self.year not in self.model_dict:
self.init_model()
self.get_file()
else:
self.label.setText(str(self.year))
self.clear_input()
self.set_model()
def clear_input(self):
self.textInput.clear()
self.timeLine.clear_timeline()
self.dayView.clearContents()
self.commentsBox.clear()
self.calculator.clear()
def select_driver(self, driver):
if self.dirty:
messageBox = QtGui.QMessageBox()
reply = messageBox.question(
messageBox, "UnSaved Data", "Save Data?", QtGui.QMessageBox.Yes
| QtGui.QMessageBox.No | QtGui.QMessageBox.Cancel)
if reply == QtGui.QMessageBox.Cancel:
pass
elif reply == QtGui.QMessageBox.Yes:
self.save()
self.init_driver(driver)
else:
prev_driver = self.drivers.index(self.driver)
self.selectDriver.setCurrentIndex(prev_driver)
else:
self.init_driver(driver)
def init_driver(self, driver):
if driver != "Driver...":
self.init_model()
self.clear_input()
self.driver = driver
self.dirty = False
self.get_file()
def get_file(self):
file = './Data/' + self.driver + str(self.year) + '.xml'
if os.path.isfile(file):
dataIO = DataIO(self)
model = Model()
self.model_dict[self.year] = model
dataIO.open(file)
self.set_model()
def set_model(self):
self.model = self.model_dict[self.year]
self.model.set_year(self.year)
self.yearView.setModel(self.model)
self.yearView.set_selection_model(self.model)
self.label.setText(str(self.year))
def read_input(self):
"""Activity change info from readesm is pasted in and converted to
minutes for start, end and duration"""
activity_list = []
text = self.textInput.toPlainText()
text = str(text)
date_line = re.search(r'Activities on (.*?) .*', text)
if date_line:
date = (date_line.group())
day = (date[22:24])
day = day.strip()
day = day.zfill(2)
day = int(day)
month_str = (date[18:21])
month = self.months[month_str]
month = int(month)
year = (date[-5:-1])
year = int(year)
self.date = QtCore.QDate(year, month, day)
# Get Activities
working = re.findall(r'work, from (.*?) to (.*?) .*', text)
for item in working:
start = item[0]
end = item[1]
start = self.TC.hrs_to_mins(start)
end = self.TC.hrs_to_mins(end)
duration = self.TC.calc_duration(start, end)
activity = Activity("Working", start, end, duration, "")
activity_list.append(activity)
driving = re.findall(r'driving, from (.*?) to (.*?) .*', text)
for item in driving:
start = item[0]
end = item[1]
start = self.TC.hrs_to_mins(start)
end = self.TC.hrs_to_mins(end)
duration = self.TC.calc_duration(start, end)
activity = Activity("Driving", start, end, duration, "")
activity_list.append(activity)
break_rest = re.findall(r'break/rest, from (.*?) to (.*?) .*', text)
for item in break_rest:
start = item[0]
end = item[1]
start = self.TC.hrs_to_mins(start)
end = self.TC.hrs_to_mins(end)
duration = self.TC.calc_duration(start, end)
activity = Activity("Break", start, end, duration, "")
activity_list.append(activity)
self.timeLine.add_activities(activity_list)
activity_list.sort(key=lambda x: x.start, reverse=False)
self.calculator.timers(activity_list)
self.show_activities(activity_list)
self.buttonAdd.setText("Add")
def show_activities(self, activity_list):
self.dayView.clearContents()
for item in reversed(activity_list):
item.duration = self.TC.mins_to_hrs(item.duration)
hours = QtGui.QTableWidgetItem(item.duration)
mode = QtGui.QTableWidgetItem(item.mode)
infr = QtGui.QTableWidgetItem(item.infr)
self.dayView.insertRow(0)
self.dayView.setItem(0, 0, hours)
self.dayView.setItem(0, 1, mode)
self.dayView.setItem(0, 2, infr)
num = len(activity_list)
self.dayView.setRowCount(num)
def add(self):
"""Invoked from add button; adds day to model"""
date = QtCore.QDate(self.date)
activities = self.textInput.toPlainText()
comments = self.commentsBox.toPlainText()
year = date.year()
self.year = year
self.label.setText(str(self.year))
if self.year not in self.model_dict:
self.init_model()
self.get_file()
else:
self.label.setText(str(self.year))
self.clear_input()
self.set_model()
month = date.month()
day = date.day()
year_instance = Year(self, year)
col = year_instance.get_column(month, day)
row = month - 1
current_model = self.model_dict[year]
item = current_model.item(row, col)
comment = QtGui.QStandardItem()
activity = QtGui.QStandardItem()
infringed = QtGui.QStandardItem()
item.setChild(0, 2, comment)
item.setChild(0, 3, activity)
item.setChild(0, 4, infringed)
comment.setData(comments)
activity.setData(activities)
infringed.setData(self.infringements)
item.setBackground(QtGui.QColor(109, 255, 174))
if infringed.data() == "hgv":
item.setBackground(QtGui.QColor(255, 150, 150))
if infringed.data() == "wtd":
item.setBackground(QtGui.QColor(255, 213, 140))
if infringed.data() == "both":
pixmap = QtGui.QPixmap('./icons/dual_infr.svg')
brush = QtGui.QBrush(pixmap)
item.setBackground(brush)
self.infringements = ""
self.dirty = True
if self.buttonAdd.text() == "Add":
self.clear_input()
self.set_model()
def save(self):
dataIO = DataIO(self)
dataIO.save(self.model_dict)
self.dirty = False
def closeEvent(self, event):
if self.dirty:
messageBox = QtGui.QMessageBox()
reply = messageBox.question(
messageBox, "UnSaved Data", "Save Data?", QtGui.QMessageBox.Yes
| QtGui.QMessageBox.No | QtGui.QMessageBox.Cancel)
if reply == QtGui.QMessageBox.Cancel:
event.ignore()
elif reply == QtGui.QMessageBox.Yes:
event.ignore()
self.save()
else:
event.accept()