mirror of https://github.com/chriskiehl/Gooey.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
3.0 KiB
109 lines
3.0 KiB
'''
|
|
Created on Dec 22, 2013
|
|
|
|
@author: Chris
|
|
'''
|
|
import subprocess
|
|
import sys
|
|
from multiprocessing.dummy import Pool, Process
|
|
import time
|
|
|
|
import wx
|
|
|
|
from gooey.gui.lang import i18n
|
|
|
|
|
|
YES = 5103
|
|
NO = 5104
|
|
|
|
|
|
|
|
class Controller(object):
|
|
'''
|
|
Main controller for the gui.
|
|
|
|
All controlls are delegated to this central control point.
|
|
|
|
Args:
|
|
base_frame = Reference to the Basewindow
|
|
head_panel = reference to the BaseWindow's Head Panel
|
|
body_panel = reference to the BaseWindow's Body Panel
|
|
footer_panel = reference to the BaseWindow's Footer Panel
|
|
model = configuration model
|
|
translator = instance of the I18N class
|
|
'''
|
|
|
|
def __init__(self, base_frame, build_spec):
|
|
self.core_gui = base_frame
|
|
self.build_spec = build_spec
|
|
|
|
def OnCancelButton(self, widget, event):
|
|
msg = i18n.translate('sure_you_want_to_exit')
|
|
dlg = wx.MessageDialog(None, msg, i18n.translate('close_program'), wx.YES_NO)
|
|
result = dlg.ShowModal()
|
|
if result == YES:
|
|
dlg.Destroy()
|
|
self.core_gui.Destroy()
|
|
sys.exit()
|
|
dlg.Destroy()
|
|
|
|
def OnStartButton(self, widget, event):
|
|
cmd_line_args = self.core_gui.GetOptions()
|
|
|
|
if not self.build_spec['manual_start']:
|
|
_required = self.core_gui.GetRequiredArgs()
|
|
if _required and any(req == '' for req in _required):
|
|
self.ShowDialog(i18n.translate('error_title'), "Must fill in all fields in the Required section!", wx.ICON_ERROR)
|
|
return
|
|
|
|
command = '{0} {1}'.format(self.build_spec['target'], cmd_line_args)
|
|
self.core_gui.NextPage()
|
|
self.RunClientCode(command)
|
|
|
|
def RunClientCode(self, command):
|
|
def doInBackground(process, callback):
|
|
while True:
|
|
line = process.stdout.readline()
|
|
if not line:
|
|
break
|
|
self.core_gui.PublishConsoleMsg(line)
|
|
callback(process)
|
|
|
|
p = subprocess.Popen(command, bufsize=1, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
_pool = Pool(1)
|
|
_pool.apply_async(doInBackground, (p, self.HandleResult))
|
|
|
|
def HandleResult(self, process):
|
|
_stdout, _stderr = process.communicate()
|
|
if process.returncode == 0:
|
|
self.core_gui.NextPage()
|
|
self.ShowGoodFinishedDialog()
|
|
else:
|
|
self.core_gui.NextPage()
|
|
self.ShowBadFinishedDialog(_stderr)
|
|
|
|
def OnRestartButton(self, widget, event):
|
|
self.OnStartButton(None, event)
|
|
|
|
def ManualStart(self):
|
|
self.OnStartButton(None, None)
|
|
|
|
def OnCloseButton(self, widget, event):
|
|
self.core_gui.Destroy()
|
|
sys.exit()
|
|
|
|
def ShowGoodFinishedDialog(self):
|
|
self.ShowDialog(i18n.translate("execution_finished"),
|
|
i18n.translate('success_message'),
|
|
wx.ICON_INFORMATION)
|
|
|
|
def ShowBadFinishedDialog(self, error_msg):
|
|
msg = i18n.translate('uh_oh').format(error_msg)
|
|
self.ShowDialog(i18n.translate('error_title'), msg, wx.ICON_ERROR)
|
|
|
|
|
|
def ShowDialog(self, title, content, style):
|
|
a = wx.MessageDialog(None, content, title, style)
|
|
a.ShowModal()
|
|
a.Destroy()
|
|
|