Wrapper class to handle ws data -> gui

extra_widgets.py

@@ -3,7 +3,7 @@ from PyQt5 import QtWidgets, QtGui, QtCore
 from datetime import datetime, timedelta
 
 class Fissure(QtWidgets.QWidget):
-    def __init__(self, parent, fissure_info, i):
+    def __init__(self, parent, fissure_info = {}, i = 0):
         super().__init__(parent)
         self.i = str(i)
         self.fissure_info = fissure_info
@@ -44,16 +44,19 @@ class Fissure(QtWidgets.QWidget):
         self.FissureNode.setAlignment(QtCore.Qt.AlignLeading|QtCore.Qt.AlignLeft|QtCore.Qt.AlignTop)
         self.FissureNode.setObjectName("FissureNode_" + self.i)
         self.gridLayout_3.addWidget(self.FissureNode, 1, 0, 1, 1)
-        self.FissureType = QtWidgets.QLabel(Fissure)
+        self.FissureType = QtWidgets.QLabel(self)
         self.FissureType.setAlignment(QtCore.Qt.AlignCenter)
         self.FissureType.setObjectName("FissureType_" + self.i)
         self.gridLayout_3.addWidget(self.FissureType, 1, 1, 1, 1)
 
     def update_gui(self):
+        try:
             self.FissureTier.setText(f"{self.fissure_info['tier']} - T{self.fissure_info['tierNum']}")
             self.update_timer(self)
             self.FissureNode.setText(f"{self.fissure_info['node']}")
             self.FissureType.setText(f"{'Steel Path' if self.fissure_info['isHard'] else ('Storm' if self.fissure_info['isStorm'] else '')}")
+        except:
+            pass
 
     def udpdate_timer(self):
         eta = datetime.strptime(self.fissure_info['expiry'], '%Y-%m-%dT%H:%M:%SZ') - datetime.now()
@@ -63,7 +66,7 @@ class Fissure(QtWidgets.QWidget):
 
 
 class Invasion(QtWidgets.QWidget):
-    def __init__(self, parent, invasion_info, i):
+    def __init__(self, parent, invasion_info = {}, i = 0):
         super().__init__(parent)
         self.i = str(i)
         self.invasion_info = invasion_info
@@ -78,7 +81,7 @@ class Invasion(QtWidgets.QWidget):
         self.setSizePolicy(sizePolicy)
         self.setMinimumSize(QtCore.QSize(0, 100))
         self.setObjectName("Invasion_" + self.i)
-        self.gridLayout_4 = QtWidgets.QGridLayout(self.Invasion)
+        self.gridLayout_4 = QtWidgets.QGridLayout(self)
         self.gridLayout_4.setObjectName("gridLayout_4_" + self.i)
         self.Reward2 = QtWidgets.QLabel(self)
         font = QtGui.QFont()
@@ -106,16 +109,19 @@ class Invasion(QtWidgets.QWidget):
         self.gridLayout_4.addWidget(self.Reward1, 0, 0, 1, 1)
 
     def update_gui(self):
+        try:
             faction_colors = {'Gineer': 'red', 'Corpus': 'blue', 'Infested': 'green'}
             self.Reward1.setText(f"{self.invasion_info['attackerReward']['itemString']}")
             self.Reward2.setText(f"{self.invasion_info['defenderReward']['itemString']}")
             self.InvasionProgress.setStyleSheet(f'QProgressBar {{background-color:{faction_colors.get(self.invasion_info["defendingFaction"])};border-color:red; border:0px}}\nQProgressBar::chunk {{background-color:{faction_colors.get(self.invasion_info["attackingFaction"])};}}')
+        except:
+            pass
 
     def update_progress(self):
         self.InvasionProgress.setProperty('value', self.invasion_info['completion'])
 
 class Alert(QtWidgets.QWidget):
-    def __init__(self, parent, alert_info, i):
+    def __init__(self, parent, alert_info = {}, i = 0):
         super().__init__(parent)
         self.i = str(i)
         self.alert_info = alert_info
@@ -151,8 +157,11 @@ class Alert(QtWidgets.QWidget):
         self.horizontalLayout_6.addWidget(self.AlertTimer)
 
     def update_gui(self):
+        try:
             self.AlertReward.setText(self.alert_info['mission']['reward']['itemString'])
             self.update_timer()
+        except:
+            pass
 
     def update_timer(self):
         eta = datetime.strptime(self.alert_info['expiry'], '%Y-%m-%dT%H:%M:%SZ') - datetime.now()
@@ -160,7 +169,7 @@ class Alert(QtWidgets.QWidget):
         self.AlertTimer.setText(f'{eta.days}d {eta.seconds // 3600}h {(eta.seconds // 60) % 60}m {eta.seconds % 60}s')
 
 class BaroItem(QtWidgets.QWidget):
-    def __init__(self, parent, item_info, i):
+    def __init__(self, parent, item_info = {}, i = 0):
         super().__init__(parent)
         self.item_info = item_info
         self.i = str(i)
@@ -182,10 +191,13 @@ class BaroItem(QtWidgets.QWidget):
         self.gridLayout_7.addWidget(self.ItemName, 0, 0, 1, 1)
 
     def update_gui(self):
+        try:
             self.ItemName.setText(self.item_info['item'])
+        except:
+            pass
 
 class Event(QtWidgets.QWidget):
-    def __init__(self, parent, event_info, i):
+    def __init__(self, parent, event_info = {}, i = 0):
         super().__init__(parent)
         self.event_info = event_info
         self.i = str(i)
@@ -220,8 +232,11 @@ class Event(QtWidgets.QWidget):
         self.horizontalLayout_11.addWidget(self.EventTimer)
 
     def update_gui(self):
+        try:
             self.EventName.setText(self.event_info['description'])
             self.update_timer()
+        except:
+            pass
 
     def update_timer(self):
         eta = datetime.strptime(self.event_info['expiry'], '%Y-%m-%dT%H:%M:%SZ') - datetime.now()
@@ -229,7 +244,7 @@ class Event(QtWidgets.QWidget):
         self.EventTimer.setText(f'{eta.days}d {eta.seconds // 3600}h {(eta.seconds // 60) % 60}m {eta.seconds % 60}s')
 
 class NewsBlurb(QtWidgets.QWidget):
-    def __init__(self, parent, news_info, i):
+    def __init__(self, parent, news_info = {}, i = 0):
         super().__init__(parent)
         self.i = str(i)
         self.news_info = news_info
@@ -271,8 +286,11 @@ class NewsBlurb(QtWidgets.QWidget):
         self.horizontalLayout_12.addWidget(self.NewsBlurbRelTimer)
 
     def update_gui(self):
+        try:
             self.NewsBlurbHyperLink.setText(f'<a href=\"{self.news_info["link"]}\">{self.news_info["message"]}</a>')
             self.update_timer()
+        except:
+            pass
 
     def update_timer(self):
         eta = datetime.strptime(self.news_info['expiry'], '%Y-%m-%dT%H:%M:%SZ') - datetime.now()

gui.py

@@ -4,7 +4,7 @@ from PyQt5.QtWidgets import QApplication
 import sys
 import gui_base
 
-class ExampleApp(QtWidgets.QMainWindow, gui_base.Ui_MainWindow):
+class GUI(QtWidgets.QMainWindow, gui_base.Ui_MainWindow):
     def __init__(self, parent=None):
         super().__init__(parent)
         self.setupUi(self)

main.py

@@ -0,0 +1,60 @@
+from gui import GUI
+from api import WorldState
+from extra_widgets import *
+
+from PyQt5.QtWidgets import QApplication, QMessageBox
+
+import sys
+
+
+class App(GUI):
+    def __init__(self):
+        super().__init__()
+        self.ws = WorldState()
+        while self.ws.ws == {}:
+            self.network_error()
+            sys.exit(1)
+        del self.Fissure, self.Invasion, self.Alert, self.BaroItem, self.Event, self.NewsBlurb
+        self.redraw_content()
+
+
+        def redraw_content(self):
+            self.redraw_content_dynamic()
+
+        def redraw_content_dynamic(self):
+            self.Fissure_list = [Fissure(self.scrollAreaWidgetContents, fissure_info, fissure_i) for (fissure_i, fissure_info) in enumerate(self.ws.fissures)]
+            [self.verticalLayout_6.addWidget(i) for i in self.Fissure_list]
+            self.Invasion_list = [Invasion(self.scrollAreaWidgetContents_2, invasion_info, invasion_i) for (invasion_i, invasion_info) in enumerate(self.ws.invasions)]
+            [self.verticalLayout_8.addWidget(i) for i in self.Invasion_list]
+            self.Alert_list = [Alert(self.scrollAreaWidgetContents_3, alerta_info, alert_i) for (alert_i, alert_info) in enumerate(self.ws.alerts)]
+            [self.verticalLayout_16.addWidget(i) for i in self.Alert_list]
+            if len(self.ws.baro_items) == 0:
+                self.stackedWidget_2.setCurrentIndex(0)
+            else:
+                self.stackedWidget_2.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.scrollAreaWidgetContents_4, event_info, event_i) for (event_i, event_info) in enumerate(self.ws.events)]
+            [self.verticalLayout_31.addWidget(i) for i in self.events_list]
+            self.newsblurb_list = [NewsBlurb(self.scrollAreaWidgetContents_5, newsblurb_info, newsblurb_i) for (newsblurb_i, newsblurb_info) in enumerate(self.ws.news)]
+            [self.verticalLayout_33.addWidget(i) for i in self.newsblurb_list]
+
+        def redraw_content_static(self):
+            pass
+        
+        
+    def network_error(self):
+        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_()
+