def setup(self):
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.series = test_util.add_series(self.session)
self.memory_db = MemoryDatabase(self.series_service)
class TestMemoryDB:
def setup(self):
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.series = test_util.add_series(self.session)
self.memory_db = MemoryDatabase(self.series_service)
def test_delete_points(self):
with pytest.raises(NotImplementedError):
self.memory_db.delete_points("filter")
def test_add_points(self):
with pytest.raises(NotImplementedError):
self.memory_db.add_points("filter")
def test_update_points(self):
with pytest.raises(NotImplementedError):
self.memory_db.update_points("filter", [1,2,3])
def test_get_data_values_is_empty(self):
dvs = self.memory_db.get_data_values()
assert len(dvs) == 0
class TestMemoryDB:
def setup(self):
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.series = test_util.add_series(self.session)
self.memory_db = MemoryDatabase(self.series_service)
def test_delete_points(self):
with pytest.raises(NotImplementedError):
self.memory_db.delete_points("filter")
def test_add_points(self):
with pytest.raises(NotImplementedError):
self.memory_db.add_points("filter")
def test_update_points(self):
with pytest.raises(NotImplementedError):
self.memory_db.update_points("filter", [1, 2, 3])
def test_get_data_values_is_empty(self):
dvs = self.memory_db.get_data_values()
assert len(dvs) == 0
def setup(self): 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.series = test_util.add_series(self.session) self.memory_db = MemoryDatabase(self.series_service)
class pnlSeriesSelector(wx.Panel):
## Radio Sizer
## def _init_coll_boxSizer5_Items(self, parent):
## # generated method, don't edit
##
## parent.AddWindow(self.rbAll, 0, border=1, flag=wx.ALL)
## parent.AddWindow(self.rbSimple, 0, border=1, flag=wx.ALL)
## parent.AddWindow(self.rbAdvanced, 0, border=1, flag=wx.ALL)
## Splitter Sizer
def _init_coll_boxSizer3_Items(self, parent):
# generated method, don't edit
parent.AddWindow(self.cpnlSimple, 0, flag=wx.RIGHT|wx.LEFT|wx.EXPAND)
parent.AddWindow(self.tableSeries, 100, flag=wx.EXPAND)
## Panel Sizer
def _init_coll_boxSizer1_Items(self, parent):
# generated method, don't edit
parent.AddSizer(self.pnlRadio, 0, border=7, flag=wx.LEFT|wx.RIGHT |wx.TOP)
parent.AddWindow(self.pnlData, 100, border=3, flag=wx.LEFT|wx.RIGHT | wx.EXPAND)
## Site Sizer
def _init_coll_boxSizer4_Items(self, parent):
# generated method, don't edit
parent.AddWindow(self.checkSite, 0, border=3, flag=wx.LEFT|wx.RIGHT)
parent.AddWindow(self.lblSite, 0, border=3, flag=wx.LEFT|wx.RIGHT )
parent.AddWindow(self.cbSites, 90, border=3, flag=wx.LEFT|wx.RIGHT | wx.EXPAND)
## Variable Sizer
def _init_coll_boxSizer2_Items(self, parent):
# generated method, don't edit
parent.AddWindow(self.checkVariable, 0, border=3, flag=wx.LEFT|wx.RIGHT )
parent.AddWindow(self.lblVariable, 0, border=3, flag=wx.LEFT|wx.RIGHT )
parent.AddWindow(self.cbVariables, 90, border=3, flag=wx.LEFT|wx.RIGHT | wx.EXPAND)
## Simple Filter Sizer
def _init_coll_boxSizer6_Items(self, parent):
parent.AddWindow(self.pnlSite, 50, flag=wx.EXPAND)
parent.AddWindow(self.pnlVar, 50, flag=wx.EXPAND)
# parent.AddSizer(self.boxSizer4, 0, border=5, flag=wx.EXPAND)
# parent.AddSizer(self.boxSizer2, 0, border=5, flag=wx.EXPAND)
def _init_sizers(self):
# generated method, don't edit
self.boxSizer1 = wx.BoxSizer(orient=wx.VERTICAL)
self.boxSizer2 = wx.BoxSizer(orient=wx.HORIZONTAL)
self.boxSizer3 = wx.BoxSizer(orient=wx.VERTICAL)
self.boxSizer4 = wx.BoxSizer(orient=wx.HORIZONTAL)
## self.boxSizer5 = wx.BoxSizer(orient=wx.HORIZONTAL)
self.boxSizer6 = wx.BoxSizer(orient=wx.VERTICAL)
self._init_coll_boxSizer1_Items(self.boxSizer1)
self._init_coll_boxSizer2_Items(self.boxSizer2)
self._init_coll_boxSizer3_Items(self.boxSizer3)
self._init_coll_boxSizer4_Items(self.boxSizer4)
## self._init_coll_boxSizer5_Items(self.boxSizer5)
self._init_coll_boxSizer6_Items(self.boxSizer6)
self.SetSizer(self.boxSizer1)
## self.pnlRadio.SetSizer(self.boxSizer5)
## self.pnlSite.SetSizer(self.boxSizer4)
## self.pnlVar.SetSizer(self.boxSizer2)
self.cpnlSimple.SetSizer(self.boxSizer6)
self.pnlData.SetSizer(self.boxSizer3)
# self.pnlRadio.SetSizer(self.boxSizer5)
def _init_ctrls(self, prnt):
# generated method, don't edit
wx.Panel.__init__(self, id=wxID_PNLSERIESSELECTOR,
name=u'pnlSeriesSelector', parent=prnt,
size=wx.Size(935, 270), style=wx.TAB_TRAVERSAL)
self.SetClientSize(wx.Size(919, 232))
self.Enable(True)
## Radio panel
self.pnlRadio = wx.Panel(id=wxID_PNLRADIO, name='pnlRadio',
parent=self, pos=wx.Point(3, 3), size=wx.Size(919, 20),
style=wx.TAB_TRAVERSAL)
self.rbAll = wx.RadioButton(id=wxID_FRAME1RBALL, label=u'All',
name=u'rbAll', parent=self.pnlRadio, pos=wx.Point(0, 0),
size=wx.Size(81, 20), style=0)
self.rbAll.SetValue(True)
self.rbAll.Bind(wx.EVT_RADIOBUTTON, self.OnRbAllRadiobutton,
id=wxID_FRAME1RBALL)
self.rbSimple = wx.RadioButton(id=wxID_FRAME1RBSIMPLE,
label=u'Simple Filter', name=u'rbSimple', parent=self.pnlRadio,
pos=wx.Point(81, 0), size=wx.Size(112, 20), style=0)
self.rbSimple.Bind(wx.EVT_RADIOBUTTON, self.OnRbSimpleRadiobutton,
id=wxID_FRAME1RBSIMPLE)
self.rbAdvanced = wx.RadioButton(id=wxID_FRAME1RBADVANCED,
label=u'Advanced Filter', name=u'rbAdvanced', parent=self.pnlRadio,
pos=wx.Point(193, 0), size=wx.Size(104, 20), style=0)
self.rbAdvanced.Bind(wx.EVT_RADIOBUTTON, self.OnRbAdvancedRadiobutton,
id=wxID_FRAME1RBADVANCED)
self.rbAdvanced.Enable(False)
## Splitter panel
self.pnlData = wx.Panel(id=wxID_PNLSPLITTER, name='pnlData',
parent=self, pos=wx.Point(0, -10), size=wx.Size(900, 349),
style=wx.TAB_TRAVERSAL)
## self.splitter = wx.SplitterWindow(id=wxID_FRAME1SPLITTER,
## name=u'splitter', parent=self.pnlData, pos=wx.Point(0, 0),
## size=wx.Size(604, 137), style=wx.NO_BORDER)
## self.splitter.SetMinSize(wx.Size(-1, -1))
self.cpnlSimple =PCP.PyCollapsiblePane( parent=self.pnlData, label="",
agwStyle=wx.CP_NO_TLW_RESIZE | wx.CP_GTK_EXPANDER | wx.CP_USE_STATICBOX,
size = wx.Size(300,20), pos =wx.Point(0,-20))
self.Bind(wx.EVT_COLLAPSIBLEPANE_CHANGED, self.OnPaneChanged, self.cpnlSimple)
## panel for simple filter(top of splitter)
## self.pnlSimple = wx.Panel(id=wxID_PNLSIMPLE, name='panel3',
## parent=self.splitter, pos=wx.Point(0, 0), size=wx.Size(919, 300),
## style=wx.TAB_TRAVERSAL)
## Site Panel
self.pnlSite = wx.Panel(id=wxID_PNLSERIESSELECTORPANEL1, name='pnlSite',
parent=self.cpnlSimple.GetPane(), pos=wx.Point(3, 0), size=wx.Size(800, 25),
style=wx.TAB_TRAVERSAL)
self.cbSites = wx.ComboBox(choices=[], id=wxID_PNLSERIESSELECTORCBSITES,
name=u'cbSites', parent=self.pnlSite, pos=wx.Point(100, 0),
size=wx.Size(700, 23), style=0, value=u'')
self.cbSites.SetLabel(u'')
self.cbSites.Bind(wx.EVT_COMBOBOX, self.OnCbSitesCombobox,
id=wxID_PNLSERIESSELECTORCBSITES)
self.checkSite = wx.CheckBox(id=wxID_PNLSERIESSELECTORCHECKSITE,
label=u'', name=u'checkSite', parent=self.pnlSite, pos=wx.Point(3,
0), size=wx.Size(21, 21), style=0)
self.checkSite.SetValue(True)
self.checkSite.Bind(wx.EVT_CHECKBOX, self.OnCheck,
id=wxID_PNLSERIESSELECTORCHECKSITE)
self.lblSite = wx.StaticText(id=wxID_PNLSERIESSELECTORLBLSITE,
label=u'Site', name=u'lblSite', parent=self.pnlSite,
pos=wx.Point(30, 0), size=wx.Size(60, 21), style=0)
self.lblSite.SetToolTipString(u'staticText1')
### Variable Panel
self.pnlVar = wx.Panel(id=wxID_PNLSERIESSELECTORPANEL2, name='pnlVar',
parent=self.cpnlSimple.GetPane(), pos=wx.Point(3, 26), size=wx.Size(800, 25),
style=wx.TAB_TRAVERSAL)
self.lblVariable = wx.StaticText(id=wxID_PNLSERIESSELECTORLBLVARIABLE,
label=u'Variable', name=u'lblVariable', parent=self.pnlVar,
pos=wx.Point(30, 0), size=wx.Size(60, 21), style=0)
self.checkVariable = wx.CheckBox(id=wxID_PNLSERIESSELECTORCHECKVARIABLE,
label=u'', name=u'checkVariable', parent=self.pnlVar,
pos=wx.Point(3, 0), size=wx.Size(21, 21), style=0)
self.checkVariable.Bind(wx.EVT_CHECKBOX, self.OnCheck,
id=wxID_PNLSERIESSELECTORCHECKVARIABLE)
self.cbVariables = wx.ComboBox(choices=[],
id=wxID_PNLSERIESSELECTORCBVARIABLES, name=u'cbVariables',
parent=self.pnlVar, pos=wx.Point(100, 0), size=wx.Size(700, 25),
style=0, value='comboBox4')
self.cbVariables.SetLabel(u'')
self.cbVariables.Enable(False)
self.cbVariables.Bind(wx.EVT_COMBOBOX, self.OnCbVariablesCombobox,
id=wxID_PNLSERIESSELECTORCBVARIABLES)
self.tableSeries = clsULC(id=wxID_PNLSERIESSELECTORtableSeries,
name=u'tableSeries', parent=self.pnlData, pos=wx.Point(5, 5),
size=wx.Size(903, 108),
agwStyle= ULC.ULC_REPORT | ULC.ULC_HRULES | ULC.ULC_VRULES | ULC.ULC_HAS_VARIABLE_ROW_HEIGHT |ULC.ULC_SINGLE_SEL)
## self.splitter.Initialize(self.tableSeries)
self.cpnlSimple.Collapse(True)
# self.splitter.SplitHorizontally(self.pnlSimple, self.tableSeries, 1)
self.tableSeries.Bind(ULC.EVT_LIST_ITEM_CHECKED,
self.OntableSeriesListItemSelected,
id=wxID_PNLSERIESSELECTORtableSeries)
self.tableSeries.Bind(wx.EVT_LIST_ITEM_RIGHT_CLICK,
self.OnTableRightDown,
id=wxID_PNLSERIESSELECTORtableSeries)
Publisher.subscribe(self.OnEditButton, ("selectEdit"))
self._init_sizers()
def __init__(self, parent, id, size, style, name, dbservice,pos=None,):
self.parent= parent
self._init_ctrls(parent)
self.dbservice = dbservice
self.initTableSeries()
self.initSVBoxes()
sm = ServiceManager()
self.export_service = sm.get_export_service()
def resetDB(self, dbservice):
#####INIT DB Connection
self.dbservice = dbservice
self.tableSeries.Clear()
self.cbVariables.Clear()
self.cbSites.Clear()
self.siteList = None
self.varList =None
self.initTableSeries()
self.initSVBoxes()
self.Layout()
def initTableSeries(self):
self.dataRep=MemoryDatabase(self.dbservice)
self.tableSeries.SetColumns(self.dataRep.getSeriesColumns())
self.tableSeries.SetObjects(self.dataRep.getSeriesCatalog())
def initSVBoxes(self):
self.site_code = None
self.variable_code = None
#####INIT drop down boxes for Simple Filter
self.siteList=self.dbservice.get_sites()
for site in self.siteList:
self.cbSites.Append(site.site_code+'-'+site.site_name)
self.cbSites.SetSelection(0)
self.site_code = self.siteList[0].site_code
self.varList= self.dbservice.get_variables(self.site_code)
for var in self.varList:
self.cbVariables.Append(var.variable_code+'-'+var.variable_name)
self.cbVariables.SetSelection(0)
def OnTableRightDown(self, event):
# build pop-up menu for right-click display
self.selectedIndex= event.m_itemIndex
self.selectedID = self.tableSeries.GetColumnText(event.m_itemIndex, 1)
# print self.selectedID
popup_edit_series = wx.NewId()
popup_plot_series = wx.NewId()
popup_export_data = wx.NewId()
popup_export_metadata = wx.NewId()
popup_select_all = wx.NewId()
popup_select_none = wx.NewId()
popup_menu = wx.Menu()
self.Bind(wx.EVT_MENU, self.OnRightPlot, popup_menu.Append(popup_plot_series, 'Plot'))
self.Bind(wx.EVT_MENU, self.OnRightEdit, popup_menu.Append(popup_edit_series, 'Edit'))
popup_menu.AppendSeparator()
self.Bind(wx.EVT_MENU, self.OnRightExData, popup_menu.Append(popup_export_data, 'Export Data'))
self.Bind(wx.EVT_MENU, self.OnRightExMeta, popup_menu.Append(popup_export_metadata, 'Export MetaData'))
self.tableSeries.PopupMenu(popup_menu)
event.Skip()
def OnPaneChanged(self, event = None):
if event:
print 'wx.EVT_COLLAPSIBLEPANE_CHANGED: %s\n' % event.Collapsed
self.Layout()
def OnRbAdvancedRadiobutton(self, event):
#open filter window and hide top Panel
# self.splitter.SetSashPosition(1)
## if self.splitter.IsSplit():
## self.splitter.Unsplit(self.pnlSimple)
self.cpnlSimple.Collapse(True)
self.Layout()
series_filter = frmQueryBuilder.frmQueryBuilder(self)
self.filterlist = series_filter.ShowModal()
# print self.filterlist
event.Skip()
def OnRbAllRadiobutton(self, event):
#Hide top panel
## if self.splitter.IsSplit():
## self.splitter.Unsplit(self.pnlSimple)
self.cpnlSimple.Collapse(True)
self.Layout()
# self.splitter.SetSashPosition(1)
self.SetFilter()
event.Skip()
def OnRbSimpleRadiobutton(self, event):
#show top Panel
## if not self.splitter.IsSplit():
## self.splitter.SplitHorizontally(self.pnlSimple, self.tableSeries, 30)
# self.splitter.SetSashPosition(70)
self.cpnlSimple.Expand()
self.Layout()
self.SetFilter(self.site_code, self.variable_code)
event.Skip()
def OnRightPlot(self, event):
# print self.tableSeries.IsItemChecked(self.selectedIndex)
# self.tableSeries.GetItem(self.selectedID, 0).Check = True
## self.tableSeries.GetColumnText(self.selectedIndex, 1).Check = True
self.tableSeries.CheckItem(self.selectedIndex)
self.SelectForPlot(self.selectedIndex)
print "click Plot"
event.Skip()
def OnRightEdit(self, event):
self.SelectForEdit(self.tableSeries.GetColumnText(self.selectedIndex, 1))
event.Skip()
def OnEditButton(self, event):
#
if self.tableSeries.GetSelectedItemCount()>0:
self.SelectForEdit(self.tableSeries.GetColumnText(self.tableSeries.GetSelection(), 1))
else:
#toggle ribbon button to be unpressed
print "no series selected"
def OnRightExData(self, event):
dlg = wx.FileDialog(self, "Choose a save location", '', "", "*.csv", wx.SAVE | wx.OVERWRITE_PROMPT)
if dlg.ShowModal() == wx.ID_OK:
full_path = os.path.join(dlg.GetDirectory(), dlg.GetFilename())
series_id = self.tableSeries.GetColumnText(self.selectedIndex, 1)
self.export_service.export_series_data(series_id, full_path, True, True, True, True, True, True, True)
self.Close()
dlg.Destroy()
event.Skip()
def OnRightExMeta(self, event):
dlg = wx.FileDialog(self, "Choose a save location", '', "", "*.xml", wx.SAVE | wx.OVERWRITE_PROMPT)
if dlg.ShowModal() == wx.ID_OK:
full_path = os.path.join(dlg.GetDirectory(), dlg.GetFilename())
series_id = self.tableSeries.GetColumnText(self.selectedIndex, 1)
self.export_service.export_series_metadata(series_id, full_path)
self.Close()
dlg.Destroy()
event.Skip()
def OnCbSitesCombobox(self, event):
self.site_code = self.siteList[event.GetSelection()].site_code
self.varList =[]
self.varList= self.dbservice.get_variables(self.site_code)
self.cbVariables.Clear()
for var in self.varList:
self.cbVariables.Append(var.variable_code+'-'+var.variable_name)
self.cbVariables.SetSelection(0)
#if (not self.checkVariable):
if (self.checkSite.GetValue() and not self.checkVariable.GetValue()):
self.variable_code = None
self.SetFilter(site_code = self.site_code, var_code = self.variable_code)
event.Skip()
def OnCbVariablesCombobox(self, event):
self.variable_code = self.varList[event.GetSelection()].variable_code
# if (self.checkSite.GetValue() and self.checkVariable.GetValue()):
# self.seriesList = self.dbservice.get_series(site_code = self.site_code, var_code= self.variable_code)
if (not self.checkSite.GetValue() and self.checkVariable.GetValue()):
self.site_code = None
self.SetFilter(site_code = self.site_code, var_code = self.variable_code)
event.Skip()
def siteAndVariables(self):
self.cbVariables.Clear()
self.varList= self.dbservice.get_variables(self.site_code)
for var in self.varList:
self.cbVariables.Append(var.variable_code+'-'+var.variable_name)
self.cbVariables.SetSelection(0)
self.variable_code=self.varList[self.cbVariables.Selection].variable_code
self.site_code = self.siteList[self.cbSites.Selection].site_code
self.SetFilter(site_code = self.site_code, var_code = self.variable_code)
self.cbVariables.Enabled =True
self.cbSites.Enabled = True
def siteOnly(self):
self.cbVariables.Enabled = False
self.cbSites.Enabled = True
self.variable_code = None
self.site_code = self.siteList[self.cbSites.Selection].site_code
self.SetFilter(site_code = self.site_code)
def variableOnly(self):
self.site_code = None
self.cbVariables.Clear()
self.varList= self.dbservice.get_variables()
for var in self.varList:
self.cbVariables.Append(var.variable_code+'-'+var.variable_name)
self.cbVariables.SetSelection(0)
self.cbSites.Enabled = False
self.cbVariables.Enabled = True
self.variable_code=self.varList[0].variable_code
self.SetFilter( var_code = self.variable_code)
def OnCheck(self, event):
# self.tableSeries.DeleteAllItems()
if self.checkSite.GetValue():
if self.checkVariable.GetValue():
self.siteAndVariables()
else:
self.siteOnly()
else:
if self.checkVariable.GetValue():
self.variableOnly()
else:
self.cbSites.Enabled = False
self.cbVariables.Enabled = False
event.Skip()
def SetFilter(self, site_code = '', var_code = '', advfilter = ''):
if (site_code and var_code):
self.siteFilter= TextSearch(self.tableSeries, columns = self.tableSeries.columns[2:4], text = site_code )
self.variableFilter = TextSearch(self.tableSeries, columns = self.tableSeries.columns[5:7], text = var_code)
self.tableSeries.SetFilter(Chain(self.siteFilter, self.variableFilter))
elif (site_code):
self.tableSeries.SetFilter(TextSearch(self.tableSeries, columns = self.tableSeries.columns[2:4], text = site_code) )
elif (var_code):
self.tableSeries.SetFilter(TextSearch(self.tableSeries, columns = self.tableSeries.columns[5:7], text = var_code) )
elif(advfilter):
self.tableSeries.SetFilter(advfilter)
else:
self.tableSeries.ClearFilter()
self.tableSeries.RepopulateList()
# print self.tableSeries.GetItemCount()
def OntableSeriesListItemSelected(self, event):
# print"in item selected", event.m_itemIndex, self.tableSeries.IsItemChecked(event.m_itemIndex)
# print dir(event)
self.SelectForPlot( event.m_itemIndex)
event.Skip()
def SelectForPlot(self, selIndex ):
sid= self.tableSeries.innerList[selIndex][0]
if not self.tableSeries.IsItemChecked(selIndex):
Publisher.sendMessage(("removePlot"), seriesID=sid)
self.tableSeries.innerList[selIndex][-1]= False
else:
#set isselected value to True
self.tableSeries.innerList[selIndex][-1]= True
self.parent.Parent.addPlot(self.dataRep ,sid)
self.Refresh()
def SelectForEdit(self, seriesID):
self.dataRep.initEditValues(seriesID)
self.parent.Parent.addEdit(seriesID, self.dataRep)
def stopEdit(self):
self.dataRep.stopEdit()
def initTableSeries(self):
self.dataRep=MemoryDatabase(self.dbservice)
self.tableSeries.SetColumns(self.dataRep.getSeriesColumns())
self.tableSeries.SetObjects(self.dataRep.getSeriesCatalog())
