|
12 | 12 | from odmtools.controller.frmDataTable import FrmDataTable |
13 | 13 | import mnuRibbon |
14 | 14 | import pnlPlot |
| 15 | +import pnlPlot |
15 | 16 | import pnlDataTable |
16 | 17 | import wx.lib.agw.aui as aui |
17 | 18 | import wx.py.crust |
@@ -55,6 +56,7 @@ def __init__(self, **kwargs): |
55 | 56 |
|
56 | 57 | self.service_manager = ServiceManager() |
57 | 58 | self.record_service = None |
| 59 | + self.scriptcreate = False |
58 | 60 |
|
59 | 61 | series_service = self._init_database() |
60 | 62 | if series_service: |
@@ -384,7 +386,7 @@ def onSetScriptTitle(self, title): |
384 | 386 | def addEdit(self, event): |
385 | 387 |
|
386 | 388 | with wx.BusyInfo("Please wait for a moment while ODMTools fetches the data and stores it in our database", parent=self): |
387 | | - |
| 389 | + self.scriptcreate = True |
388 | 390 | isSelected, seriesID = self.pnlSelector.onReadyToEdit() |
389 | 391 | logger.info("Beginning editing seriesID: %s"%str(seriesID)) |
390 | 392 |
|
@@ -496,6 +498,15 @@ def onClose(self, event): |
496 | 498 | Closes ODMTools Python |
497 | 499 | Closes AUI Manager then closes MainWindow |
498 | 500 | """ |
| 501 | + |
| 502 | + #check to see if a script has been created |
| 503 | + if self.scriptcreate: |
| 504 | + msg = wx.MessageDialog(None, 'Would you like to save your editing script', |
| 505 | + 'Save Script', wx.YES_NO | wx.ICON_QUESTION) |
| 506 | + value = msg.ShowModal() |
| 507 | + if value == wx.ID_YES: |
| 508 | + pnlScript.OnSaveAs(event) |
| 509 | + |
499 | 510 | # deinitialize the frame manager |
500 | 511 | self.pnlPlot.Close() |
501 | 512 | try: |
|
0 commit comments