def setup(self):
#set up remote Database
self.connection_string = "sqlite:///:memory:"
self.series_service = SeriesService(connection_string=self.connection_string, debug=False)
self.session = self.series_service._session_factory.get_session()
engine = self.series_service._session_factory.engine
test_util.build_db(engine)
self.dvs_size = 100
self.series = test_util.add_series_bulk_data(self.session, dvs_size=self.dvs_size)
assert self.series
assert len(self.series.data_values) == self.dvs_size
self.memory_database = MemoryDatabase()
self.memory_database.set_series_service(self.series_service)
self.memory_database.initEditValues(self.series.id)
self.app = wx.App()
self.frame = wx.Frame(None)
self.dataTable = FrmDataTable(self.frame)
class frmODMToolsMain(wx.Frame):
"""
"""
def __init__(self, **kwargs):
"""
"""
self.taskserver = kwargs.pop('taskServer')
self.memDB = kwargs.pop('memdb')
# Determine the optimal size of the screen resolution
size = self._obtainScreenResolution()
kwargs['size'] = size
wx.Frame.__init__(self, **kwargs)
self.service_manager = ServiceManager()
self.record_service = None
self.scriptcreate = False
series_service = self._init_database()
if series_service:
self._init_ctrls(series_service)
self._init_aui_manager()
self._init_sizers()
self._ribbon.Realize()
self.Refresh()
logger.info("System starting ...")
else:
logger.info("System shutting down... ")
sys.exit(0)
def _obtainScreenResolution(self):
""" Calculates the size of ODMTools. Prevents ODMTools being larger than the available screen size
typically a problem on laptops
:return wx.Size:
"""
defaultSize = wx.Size(850, 800)
defaultHeight, defaultWidth = defaultSize
screenHeight, screenWidth = wx.GetDisplaySize()
minimumAllowedSize = wx.Size(640, 480)
'''
if minimumAllowedSize >= wx.GetDisplaySize():
logger.fatal("ODMTools cannot be displayed in this resolution: %s \n\tPlease use a larger resolution"
% wx.GetDisplaySize())
print "minimumAllowedsize: ", minimumAllowedSize, "display: ", wx.GetDisplaySize()
sys.exit(0)
'''
newSize = defaultSize
## Screen size is greater than ODMTools' default size
if screenHeight > defaultHeight and screenWidth > defaultWidth:
pass
## Screen size is smaller than ODMTools' default size
elif screenHeight < defaultHeight and screenWidth < defaultWidth:
newSize = wx.Size(screenHeight / 1.5, screenWidth / 1.5)
elif screenHeight < defaultHeight:
newSize = wx.Size(screenHeight / 1.5, defaultWidth)
elif screenWidth < defaultWidth:
newSize = wx.Size(defaultHeight, screenWidth / 1.5)
logger.debug("ODMTools Window Size: %s" % newSize)
return newSize
#############Entire Form Sizers##########
def _init_sizers(self):
# generated method, don't edit
self.s = wx.BoxSizer(wx.VERTICAL)
self._init_s_Items(self.s)
self.SetSizer(self.s)
def _init_s_Items(self, parent):
# generated method, don't edit
parent.AddWindow(self._ribbon, 0, wx.EXPAND)
parent.AddWindow(self.pnlDocking, 85, flag=wx.ALL | wx.EXPAND)
def _init_database(self, quit_if_cancel=True):
logger.info("Loading Database...")
while True:
## Database connection is valid, therefore proceed through the rest of the program
if self.service_manager.is_valid_connection():
conn_dict = None
series_service = self.createService()
conn_dict = self.service_manager.get_current_conn_dict()
if self.servicesValid(series_service):
self.service_manager.add_connection(conn_dict)
break
db_config = frmDBConfig.frmDBConfig(None, self.service_manager, False)
value = db_config.ShowModal()
if value == wx.ID_CANCEL and quit_if_cancel:
logger.fatal("ODMTools is now closing because there is no database connection.")
sys.exit(0)
elif not quit_if_cancel:
return series_service
newConnection = db_config.panel.getFieldValues()
self.service_manager.set_current_conn_dict(newConnection)
db_config.Destroy()
conn_dict = self.service_manager.get_current_conn_dict()
msg = '%s://%s@%s/%s' % (
conn_dict['engine'], conn_dict['user'], conn_dict['address'], conn_dict['db']
)
logger.debug("...Connected to '%s'" % msg)
return series_service
def onChangeDBConn(self, event):
db_config = frmDBConfig.frmDBConfig(None, self.service_manager, False)
value = db_config.ShowModal()
if value == wx.ID_CANCEL:
return
newConnection = db_config.panel.getFieldValues()
self.service_manager.set_current_conn_dict(newConnection)
db_config.Destroy()
if self._init_database(quit_if_cancel=False):
# if editing, stop editing...
if self._ribbon.getEditStatus():
self.stopEdit(event=None)
if value == wx.ID_OK:
series_service = self.createService(newConnection)
self.pnlSelector.resetDB(series_service)
self.refreshConnectionInfo()
self.pnlPlot.clear()
self.dataTable.clear()
def servicesValid(self, service, displayMsg=True):
"""
:param displayMsg:
Option to display a message box if there is an issue with a service. Default: True
:return:
"""
valid = True
## Test if Series Catalog is empty
if not service.get_used_sites():
if displayMsg:
msg = wx.MessageDialog(None, 'Series Catalog cannot be empty. Please enter in a new database connection',
'Series Catalog is empty', wx.OK | wx.ICON_ERROR )
msg.ShowModal()
valid = False
# @TODO If Jeff runs into other issues with services not being available, we can simply test different services here
#if not service.get_all_variables():
# valid = False
return valid
def on_about_request(self, event):
frmAbout(self)
def MacReopenApp(self):
"""Called when the doc icon is clicked, and ???"""
try: # it's possible for this event to come when the frame is closed
self.GetTopWindow().Raise()
except:
pass
###################### Frame ################
def _init_ctrls(self, series_service):
# generated method, don't edit
logger.debug("Loading frame...")
self.SetIcon(gtk_execute.getIcon())
self.SetFont(wx.Font(9, wx.SWISS, wx.NORMAL, wx.NORMAL,
False, u'Tahoma'))
############### Ribbon ###################
logger.debug("Loading Ribbon Menu...")
self._ribbon = mnuRibbon.mnuRibbon(parent=self, id=wx.ID_ANY, name='ribbon')
if sys.platform == 'darwin':
self.menu_bar = wx.MenuBar()
self.help_menu = wx.Menu()
self.help_menu.Append(wx.ID_ABOUT, "&About ODMTools")
self.menu_bar.Append(self.help_menu, "&Help")
self.SetMenuBar(self.menu_bar)
self.Bind(wx.EVT_MENU, self.on_about_request, id=wx.ID_ABOUT)
# self.menu_bar.SetAutoWindowMenu()
################ Docking Tools##############
self.pnlDocking = wx.Panel(name='pnlDocking',
parent=self, size=wx.Size(605, 458),
style=wx.TAB_TRAVERSAL)
############# Graph ###############
logger.debug("Loading Plot ...")
self.pnlPlot = pnlPlot.pnlPlot(self.pnlDocking, self.taskserver)
################ Series Selection Panel ##################
logger.debug("Loading Series Selector ...")
self.pnlSelector = FrmSeriesSelector(self.pnlDocking, series_service, plot=self.pnlPlot, taskserver=self.taskserver, memdb = self.memDB)
####################grid Table View##################
logger.debug("Loading DataTable ...")
self.dataTable = FrmDataTable(self.pnlDocking)
# self.dataTable.toggleBindings()
############# Script & Console ###############
logger.debug("Loading Python Console ...")
self.txtPythonConsole = ODMToolsConsole(parent=self.pnlDocking, size=wx.Size(200, 200))
wx.CallAfter(self._postStartup)
logger.debug("Loading Python Script ...")
self.txtPythonScript = pnlScript(name=u'txtPython', parent=self,
size=wx.Size(200, 200))
self.Bind(wx.EVT_CLOSE, self.onClose)
Publisher.subscribe(self.onDocking, ("adjust.Docking"))
Publisher.subscribe(self.onPlotSelection, ("select.Plot"))
Publisher.subscribe(self.onExecuteScript, ("execute.script"))
Publisher.subscribe(self.onChangeDBConn, ("change.dbConfig"))
Publisher.subscribe(self.onSetScriptTitle, ("script.title"))
Publisher.subscribe(self.onClose, ("onClose"))
Publisher.subscribe(self.addEdit, ("selectEdit"))
Publisher.subscribe(self.stopEdit, ("stopEdit"))
def _init_aui_manager(self):
############ Docking ###################
logger.debug("Loading AuiManager ...")
self._mgr = aui.AuiManager()
self._mgr.SetManagedWindow(self.pnlDocking)
self._mgr.AddPane(self.pnlPlot, aui.AuiPaneInfo().CenterPane()
.Name("Plot").Caption("Plot").MaximizeButton(True).DestroyOnClose(False)
)
self._mgr.AddPane(self.dataTable, aui.AuiPaneInfo().Right().Name("Table").
Show(show=False).Caption('Table View').MinSize(wx.Size(200, 200)).Floatable().Movable().
Position(1).MinimizeButton(True).MaximizeButton(True).DestroyOnClose(False)
)
self._mgr.AddPane(self.pnlSelector, aui.AuiPaneInfo().Bottom().Name("Selector").MinSize(wx.Size(50, 200)).
Movable().Floatable().Position(0).MinimizeButton(True).MaximizeButton(True).CloseButton(True)
.DestroyOnClose(False)
)
self._mgr.AddPane(self.txtPythonScript, aui.AuiPaneInfo().Caption('Script').
Name("Script").Movable().Floatable().Right()
.MinimizeButton(True).MaximizeButton(True).FloatingSize(size=(400, 400))
.CloseButton(True).Float().FloatingPosition(pos=(self.Position))
.Hide().CloseButton(True).DestroyOnClose(False)
)
self._mgr.AddPane(self.txtPythonConsole, aui.AuiPaneInfo().Caption('Python Console').
Name("Console").FloatingSize(size=(300, 400)).MinimizeButton(
True).Movable().Floatable().MaximizeButton(True).CloseButton(True).Float()
.FloatingPosition(pos=(self.Position)).Show(show=False).DestroyOnClose(False)
)
## TODO Fix loadingDockingSettings as it doesn't load it correctly.
# self.loadDockingSettings()
self.refreshConnectionInfo()
self._mgr.Update()
def refreshConnectionInfo(self):
"""Updates the Series Selector Connection Information for the user"""
conn_dict = self.service_manager.get_current_conn_dict()
msg = 'Series: %s://%s@%s/%s' % (
conn_dict['engine'], conn_dict['user'], conn_dict['address'], conn_dict['db']
)
self._mgr.GetPane('Selector').Caption(msg)
self._mgr.RefreshCaptions()
def onDocking(self, value):
paneDetails = self._mgr.GetPane(self.pnlPlot)
if value == "Table":
paneDetails = self._mgr.GetPane(self.dataTable)
# self.dataTable.toggleBindings()
elif value == "Selector":
paneDetails = self._mgr.GetPane(self.pnlSelector)
elif value == "Script":
paneDetails = self._mgr.GetPane(self.txtPythonScript)
elif value == "Console":
paneDetails = self._mgr.GetPane(self.txtPythonConsole)
self.txtPythonConsole.crust.OnSashDClick(event=None)
if paneDetails.IsNotebookPage() or paneDetails.dock_direction == 0:
paneDetails.FloatingPosition(pos=self.Position)
paneDetails.Float()
if paneDetails.IsShown():
paneDetails.Show(show=False)
else:
paneDetails.Show(show=True)
self._mgr.Update()
def getDBService(self):
return self.service_manager
def onPlotSelection(self, value):
self.pnlPlot.selectPlot(value)
def onSetScriptTitle(self, title):
scriptPane = self._mgr.GetPane(self.txtPythonScript)
scriptPane.Caption(title)
if scriptPane.IsFloating():
scriptPane.Restore()
self._mgr.Update()
def addEdit(self, event):
with wx.BusyInfo("Please wait for a moment while ODMTools fetches the data and stores it in our database", parent=self):
self.scriptcreate = True
isSelected, seriesID = self.pnlSelector.onReadyToEdit()
logger.info("Beginning editing seriesID: %s"%str(seriesID))
# logger.debug("Initializing DataTable")
# # tasks = [("dataTable", (memDB.conn, self.dataTable.myOlv))]
# tasks = [("dataTable", (self.dataTable.myOlv))]
# self.taskserver.setTasks(tasks)
# self.taskserver.processTasks()
if isSelected:
self.record_service = self.service_manager.get_record_service(self.txtPythonScript, seriesID,
connection=self.memDB)
self._ribbon.toggleEditButtons(True)
logger.debug("Initializing Plot")
self.pnlPlot.addEditPlot(self.memDB, seriesID, self.record_service)
logger.debug("Initializing DataTable")
self.dataTable.init(self.memDB)
# set record service for console
Publisher.sendMessage("setEdit", isEdit=True)
logger.debug("Enabling Edit")
self.record_service.toggle_record(True)
# set the cursor for matplotlib
selectedObject = self.record_service.get_series()
Publisher.sendMessage("updateCursor", selectedObject=selectedObject)
else:
logger.debug("disabling Edit")
Publisher.sendMessage("setEdit", isEdit=False)
self.record_service.toggle_record(False)
# disable cursor for matplotlib
selectedObject = self.record_service.get_series()
Publisher.sendMessage("updateCursor", deselectedObject=selectedObject)
# self._mgr.Update()
logger.debug("Recording? %s" % self.record_service._record)
#self.record_service = None
self.txtPythonConsole.shell.run("edit_service = app.TopWindow.record_service", prompt=False, verbose=False)
self.txtPythonConsole.shell.run("series_service = edit_service.get_series_service()", prompt=False,
verbose=False)
#from meliae import scanner
#scanner.dump_all_objects("edit_plotting.dat")
logger.info("Finished Setting up Editing Series: %s " % seriesID)
def stopEdit(self, event):
self.pnlSelector.stopEdit()
self.dataTable.stopEdit()
self.pnlPlot.stopEdit()
Publisher.sendMessage("toggleEdit", checked=False)
self.memDB.reset_edit()
self.record_service = None
self._ribbon.toggleEditButtons(False)
def getRecordService(self):
return self.record_service
def createService(self, conn_dict=""):
"""
:param conn_dict:
Provides the ability to send in your own conn_dict instead
of relying on reading one in from connection.config
:return:
"""
series_service = self.service_manager.get_series_service(conn_dict=conn_dict)
return series_service
def getServiceManager(self):
return self.service_manager
def toggleConsoleTools(self):
self.txtPythonConsole.ToggleTools()
def onExecuteScript(self, value):
for i in ('red', 'blue', 'green', 'magenta', 'gold', 'cyan', 'brown', 'lime', 'purple', 'navy'):
self.txtPythonScript('This is a test\n', i)
def loadDockingSettings(self):
# test if there is a perspective to load
try:
# TODO Fix resource_path to appdirs
os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config')
f = open(os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config'), 'r')
except:
# Create the file if it doesn't exist
open(os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config'), 'w').close()
f = open(os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config'), 'r')
self._mgr.LoadPerspective(f.read(), True)
def onClose(self, event):
"""
Closes ODMTools Python
Closes AUI Manager then closes MainWindow
"""
#check to see if a script has been created
if self.scriptcreate:
msg = wx.MessageDialog(None, 'Would you like to save your editing script?',
'Save Script', wx.YES_NO | wx.ICON_QUESTION)
value = msg.ShowModal()
if value == wx.ID_YES:
self.txtPythonScript.OnSaveAs(event)
# deinitialize the frame manager
self.pnlPlot.Close()
try:
f = open(os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config'), 'w')
f.write(self._mgr.SavePerspective())
except:
print "error saving docking data"
self._mgr.UnInit()
# Shut down processes running in background
if self.taskserver.numprocesses > 0 and self.taskserver.anyAlive:
busy = wx.BusyInfo("Closing ODMTools ...", parent=self)
# Terminate the processes
self.taskserver.processTerminate()
# IMPORTANT! if wx.TaskBarIcons exist, it will keep mainloop running
windowsRemaining = len(wx.GetTopLevelWindows())
if windowsRemaining > 0:
import wx.lib.agw.aui.framemanager as aui
# logger.debug("Windows left to close: %d" % windowsRemaining)
for item in wx.GetTopLevelWindows():
# logger.debug("Windows %s" % item)
if not isinstance(item, self.__class__):
if isinstance(item, aui.AuiFloatingFrame):
item.Destroy()
elif isinstance(item, aui.AuiSingleDockingGuide):
item.Destroy()
elif isinstance(item, aui.AuiDockingHintWindow):
item.Destroy()
elif isinstance(item, wx.Dialog):
item.Destroy()
item.Close()
logger.info("Closing ODMTools\n")
self.Destroy()
wx.GetApp().ExitMainLoop()
def _postStartup(self):
"""
Called after MainLoop is initialized
Hides Python Console Tools
"""
if self.txtPythonConsole.ToolsShown():
self.txtPythonConsole.ToggleTools()
self.txtPythonConsole.shell.run("import datetime", prompt=False, verbose=False)
self.txtPythonConsole.shell.run("edit_service = app.TopWindow.record_service", prompt=False, verbose=False)
def _init_ctrls(self, series_service):
# generated method, don't edit
logger.debug("Loading frame...")
self.SetIcon(gtk_execute.getIcon())
self.SetFont(wx.Font(9, wx.SWISS, wx.NORMAL, wx.NORMAL,
False, u'Tahoma'))
############### Ribbon ###################
logger.debug("Loading Ribbon Menu...")
self._ribbon = mnuRibbon.mnuRibbon(parent=self, id=wx.ID_ANY, name='ribbon')
if sys.platform == 'darwin':
self.menu_bar = wx.MenuBar()
self.help_menu = wx.Menu()
self.help_menu.Append(wx.ID_ABOUT, "&About ODMTools")
self.menu_bar.Append(self.help_menu, "&Help")
self.SetMenuBar(self.menu_bar)
self.Bind(wx.EVT_MENU, self.on_about_request, id=wx.ID_ABOUT)
# self.menu_bar.SetAutoWindowMenu()
################ Docking Tools##############
self.pnlDocking = wx.Panel(name='pnlDocking',
parent=self, size=wx.Size(605, 458),
style=wx.TAB_TRAVERSAL)
############# Graph ###############
logger.debug("Loading Plot ...")
self.pnlPlot = pnlPlot.pnlPlot(self.pnlDocking, self.taskserver)
################ Series Selection Panel ##################
logger.debug("Loading Series Selector ...")
self.pnlSelector = FrmSeriesSelector(self.pnlDocking, series_service, plot=self.pnlPlot, taskserver=self.taskserver, memdb = self.memDB)
####################grid Table View##################
logger.debug("Loading DataTable ...")
self.dataTable = FrmDataTable(self.pnlDocking)
# self.dataTable.toggleBindings()
############# Script & Console ###############
logger.debug("Loading Python Console ...")
self.txtPythonConsole = ODMToolsConsole(parent=self.pnlDocking, size=wx.Size(200, 200))
wx.CallAfter(self._postStartup)
logger.debug("Loading Python Script ...")
self.txtPythonScript = pnlScript(name=u'txtPython', parent=self,
size=wx.Size(200, 200))
self.Bind(wx.EVT_CLOSE, self.onClose)
Publisher.subscribe(self.onDocking, ("adjust.Docking"))
Publisher.subscribe(self.onPlotSelection, ("select.Plot"))
Publisher.subscribe(self.onExecuteScript, ("execute.script"))
Publisher.subscribe(self.onChangeDBConn, ("change.dbConfig"))
Publisher.subscribe(self.onSetScriptTitle, ("script.title"))
Publisher.subscribe(self.onClose, ("onClose"))
Publisher.subscribe(self.addEdit, ("selectEdit"))
Publisher.subscribe(self.stopEdit, ("stopEdit"))
class TestPnlDataTable:
def setup(self):
#set up remote Database
self.connection_string = "sqlite:///:memory:"
self.series_service = SeriesService(connection_string=self.connection_string, debug=False)
self.session = self.series_service._session_factory.get_session()
engine = self.series_service._session_factory.engine
test_util.build_db(engine)
self.dvs_size = 100
self.series = test_util.add_series_bulk_data(self.session, dvs_size=self.dvs_size)
assert self.series
assert len(self.series.data_values) == self.dvs_size
self.memory_database = MemoryDatabase()
self.memory_database.set_series_service(self.series_service)
self.memory_database.initEditValues(self.series.id)
self.app = wx.App()
self.frame = wx.Frame(None)
self.dataTable = FrmDataTable(self.frame)
def test_build_series(self):
dvs = self.session.query(DataValue).all()
assert len(dvs) == self.dvs_size
dvs = self.memory_database.mem_service._edit_session.query(DataValue).all()
assert len(dvs) == self.dvs_size
def test_get_data_values_data_frame(self):
df = self.memory_database.getDataValuesDF()
assert not df.empty
def test_init_pnlDataTable(self):
assert self.frame
assert self.dataTable
self.dataTable.init(self.memory_database)
def test_selecting_points(self):
self.dataTable.init(self.memory_database)
values = self.dataTable.olvDataTable.dataframe
assert not values.empty
self.dataTable.onChangeSelection(values)
myOlv = self.dataTable.olvDataTable
count = 0
selected_item = myOlv.GetFirstSelected()
assert selected_item != -1
# loop through selected items
while selected_item != -1:
selected_item = myOlv.GetNextSelected(selected_item)
count += 1
assert count == self.dvs_size
def test_deselecting_all(self):
self.dataTable.init(self.memory_database)
assert self.dataTable.olvDataTable.GetItemCount() == self.dvs_size
values = self.dataTable.olvDataTable.dataframe
self.dataTable.onChangeSelection(values)
self.dataTable.olvDataTable.onDeselectAll()
selected_item = self.dataTable.olvDataTable.GetFirstSelected()
assert selected_item == -1
def test_clear_data_table(self):
self.dataTable.init(self.memory_database)
assert self.dataTable.olvDataTable.GetItemCount() == self.dvs_size
self.dataTable.clear()
assert not self.dataTable.olvDataTable.dataframe
assert self.dataTable.olvDataTable.GetItemCount() == 0
class frmODMToolsMain(wx.Frame):
"""
"""
def __init__(self, **kwargs):
"""
"""
self.taskserver = kwargs.pop('taskServer')
self.memDB = kwargs.pop('memdb')
# Determine the optimal size of the screen resolution
size = self._obtainScreenResolution()
kwargs['size'] = size
wx.Frame.__init__(self, **kwargs)
self.service_manager = ServiceManager()
self.record_service = None
series_service = self._init_database()
if series_service:
self._init_ctrls(series_service)
self._init_aui_manager()
self._init_sizers()
self._ribbon.Realize()
self.Refresh()
logger.debug("System starting ...")
else:
logger.debug("System shutting down... ")
sys.exit(0)
def _obtainScreenResolution(self):
""" Calculates the size of ODMTools. Prevents ODMTools being larger than the available screen size
typically a problem on laptops
:return wx.Size:
"""
defaultSize = wx.Size(850, 800)
defaultHeight, defaultWidth = defaultSize
screenHeight, screenWidth = wx.GetDisplaySize()
minimumAllowedSize = wx.Size(640, 480)
'''
if minimumAllowedSize >= wx.GetDisplaySize():
logger.fatal("ODMTools cannot be displayed in this resolution: %s \n\tPlease use a larger resolution"
% wx.GetDisplaySize())
print "minimumAllowedsize: ", minimumAllowedSize, "display: ", wx.GetDisplaySize()
sys.exit(0)
'''
newSize = defaultSize
## Screen size is greater than ODMTools' default size
if screenHeight > defaultHeight and screenWidth > defaultWidth:
pass
## Screen size is smaller than ODMTools' default size
elif screenHeight < defaultHeight and screenWidth < defaultWidth:
newSize = wx.Size(screenHeight / 1.5, screenWidth / 1.5)
elif screenHeight < defaultHeight:
newSize = wx.Size(screenHeight / 1.5, defaultWidth)
elif screenWidth < defaultWidth:
newSize = wx.Size(defaultHeight, screenWidth / 1.5)
logger.debug("ODMTools Window Size: %s" % newSize)
return newSize
#############Entire Form Sizers##########
def _init_sizers(self):
# generated method, don't edit
self.s = wx.BoxSizer(wx.VERTICAL)
self._init_s_Items(self.s)
self.SetSizer(self.s)
def _init_s_Items(self, parent):
# generated method, don't edit
parent.AddWindow(self._ribbon, 0, wx.EXPAND)
parent.AddWindow(self.pnlDocking, 85, flag=wx.ALL | wx.EXPAND)
def _init_database(self, quit_if_cancel=True):
logger.debug("Loading Database...")
while True:
## Database connection is valid, therefore proceed through the rest of the program
if self.service_manager.is_valid_connection():
conn_dict = None
series_service = self.createService()
conn_dict = self.service_manager.get_current_conn_dict()
if self.servicesValid(series_service):
self.service_manager.add_connection(conn_dict)
break
db_config = frmDBConfig.frmDBConfig(None, self.service_manager, False)
value = db_config.ShowModal()
if value == wx.ID_CANCEL and quit_if_cancel:
logger.fatal("ODMTools is now closing because there is no database connection.")
sys.exit(0)
elif not quit_if_cancel:
return series_service
newConnection = db_config.panel.getFieldValues()
self.service_manager.set_current_conn_dict(newConnection)
db_config.Destroy()
conn_dict = self.service_manager.get_current_conn_dict()
msg = '%s://%s@%s/%s' % (
conn_dict['engine'], conn_dict['user'], conn_dict['address'], conn_dict['db']
)
logger.debug("...Connected to '%s'" % msg)
return series_service
def onChangeDBConn(self, event):
db_config = frmDBConfig.frmDBConfig(None, self.service_manager, False)
value = db_config.ShowModal()
if value == wx.ID_CANCEL:
return
newConnection = db_config.panel.getFieldValues()
self.service_manager.set_current_conn_dict(newConnection)
db_config.Destroy()
if self._init_database(quit_if_cancel=False):
# if editing, stop editing...
if self._ribbon.getEditStatus():
self.stopEdit(event=None)
if value == wx.ID_OK:
series_service = self.createService(newConnection)
self.pnlSelector.resetDB(series_service)
self.refreshConnectionInfo()
self.pnlPlot.clear()
self.dataTable.clear()
def servicesValid(self, service, displayMsg=True):
"""
:param displayMsg:
Option to display a message box if there is an issue with a service. Default: True
:return:
"""
valid = True
## Test if Series Catalog is empty
if not service.get_used_sites():
if displayMsg:
msg = wx.MessageDialog(None, 'Series Catalog cannot be empty. Please enter in a new database connection',
'Series Catalog is empty', wx.OK | wx.ICON_ERROR )
msg.ShowModal()
valid = False
# @TODO If Jeff runs into other issues with services not being available, we can simply test different services here
#if not service.get_all_variables():
# valid = False
return valid
def on_about_request(self, event):
frmAbout(self)
def MacReopenApp(self):
"""Called when the doc icon is clicked, and ???"""
try: # it's possible for this event to come when the frame is closed
self.GetTopWindow().Raise()
except:
pass
###################### Frame ################
def _init_ctrls(self, series_service):
# generated method, don't edit
logger.debug("Loading frame...")
self.SetIcon(gtk_execute.getIcon())
self.SetFont(wx.Font(9, wx.SWISS, wx.NORMAL, wx.NORMAL,
False, u'Tahoma'))
############### Ribbon ###################
logger.debug("Loading Ribbon Menu...")
self._ribbon = mnuRibbon.mnuRibbon(parent=self, id=wx.ID_ANY, name='ribbon')
if sys.platform == 'darwin':
self.menu_bar = wx.MenuBar()
self.help_menu = wx.Menu()
self.help_menu.Append(wx.ID_ABOUT, "&About ODMTools")
self.menu_bar.Append(self.help_menu, "&Help")
self.SetMenuBar(self.menu_bar)
self.Bind(wx.EVT_MENU, self.on_about_request, id=wx.ID_ABOUT)
# self.menu_bar.SetAutoWindowMenu()
################ Docking Tools##############
self.pnlDocking = wx.Panel(name='pnlDocking',
parent=self, size=wx.Size(605, 458),
style=wx.TAB_TRAVERSAL)
############# Graph ###############
logger.debug("Loading Plot ...")
self.pnlPlot = pnlPlot.pnlPlot(self.pnlDocking, self.taskserver)
################ Series Selection Panel ##################
logger.debug("Loading Series Selector ...")
self.pnlSelector = FrmSeriesSelector(self.pnlDocking, series_service, plot=self.pnlPlot, taskserver=self.taskserver, memdb = self.memDB)
####################grid Table View##################
logger.debug("Loading DataTable ...")
self.dataTable = FrmDataTable(self.pnlDocking)
# self.dataTable = pnlDataTable.pnlDataTable(self.pnlDocking)
# self.dataTable.toggleBindings()
############# Script & Console ###############
logger.debug("Loading Python Console ...")
self.txtPythonConsole = ODMToolsConsole(parent=self.pnlDocking, size=wx.Size(200, 200))
wx.CallAfter(self._postStartup)
logger.debug("Loading Python Script ...")
self.txtPythonScript = pnlScript(name=u'txtPython', parent=self,
size=wx.Size(200, 200))
self.Bind(wx.EVT_CLOSE, self.onClose)
Publisher.subscribe(self.onDocking, ("adjust.Docking"))
Publisher.subscribe(self.onPlotSelection, ("select.Plot"))
Publisher.subscribe(self.onExecuteScript, ("execute.script"))
Publisher.subscribe(self.onChangeDBConn, ("change.dbConfig"))
Publisher.subscribe(self.onSetScriptTitle, ("script.title"))
#.subscribe(self.onSetScriptTitle, ("script.title"))
Publisher.subscribe(self.onClose, ("onClose"))
Publisher.subscribe(self.addEdit, ("selectEdit"))
Publisher.subscribe(self.stopEdit, ("stopEdit"))
def _init_aui_manager(self):
############ Docking ###################
logger.debug("Loading AuiManager ...")
self._mgr = aui.AuiManager()
self._mgr.SetManagedWindow(self.pnlDocking)
self._mgr.AddPane(self.pnlPlot, aui.AuiPaneInfo().CenterPane()
.Name("Plot").Caption("Plot").MaximizeButton(True).DestroyOnClose(False)
)
self._mgr.AddPane(self.dataTable, aui.AuiPaneInfo().Right().Name("Table").
Show(show=False).Caption('Table View').MinSize(wx.Size(200, 200)).Floatable().Movable().
Position(1).MinimizeButton(True).MaximizeButton(True).DestroyOnClose(False)
)
self._mgr.AddPane(self.pnlSelector, aui.AuiPaneInfo().Bottom().Name("Selector").MinSize(wx.Size(50, 200)).
Movable().Floatable().Position(0).MinimizeButton(True).MaximizeButton(True).CloseButton(True)
.DestroyOnClose(False)
)
self._mgr.AddPane(self.txtPythonScript, aui.AuiPaneInfo().Caption('Script').
Name("Script").Movable().Floatable().Right()
.MinimizeButton(True).MaximizeButton(True).FloatingSize(size=(400, 400))
.CloseButton(True).Float().FloatingPosition(pos=(self.Position))
.Hide().CloseButton(True).DestroyOnClose(False)
)
self._mgr.AddPane(self.txtPythonConsole, aui.AuiPaneInfo().Caption('Python Console').
Name("Console").FloatingSize(size=(300, 400)).MinimizeButton(
True).Movable().Floatable().MaximizeButton(True).CloseButton(True).Float()
.FloatingPosition(pos=(self.Position)).Show(show=False).DestroyOnClose(False)
)
## TODO Fix loadingDockingSettings as it doesn't load it correctly.
# self.loadDockingSettings()
self.refreshConnectionInfo()
self._mgr.Update()
def refreshConnectionInfo(self):
"""Updates the Series Selector Connection Information for the user"""
conn_dict = self.service_manager.get_current_conn_dict()
msg = 'Series: %s://%s@%s/%s' % (
conn_dict['engine'], conn_dict['user'], conn_dict['address'], conn_dict['db']
)
self._mgr.GetPane('Selector').Caption(msg)
self._mgr.RefreshCaptions()
def onDocking(self, value):
paneDetails = self._mgr.GetPane(self.pnlPlot)
if value == "Table":
paneDetails = self._mgr.GetPane(self.dataTable)
# self.dataTable.toggleBindings()
elif value == "Selector":
paneDetails = self._mgr.GetPane(self.pnlSelector)
elif value == "Script":
paneDetails = self._mgr.GetPane(self.txtPythonScript)
elif value == "Console":
paneDetails = self._mgr.GetPane(self.txtPythonConsole)
self.txtPythonConsole.crust.OnSashDClick(event=None)
if paneDetails.IsNotebookPage() or paneDetails.dock_direction == 0:
paneDetails.FloatingPosition(pos=self.Position)
paneDetails.Float()
if paneDetails.IsShown():
paneDetails.Show(show=False)
else:
paneDetails.Show(show=True)
self._mgr.Update()
def getDBService(self):
return self.service_manager
def onPlotSelection(self, value):
self.pnlPlot.selectPlot(value)
def onSetScriptTitle(self, title):
scriptPane = self._mgr.GetPane(self.txtPythonScript)
scriptPane.Caption(title)
if scriptPane.IsFloating():
scriptPane.Restore()
self._mgr.Update()
def addEdit(self, event):
with wx.BusyInfo("Please wait for a moment while ODMTools fetches the data and stores it in our database", parent=self):
logger.debug("Beginning editing")
isSelected, seriesID = self.pnlSelector.onReadyToEdit()
# logger.debug("Initializing DataTable")
# # tasks = [("dataTable", (memDB.conn, self.dataTable.myOlv))]
# tasks = [("dataTable", (self.dataTable.myOlv))]
# self.taskserver.setTasks(tasks)
# self.taskserver.processTasks()
if isSelected:
self.record_service = self.service_manager.get_record_service(self.txtPythonScript, seriesID,
connection=self.memDB)
self._ribbon.toggleEditButtons(True)
logger.debug("Initializing Plot")
self.pnlPlot.addEditPlot(self.memDB, seriesID, self.record_service)
logger.debug("Initializing DataTable")
self.dataTable.init(self.memDB)
# set record service for console
Publisher.sendMessage("setEdit", isEdit=True)
logger.debug("Enabling Edit")
self.record_service.toggle_record(True)
# set the cursor for matplotlib
selectedObject = self.record_service.get_series()
Publisher.sendMessage("updateCursor", selectedObject=selectedObject)
else:
logger.debug("disabling Edit")
Publisher.sendMessage("setEdit", isEdit=False)
self.record_service.toggle_record(False)
# disable cursor for matplotlib
selectedObject = self.record_service.get_series()
Publisher.sendMessage("updateCursor", deselectedObject=selectedObject)
# self._mgr.Update()
logger.debug("Recording? %s" % self.record_service._record)
#self.record_service = None
self.txtPythonConsole.shell.run("edit_service = app.TopWindow.record_service", prompt=False, verbose=False)
self.txtPythonConsole.shell.run("series_service = edit_service.get_series_service()", prompt=False,
verbose=False)
#from meliae import scanner
#scanner.dump_all_objects("edit_plotting.dat")
logger.info("Finished Setting up Editing Series: %s " % seriesID)
def stopEdit(self, event):
self.pnlSelector.stopEdit()
self.dataTable.stopEdit()
self.pnlPlot.stopEdit()
Publisher.sendMessage("toggleEdit", checked=False)
self.record_service = None
self._ribbon.toggleEditButtons(False)
def getRecordService(self):
return self.record_service
def createService(self, conn_dict=""):
"""
:param conn_dict:
Provides the ability to send in your own conn_dict instead
of relying on reading one in from connection.config
:return:
"""
series_service = self.service_manager.get_series_service(conn_dict=conn_dict)
return series_service
def getServiceManager(self):
return self.service_manager
def toggleConsoleTools(self):
self.txtPythonConsole.ToggleTools()
def onExecuteScript(self, value):
for i in ('red', 'blue', 'green', 'magenta', 'gold', 'cyan', 'brown', 'lime', 'purple', 'navy'):
self.txtPythonScript('This is a test\n', i)
def loadDockingSettings(self):
# test if there is a perspective to load
try:
# TODO Fix resource_path to appdirs
os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config')
f = open(os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config'), 'r')
except:
# Create the file if it doesn't exist
open(os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config'), 'w').close()
f = open(os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config'), 'r')
self._mgr.LoadPerspective(f.read(), True)
def onClose(self, event):
"""
Closes ODMTools Python
Closes AUI Manager then closes MainWindow
"""
# deinitialize the frame manager
self.pnlPlot.Close()
try:
f = open(os.path.join(user_config_dir("ODMTools", "UCHIC"), 'ODMTools.config'), 'w')
f.write(self._mgr.SavePerspective())
except:
print "error saving docking data"
self._mgr.UnInit()
# Shut down processes running in background
if self.taskserver.numprocesses > 0 and self.taskserver.anyAlive:
busy = wx.BusyInfo("Closing ODMTools ...", parent=self)
# Terminate the processes
self.taskserver.processTerminate()
# IMPORTANT! if wx.TaskBarIcons exist, it will keep mainloop running
windowsRemaining = len(wx.GetTopLevelWindows())
if windowsRemaining > 0:
import wx.lib.agw.aui.framemanager as aui
# logger.debug("Windows left to close: %d" % windowsRemaining)
for item in wx.GetTopLevelWindows():
# logger.debug("Windows %s" % item)
if not isinstance(item, self.__class__):
if isinstance(item, aui.AuiFloatingFrame):
item.Destroy()
elif isinstance(item, aui.AuiSingleDockingGuide):
item.Destroy()
elif isinstance(item, aui.AuiDockingHintWindow):
item.Destroy()
elif isinstance(item, wx.Dialog):
item.Destroy()
item.Close()
self.Destroy()
wx.GetApp().ExitMainLoop()
def _postStartup(self):
"""
Called after MainLoop is initialized
Hides Python Console Tools
"""
if self.txtPythonConsole.ToolsShown():
self.txtPythonConsole.ToggleTools()
self.txtPythonConsole.shell.run("import datetime", prompt=False, verbose=False)
self.txtPythonConsole.shell.run("edit_service = app.TopWindow.record_service", prompt=False, verbose=False)
def _init_ctrls(self, series_service):
# generated method, don't edit
logger.debug("Loading frame...")
self.SetIcon(gtk_execute.getIcon())
self.SetFont(wx.Font(9, wx.SWISS, wx.NORMAL, wx.NORMAL,
False, u'Tahoma'))
############### Ribbon ###################
logger.debug("Loading Ribbon Menu...")
self._ribbon = mnuRibbon.mnuRibbon(parent=self, id=wx.ID_ANY, name='ribbon')
if sys.platform == 'darwin':
self.menu_bar = wx.MenuBar()
self.help_menu = wx.Menu()
self.help_menu.Append(wx.ID_ABOUT, "&About ODMTools")
self.menu_bar.Append(self.help_menu, "&Help")
self.SetMenuBar(self.menu_bar)
self.Bind(wx.EVT_MENU, self.on_about_request, id=wx.ID_ABOUT)
# self.menu_bar.SetAutoWindowMenu()
################ Docking Tools##############
self.pnlDocking = wx.Panel(name='pnlDocking',
parent=self, size=wx.Size(605, 458),
style=wx.TAB_TRAVERSAL)
############# Graph ###############
logger.debug("Loading Plot ...")
self.pnlPlot = pnlPlot.pnlPlot(self.pnlDocking, self.taskserver)
################ Series Selection Panel ##################
logger.debug("Loading Series Selector ...")
self.pnlSelector = FrmSeriesSelector(self.pnlDocking, series_service, plot=self.pnlPlot, taskserver=self.taskserver, memdb = self.memDB)
####################grid Table View##################
logger.debug("Loading DataTable ...")
self.dataTable = FrmDataTable(self.pnlDocking)
# self.dataTable = pnlDataTable.pnlDataTable(self.pnlDocking)
# self.dataTable.toggleBindings()
############# Script & Console ###############
logger.debug("Loading Python Console ...")
self.txtPythonConsole = ODMToolsConsole(parent=self.pnlDocking, size=wx.Size(200, 200))
wx.CallAfter(self._postStartup)
logger.debug("Loading Python Script ...")
self.txtPythonScript = pnlScript(name=u'txtPython', parent=self,
size=wx.Size(200, 200))
self.Bind(wx.EVT_CLOSE, self.onClose)
Publisher.subscribe(self.onDocking, ("adjust.Docking"))
Publisher.subscribe(self.onPlotSelection, ("select.Plot"))
Publisher.subscribe(self.onExecuteScript, ("execute.script"))
Publisher.subscribe(self.onChangeDBConn, ("change.dbConfig"))
Publisher.subscribe(self.onSetScriptTitle, ("script.title"))
#.subscribe(self.onSetScriptTitle, ("script.title"))
Publisher.subscribe(self.onClose, ("onClose"))
Publisher.subscribe(self.addEdit, ("selectEdit"))
Publisher.subscribe(self.stopEdit, ("stopEdit"))