diff --git a/alfred_do_get_week.py b/alfred_do_get_week.py index 462d165..b8000a0 100755 --- a/alfred_do_get_week.py +++ b/alfred_do_get_week.py @@ -8,21 +8,32 @@ from requests.exceptions import ConnectionError def main(wf): num = None subtt = u'' + try: num = str(get_week.get_week()) - title = u'Épp {}.'.format(num) + title = u'Épp {}.' subtt = u'hét van!' - if num == 14: title = u'Póthét ({}. hét)'.format(num); subtt = u'van!' - if num > 14: title = u'Vizsgaidőszak ({}. hét)'.format(num); subtt = u'van!' + + if int(num) == 14: + title = u'Póthét ({}. hét)' + subtt = u'van!' + if int(num) > 14: + title = u'Vizsgaidőszak ({}. hét)' + subtt = u'van!' + + title = title.format(num) + except ConnectionError: title = u'Nincs internet!' except (IndexError, RuntimeError): title = u'A KTH összebarmolta a honlapját!' except EnvironmentError: title = u'Nyár van, nyugi!' + wf.add_item(title=title, subtitle=subtt, arg=num, valid=True) wf.send_feedback() + if __name__ == '__main__': wf = Workflow() sys.exit(wf.run(main))