wf-worldstate/main.py

# central controller, main loop

from gui import GUI
from api import WorldState
from extra_widgets import *

from PyQt5.QtWidgets import QApplication, QMessageBox
from PyQt5.QtCore import QTimer

import sys


class App(GUI):
    def __init__(self):
        super().__init__()
        # fetch world state info from API
        self.ws = WorldState()
        # if bad fetch, error for retry until quit or fetched
        while self.ws.ws == {}:
            self.network_error()
            sys.exit(1)

        # remove placeholder templates for given widgets, replaced with actual data later in redraw
        [widget.setParent(None) for widget in [self.Fissure, self.Invasion, self.Alert, self.BaroItem, self.Event, self.NewsBlurb]]
        
        # update interval for clocks
        self.timers_timer = QTimer(self)
        self.timers_timer.timeout.connect(self.update_timers)
        self.timers_timer.start(1000)

        # update interval for data refresh
        self.ws_update_timer = QTimer(self)
        self.ws_update_timer.timeout.connect(self.update_world_state)
        self.ws_update_timer.start(1000 * 60 * 5)

        # largely just creating custom widgets, and mapping to existing parent widgets
        self.redraw_content()

    def redraw_content(self):
        self.redraw_content_dynamic() # content that varies in number, invasions, fissures, etc
        self.redraw_content_static() # static content that won't change/only one, arbi, archons, etc

    def redraw_content_dynamic(self):
        self.Fissure_list = [Fissure(self.FissureHolder, fissure_info, fissure_i) for (fissure_i, fissure_info) in enumerate(self.ws.fissures)]
        [self.FissureList.addWidget(i) for i in self.Fissure_list]

        self.Invasion_list = [Invasion(self.InvasionHolder, invasion_info, invasion_i) for (invasion_i, invasion_info) in enumerate(self.ws.invasions)]
        [self.InvasionList.addWidget(i) for i in self.Invasion_list]

        self.Alert_list = [Alert(self.AlertHolder, alert_info, alert_i) for (alert_i, alert_info) in enumerate(self.ws.alerts)]
        [self.AlertList.addWidget(i) for i in self.Alert_list]

        # if baro key is [], no baro, switch tab to inactive, else populate grid
        if len(self.ws.baro_items) == 0:
            self.BaroTabs.setCurrentIndex(0)
        else:
            self.BaroTabs.setCurrentIndex(1)
            self.baro_items_list = [BaroItem(self.BaroData, item_info, item_i) for (item_i, item_info) in enumerate(self.ws.baro_items)]
            [self.gridLayout_6.addWidget(i, index // 7, index % 7, 1, 1) for (index, i) in enumerate(self.baro_items_list)]

        self.events_list = [Event(self.EventHolder, event_info, event_i) for (event_i, event_info) in enumerate(self.ws.events)]
        [self.EventsList.addWidget(i) for i in self.events_list]

        self.newsblurb_list = [NewsBlurb(self.NewsHolder, newsblurb_info, newsblurb_i) for (newsblurb_i, newsblurb_info) in enumerate(self.ws.news)]
        [self.NewsList.addWidget(i) for i in self.newsblurb_list]

    def redraw_content_static(self):
        for i in range(1, 4):
            getattr(self, 'SortieM' + str(i)).setText(self.ws.sorties['variants'][i - 1]['missionType'])
            getattr(self, 'SortieM' + str(i) + 'Mod').setText(self.ws.sorties['variants'][i-1]['modifier'])

        self.ArchonHuntStr.setText('Archon Hunt: ' + self.ws.archon_hunt['boss'])
        for i in range(1, 4):
            getattr(self, 'ArchonHuntM' + str(i)).setText(self.ws.archon_hunt['missions'][i - 1]['type'])

        # warframestat.us arbi jank fix until content API adds arbis and can use that directly
        if not self.ws.arbitration:
            self.ArbitrationTabs.setCurrentIndex(0)
        else:
            self.ArbitrationTabs.setCurrentIndex(1)
            self.ArbitrationMT.setText(self.ws.arbitration['type'])
            self.ArbitrationFaction.setText(self.ws.arbitration['enemy'])

        # nightwave mission mapping if active
        daily_i = 1
        weekly_i = 1
        elite_i = 1
        if self.ws.nightwave:
            self.NightwaveTabs.setCurrentIndex(1)
            for challenge in self.ws.nightwave['activeChallenges']:
                if challenge.get('isDaily'):
                    challenge_type = 'Daily'
                    challenge_i = daily_i
                    daily_i += 1
                elif challenge.get('isElite'):
                    challenge_type = 'Elite'
                    challenge_i = weekly_i
                    weekly_i += 1
                else:
                    challenge_type = 'Weekly'
                    challenge_i = elite_i
                    elite_i += 1
                getattr(self, challenge_type + str(challenge_i) + 'Name').setText(challenge['title'])
                getattr(self, challenge_type + str(challenge_i) + 'Desc').setText(challenge['desc'])
        else:
            self.NightwaveTabs.setCurrentIndex(0)

    def update_timers(self):
        # call timer updates and invasion progress changes
        [i.update_timer() for i in self.Fissure_list]
        [i.update_progress() for i in self.Invasion_list]
        [i.update_timer() for i in self.Alert_list]
        [i.update_timer() for i in self.events_list]
        [i.update_timer() for i in self.newsblurb_list]

        # parse expiry datestring and calc sortie expiry time delta from now to listed expiry
        sortie_eta = round(time.mktime(time.strptime(self.ws.sorties['expiry'], '%Y-%m-%dT%H:%M:%S.%fZ')) - time.mktime(time.gmtime()))
        if sortie_eta >= 0:
            self.SortieResetTimer.setText(f'{str(sortie_eta // 3600) + "h " if sortie_eta // 3600 > 0 else ""}{(sortie_eta // 60) % 60}m {sortie_eta % 60}s')
        else:
            self.SortieResetTimer.setText('Expired')

        # same as sortie, except expiry happens at Monday 00:00 UTC), 3 days for epoch adjust from Wed
        # % week for weekly timer diff since last Monday
        # day str listed separately for display formatting
        weekly_eta = round(time.mktime(time.strptime('1970-1-1T00:00:00.000Z', '%Y-%m-%dT%H:%M:%S.%fZ')) - time.mktime(time.gmtime()) - 3 * 86400) % (7 * 86400)
        self.WeeklyResetTimeDayStr.setText(str(weekly_eta // 86400) + 'd')
        self.WeeklyResetTimeStr.setText(f'{(weekly_eta // 3600) % 24}h {(weekly_eta // 60) % 60}m {weekly_eta % 60}s')

        # same as above, bounties reset at given time adjust, 2.5hr cycle
        bounty_eta = round(time.mktime(time.strptime('1970-1-1T00:00:00.000Z', '%Y-%m-%dT%H:%M:%S.%fZ')) - time.mktime(time.gmtime()) + 120) % (9000)
        self.BountyTimeStr.setText(f'{bounty_eta // 3600}h {(bounty_eta // 60) % 60}m {bounty_eta % 60}s')

        # baro reset adjust, 2 week cycle in full.
        # arrives at 2 days left, present < 2 days.
        baro_eta = round(time.mktime(time.strptime('1970-1-1T00:00:00.000Z', '%Y-%m-%dT%H:%M:%S.%fZ')) - time.mktime(time.gmtime()) + (10*86400 + 13 * 3600)) % (14 * 86400)
        if baro_eta >= 2 * 86400:
            self.BaroStateStr.setText('Baro Arrives in:')
            baro_eta = baro_eta - 2 * 86400
        else:
            self.BaroStateStr.setText('Baro Leaves in:')
        self.BaroTimeStr.setText(f'{str(baro_eta // 86400) + "d " if baro_eta // 86400 > 0 else ""}{(baro_eta // 3600) % 24}h {(baro_eta // 60) % 60}m {baro_eta % 60}s')


    def update_world_state(self):
        if self.ws.get_ws():
            self.redraw_content()
        else:
            self.network_error()
        
    def network_error(self):
        # slapped on later, just a dialog box for network error
        msg = QMessageBox()
        msg.setIcon(QMessageBox.Critical)
        msg.setText('Network Error')
        msg.setInformativeText('Could not connect to api.warframestat.us')
        msg.setWindowTitle('Error')
        msg.exec_()


if __name__ == '__main__':
    app_handle = QApplication(sys.argv)
    app = App()
    app.show()
    app_handle.exec_()