def filled_database():
database = Database('sqlite:///:memory:')
for i in range(1, 11):
entry = DatabaseEntry()
database.add(entry)
# every fourth entry gets the tag 'foo'
if i % 4 == 0:
database.tag(entry, 'foo')
# every fifth entry gets the tag 'bar'
if i % 5 == 0:
database.tag(entry, 'bar')
database.commit()
return database
def filled_database():
database = Database('sqlite:///:memory:')
for i in xrange(1, 11):
entry = DatabaseEntry()
database.add(entry)
# every fourth entry gets the tag 'foo'
if i % 4 == 0:
database.tag(entry, 'foo')
# every fifth entry gets the tag 'bar'
if i % 5 == 0:
database.tag(entry, 'bar')
database.commit()
return database
def split_function_database():
"""
Generates a custom database to test the split_database function
"""
database = Database('sqlite:///:memory:')
for i in range(1, 11):
entry = DatabaseEntry()
database.add(entry)
# every fourth entry gets the instrument 'EIA'
if i % 4 == 0:
database.edit(entry, instrument='EIA')
# every fifth entry gets the instrument 'AIA_3'
elif i % 5 == 0:
database.edit(entry, instrument='AIA_3')
# every other entry gets instrument 'RHESSI'
else:
database.edit(entry, instrument='RHESSI')
# all entries have provider 'xyz'
database.edit(entry, provider='xyz')
database.commit()
return database
def split_function_database():
"""
Generates a custom database to test the split_database function
"""
database = Database('sqlite:///:memory:')
for i in range(1, 11):
entry = DatabaseEntry()
database.add(entry)
# every fourth entry gets the instrument 'EIA'
if i % 4 == 0:
database.edit(entry, instrument='EIA')
# every fifth entry gets the instrument 'AIA_3'
elif i % 5 == 0:
database.edit(entry, instrument='AIA_3')
# every other entry gets instrument 'RHESSI'
else:
database.edit(entry, instrument='RHESSI')
# all entries have provider 'xyz'
database.edit(entry, provider='xyz')
database.commit()
return database
class DownloadResultController(ToolController):
queries = {}
app_ctrl: AppController = RequiredFeature(AppController.__name__)
content_ctrl: ContentController = RequiredFeature(ContentController.name)
def __init__(self):
self._view = QtWidgets.QWidget()
self._ui = Ui_DownloadResult()
self._ui.setupUi(self._view)
self._ui.tabs.clear()
self._ui.tabs.tabCloseRequested.connect(self._onRemoveTab)
self.database = Database()
self.tabs = {}
self.queries = {}
self.query_id = 0
self.loading = []
self.loaded = {
entry.fileid: entry.path
for entry in list(self.database)
}
self._ui.download_button.clicked.connect(
lambda evt: self._onDownloadSelected())
self._ui.open_button.clicked.connect(
lambda evt: self._onOpenSelected())
def query(self, attrs):
# open tool if not already opened
self.app_ctrl.openController(self.name)
self.query_id += 1
# add pending tab
tab = ResultTab(self.query_id)
self.tabs[self.query_id] = tab
index = self._ui.tabs.addTab(tab, "Query " + str(self.query_id))
self._ui.tabs.setCurrentIndex(index)
# start query
executeTask(Fido.search, attrs, self._onQueryResult, [self.query_id])
# register events
tab.download.connect(
lambda f_id, q_id=self.query_id: self.download(q_id, f_id))
tab.open.connect(lambda f_id: self._onOpen(f_id))
def _onQueryResult(self, query, id):
if id not in self.tabs:
return
query_model = self._convertQuery(query)
self.tabs[id].loadQuery(query_model)
self.tabs[id].setLoading(self.loading)
self.tabs[id].setLoaded(self.loaded.keys())
self.queries[id] = query
def download(self, q_id, f_id):
req = copy.copy(self.queries[q_id])
req._list = [copy.copy(r) for r in req]
for resp in req:
resp[:] = [item for item in resp if item.fileid == f_id]
self._addLoading([f_id])
executeTask(Fido.fetch, [req], self._onDownloadResult, [f_id, req])
def _onDownloadResult(self, paths, f_id, request):
path = paths[0]
entry = list(
tables.entries_from_fido_search_result(
request, self.database.default_waveunit))[0]
entry.path = path
self.database.add(entry)
self.database.commit()
self._addLoaded({f_id: path})
def _onOpen(self, f_id):
viewer = MapViewerController.fromFile(self.loaded[f_id])
self.content_ctrl.addViewerController(viewer)
def _onRemoveTab(self, index):
tab = self._ui.tabs.widget(index)
self._ui.tabs.removeTab(index)
self.tabs.pop(tab.q_id)
def _onDownloadSelected(self):
tab = self._ui.tabs.currentWidget()
f_ids = tab.getSelectedFIds()
for f_id in f_ids:
if f_id in self.loading or f_id in self.loaded:
continue
self.download(tab.q_id, f_id)
def _onOpenSelected(self):
tab = self._ui.tabs.currentWidget()
f_ids = tab.getSelectedFIds()
for f_id in f_ids:
if f_id not in self.loaded:
continue
self._onOpen(f_id)
def _convertQuery(self, query):
items = [item for response in query for item in response]
return [[c[1](item) for c in columns] for item in items]
def _addLoading(self, f_ids):
self.loading.extend(f_ids)
for tab in self.tabs.values():
tab.setLoading(f_ids)
def _addLoaded(self, dict):
self.loading = [
f_id for f_id in self.loading if f_id not in dict.keys()
]
self.loaded.update(dict)
for tab in self.tabs.values():
tab.setLoaded(dict.keys())
@property
def item_config(self) -> ItemConfig:
return ItemConfig().setTitle("Download Results").setOrientation(
QtCore.Qt.BottomDockWidgetArea)
@property
def view(self) -> QtWidgets:
return self._view
