class MainWindow(QMainWindow, Ui_MainWindow):
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
self.invoice_form = InvoiceForm(self)
self.action_Account.triggered.connect(self.show_account_editor)
self.action_Update.triggered.connect(self.update_usage)
self.action_About.triggered.connect(self.about)
self.actionInvoice.triggered.connect(self.invoice_form.show)
self.network_access_manager = QNetworkAccessManager()
self.network_access_manager.finished.connect(self.fetched_usages)
self.network_access_manager.sslErrors.connect(self.allow_connection)
self.settings = QSettings('Nasim', 'BLUsage')
self.file_name = os.path.expanduser('~/.BLUsage.dat')
self.usage_model = BLUsage()
self.read_usage()
self.tree_model = TreeModel(self.usage_model.usage)
self.treeView.setModel(self.tree_model)
self.treeView.resizeColumnToContents(0)
if not self.usage_model.username:
self.show_account_editor()
self.treeView.header().setResizeMode(QHeaderView.Stretch)
self.treeView.header().setResizeMode(0, QHeaderView.ResizeToContents)
self.treeView.setAlternatingRowColors(True)
self.accountName.setText(self.usage_model.name)
self.progressBar.setHidden(True)
self.show_lastupdate()
def show_account_editor(self):
AccountDialog(self.usage_model, self).exec_()
self.accountName.setText(self.usage_model.name)
self.show_lastupdate()
self.write_usage()
def about(self):
QMessageBox.about(self, "BLUsage",
"""Bangla Lion bandwidth usage viewer.
Version 2.0
Author: M. Nasimul Haque
email: [email protected]
Web: http://www.nasim.me.uk""")
def update_usage(self):
if not all([self.usage_model.username, self.usage_model.password]):
QMessageBox.critical(self, 'No account',
'Please enter your account details first.')
return
request = QNetworkRequest(QUrl(self.usage_model.user_endpoint))
self.network_access_manager.post(request, self.usage_model.post_data)
self.updateButton.setEnabled(False)
self.progressBar.setHidden(False)
self.progressBar.setValue(0)
self.statusBar.showMessage("Please wait...")
def fetched_usages(self, reply):
self.updateButton.setEnabled(True)
self.progressBar.setHidden(True)
if reply.error() == QNetworkReply.NoError:
if not self.usage_model.parse(str(reply.readAll())):
title = "Parsing error"
message = self.usage_model.error
else:
title = None
self.tree_model.deleteLater()
self.tree_model = TreeModel(self.usage_model.usage)
self.treeView.setModel(self.tree_model)
self.show_lastupdate()
self.write_usage()
elif reply.error() in [QNetworkReply.AuthenticationRequiredError,
QNetworkReply.ContentAccessDenied]:
title = "Authentication error"
message = "Please check your account credentials."
else:
title = "An error occured"
message = reply.errorString() \
+ ".\nPlease check your internet connection."
if title:
QMessageBox.critical(self, title, message)
reply.deleteLater()
def allow_connection(self, reply):
reply.ignoreSslErrors()
def read_usage(self):
self.usage_model.name = self.settings.value('name')
self.usage_model.username = self.settings.value('username')
self.usage_model.password = self.settings.value('password')
self.usage_model.last_update = self.settings.value('last_update')
self.usage_model.capKB = int(self.settings.value('cap') or 0)
self.usage_model.totalKB = int(self.settings.value('total') or 0)
start = self.settings.value('start')
if not start:
today = QDate.currentDate()
start = QDate(today.year(), today.month(), 1)
self.usage_model.start = start
end = self.settings.value('end')
if not end:
end = start.addDays(29)
self.usage_model.end = end
def write_usage(self):
self.settings.setValue('name', self.usage_model.name)
self.settings.setValue('username', self.usage_model.username)
self.settings.setValue('password', self.usage_model.password)
self.settings.setValue('start', self.usage_model.start)
self.settings.setValue('end', self.usage_model.end)
self.settings.setValue('last_update', self.usage_model.last_update)
self.settings.setValue('cap', self.usage_model.capKB)
self.settings.setValue('total', self.usage_model.totalKB)
def show_lastupdate(self):
self.totalLabel.setText("Totals usage: <b>%s</b>" %
(self.usage_model.smart_bytes(self.usage_model.totalKB),))
remaining = self.usage_model.remaining()
if isinstance(remaining, int):
self.remainingLabel.setText("Remaining: <b>%s</b>" %
(self.usage_model.smart_bytes(remaining),))
else:
self.remainingLabel.setText("Remaining: <b>Unlimited</b>")
try:
if not self.usage_model.last_update:
self.statusBar.showMessage("Ready")
else:
self.statusBar.showMessage("Last updated on: " +
self.usage_model.last_update.toString())
except:
pass
