Commit c1e8ec57 authored by Manuel Luis Alcazar Laynez's avatar Manuel Luis Alcazar Laynez
Browse files

Reducing code in galassify.py

parent 0e5672db
Loading
Loading
Loading
Loading
+20 −23
Original line number Diff line number Diff line
@@ -16,16 +16,7 @@ def main():
    args = utils.getOptions(VERSION)
    if not len(sys.argv) > 1:
        if utils.exist_basic_files():
            selectedFiles, selectedGroups = utils.getFiles()
            if len(selectedFiles) > 0:
                if args.list:
                    print(selectedFiles.filename)
                else:
                    print(f"INFO:\t {str(len(selectedFiles))} galaxies found in selected group(s).")
                    df = selectedFiles # utils.expand_df(selectedFiles)
                    app = QtWidgets.QApplication(sys.argv)
                    window = gui.Ui(df, selectedGroups)
                    app.exec_()
            run_gui(args)
        else:    
            selec = utils.load_menu()
            if selec == 0:
@@ -41,6 +32,10 @@ def main():
            print(f"INFO:\t Create the {args.example} example files")
            utils.example_flag(args.example)
        else:
            run_gui(args)


def run_gui(args):
    selectedFiles, selectedGroups = utils.getFiles()
    if len(selectedFiles) > 0:
        if args.list:
@@ -52,7 +47,9 @@ def main():
            window = gui.Ui(df, selectedGroups)
            app.exec_()
    else:
                print('ERROR: No files found with the arguments given.')
        print('ERROR:\tNo files found with the arguments given.')
        print('HINT:\tDid you populate your .csv file?')


if __name__ == '__main__':
    sys.exit(main())
 No newline at end of file