astrodatagui: NewStarDialog: download files on ok button click and set return values

main/astrodatagui/ui/NewStarDialog.py

@@ -3,7 +3,8 @@ from PyQt5.QtWidgets import QDialog, QListWidgetItem
 from PyQt5 import uic
 import os
 
-from ...astrodatadownloader.astrodatadownloader import ObservationSource, getStarObservations
+from ...astrodatadownloader.astrodatadownloader import (ObservationSource,
+                                                        getStarObservations, downloadStarProducts)
 
 class NewStarDialog(QDialog):
     def __init__(self):
@@ -16,6 +17,9 @@ class NewStarDialog(QDialog):
         self.btOk.clicked.connect(self.btOk_clicked)
         self.btCancel.clicked.connect(self.btCancel_clicked)
 
+        self.currentObservations = None
+        self.productManifest = None
+
     def showErrorMessage(self, title: str, msg: str):
         QtWidgets.QMessageBox.question(self, title, msg,
                                                       QtWidgets.QMessageBox.Ok)
@@ -50,8 +54,18 @@ class NewStarDialog(QDialog):
         for o in obs:
             self.listPreview.addItem(QListWidgetItem(f"{o['target_name']} - {o['obs_collection']} - {o['sequence_number']}"))
 
+        self.currentObservations = obs
+
     def btOk_clicked(self):
-        return
+        if self.currentObservations:
+            keplerKadences = [self.cbKeplerShortCadence.isChecked(),
+                              self.cbKeplerLongCadence.isChecked()]
+            k2Kadences = [self.cbK2ShortCadence.isChecked(),
+                              self.cbK2LongCadence.isChecked()]
+
+            self.productManifest = downloadStarProducts(self.currentObservations,
+                                                        keplerKadences, k2Kadences)
+            self.accept()
 
     def btCancel_clicked(self):
-        return
+        self.reject()