From e6dbfdb110b1aa227945c1903ff418979eb23381 Mon Sep 17 00:00:00 2001
From: SGCMarkus <markusornik@gmail.com>
Date: Wed, 6 Mar 2024 14:01:11 +0100
Subject: [PATCH] astrodatagui: NewStarDialog: download files on ok button
 click and set return values

---
 main/astrodatagui/ui/NewStarDialog.py | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/main/astrodatagui/ui/NewStarDialog.py b/main/astrodatagui/ui/NewStarDialog.py
index 4dd6b21..b3ea897 100644
--- a/main/astrodatagui/ui/NewStarDialog.py
+++ b/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
\ No newline at end of file
+        self.reject()
\ No newline at end of file