Prepare for being a helper script and stop messing with it

The 'main' trick is cool, all that stuff within it only happens if it's called
directly.
This commit is contained in:
mlncn 2021-05-02 12:04:02 -04:00
parent 1d07fe08b9
commit 596bf7e6a0

View file

@ -55,10 +55,17 @@ class PromptWindow(Gtk.Window):
self.entry.set_text(self.orig_task) self.entry.set_text(self.orig_task)
def retrieve(self): def retrieve(self):
if self.task:
return self.task return self.task
else:
return None
# started to do conditional like this but maybe main program handles.
# else if self.orig_task:
# return "Pomodoro cancelled, original goal was:"
win = PromptWindow() if __name__ == '__main__':
win = PromptWindow("I'mma tell you what i'm gonna do.")
win.show_all() win.show_all()
Gtk.main() Gtk.main()
tha_task = win.retrieve() tha_task = win.retrieve()