staplusconfigdialog.py

import sys, logging,  time, os, requests
from PyQt5.QtWidgets import QDialog, QMessageBox
from PyQt5 import uic
from PyQt5.QtCore import QEvent
from PyQt5.QtGui import QStandardItemModel, QStandardItem

from authorizedialog import AuthorizeDialog
from staplusconfig import STAplusConfig

logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
logger = logging.getLogger()

class STAplusConfigDialog(QDialog):
    token_data = {}
    staplusConfig = {}
    def __init__(self, parent=None):
        QDialog.__init__(self, parent)
        self.setModal(True)
        self.ui = uic.loadUi(os.path.join(os.path.dirname(os.path.abspath(__file__)), "stapusconfigdialog.ui"), self)
        self.ui.configButton.clicked.connect(self.startAuthorization)
        self.ui.buttonBox.rejected.connect(self.close)
        self.ui.buttonBox.accepted.connect(self.hide)

        self.ui.progress_bar.setStyleSheet("QProgressBar::chunk {background-color: green;}")

        self.staplusConfig = STAplusConfig()
        self.staplusConfig.signalProgress.connect(self.updateProgress)

        self.authorizeDialog = AuthorizeDialog(self)
        self.authorizeDialog.authorize.signalAuthApproval.connect(self.authorizationApproved)

        self.ui.license.installEventFilter(self)

    def eventFilter(self,target,event):
            if target == self.ui.license and event.type() == QEvent.MouseButtonPress:
                logger.debug("Button press")
                self.fillComboBox()

            return False

    def fillComboBox(self):
        licenses = self.getLicenses()
        logger.debug(f"licenses: {licenses}")
        model = QStandardItemModel()
        for l in licenses:
            logger.debug(f"id: {l['@iot.id']}")
            logger.debug(f"name: {l['name']}")
            it = QStandardItem()
            it.setData(l['@iot.id'])
            it.setText(l['name'])
            model.appendRow(it)

        self.ui.license.setModel(model)

    def getLicenses(self):
        logger.debug("getLicenses()")
        try:
            staplus_url = self.ui.staplus_url.text()
            url = staplus_url + 'Licenses' if staplus_url.endswith('/') else staplus_url + '/Licenses'
            res = requests.get(url, timeout=2) # 2 seconds
            return res.json().get("value")
        except requests.Timeout:
            logger.error("Error timeout")
        except requests.RequestException:
            logger.error("Error request exception")
        except requests.ConnectionError:
            logger.error("Error connction")
        except requests.HTTPError:
            logger.error("Error HTTP")


    def setDevice(self, device):
        logger.debug(f"setDevice(): {device}")
        self.staplusConfig.setDevice(device)

    def startAuthorization(self):
        if self.ui.license.currentText() is None or self.ui.license.currentText() == '':
            qm = QMessageBox()
            ret = qm.warning(self,'', "Please select a license!", QMessageBox.Ok)
            return

        if self.token_data == {} or self.token_data['access_token_expires'] < (int(time.time()) + 300): #5min expiration remaining at least
            self.authorizeDialog.show()
        else:
            self.authorizationApproved(self.token_data)

    def authorizationApproved(self, token_data):
        logger.debug(f"configFinished(): {token_data}")
        self.authorizeDialog.hide()
        self.token_data = token_data

        ix = self.ui.license.currentIndex()
        item = self.ui.license.model().item(ix,0)
        logger.debug(f"text: {item.text()}")
        logger.debug(f"data: {item.data()}")
        license_id = item.data()
        logger.debug(f"license: {license_id}")

        app_metadata = {
            'place_name': self.ui.place_name.text(),
            'place_description': self.ui.place_description.text(),
            'longitude': float(self.ui.longitude.text()),
            'latitude': float(self.ui.latitude.text()),
            'license': license_id,
            'staplus_url': self.ui.staplus_url.text()
        }
        logger.debug(f"preferred_username: {self.ui.preferred_username.text()}")
        logger.debug(f"token_data: {token_data}")
        logger.debug(f"app_metadata: {app_metadata}")
        self.staplusConfig.run(self.ui.preferred_username.text(), token_data, app_metadata)


    def updateProgress(self, x):
        logger.debug(f"updateProgress(): {x}")
        self.ui.progress_bar.setValue(x)