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