#!/usr/bin/env python


from PyQt5.QtGui import QIcon
from PyQt5.QtWidgets import (QAction, QApplication, QDialog, QLineEdit, QMessageBox, QMenu, QSystemTrayIcon,
                             QTextEdit, )
from PyQt5.QtCore import (QThread)
from PyQt5 import QtCore
import json, time
from grab import Grab


def checkBalance():
    g = Grab()
    url = 'https://lk.ttk.ru/po/login'
    urlAccount = 'https://lk.ttk.ru/po/rest/client/accounts/'

    login = '*****'
    password = '******'
    uagent = "Opera/10.80 (Windows NT 6.1; WOW64) Presto-parser/2.12.388 Version/12.14"
    params = {
        'username': login,
        'j_username': login,
        'password': password,
        'loginSource': 'form'
    }

    g.setup(post=params, user_agent=uagent)
    g.go(url)
    text = g.go(urlAccount)
    jsonText = json.loads(text.body.decode('utf-8'))
    balance = jsonText[0]['accountBalance']
    return balance


class Window(QDialog):
    Signal = QtCore.pyqtSignal(str)

    def __init__(self):
        super(Window, self).__init__()
        self.Signal.connect(lambda s: self.setTextTrigger(s))
        self.initUi()

    def setTextTrigger(self, s):
        self.setTextToBaloon(s)
        self.showMessage()

    def initUi(self):
        self.trayIconMenu = QMenu(self)
        self.trayIcon = QSystemTrayIcon(self)
        self.titleEdit = QLineEdit("Ваш баланс").text()

        self.createActions()
        self.createTrayIcon()
        self.trayIcon.setToolTip("Чекер баланса")
        self.setTextToBaloon('пока нету баланса')

    def iconActivated(self, reason):
        if reason in (QSystemTrayIcon.Trigger, QSystemTrayIcon.DoubleClick):
            self.showMessage()

    def showMessage(self):
        icon = QSystemTrayIcon.Information
        self.trayIcon.showMessage(self.titleEdit,
                                  self.bodyEdit.toPlainText(), icon, 1000)

    def setTextToBaloon(self, text):
        self.bodyEdit = QTextEdit()
        self.bodyEdit.setPlainText(text)

    def closeEvent(self, event):
        pass

    def createActions(self):
        self.quitAction = QAction("&Quit", self, triggered=QApplication.instance().quit)

    def createTrayIcon(self):
        self.trayIconMenu.addAction(self.quitAction)

        self.trayIcon.setContextMenu(self.trayIconMenu)
        self.trayIcon.setIcon(QIcon('./images/bad.png'))
        self.trayIcon.show()
        self.trayIcon.activated.connect(self.iconActivated)


class Checker(QThread):
    def run(self):
        self.timer()

    def timer(self):
        self.firstBalance = 0
        while 1:
            balance = round(checkBalance(), 2)
            if balance != self.firstBalance:
                self.firstBalance = balance
                window.Signal.emit(str(balance))
            time.sleep(60)


if __name__ == '__main__':

    import sys

    app = QApplication(sys.argv)
    QApplication.setQuitOnLastWindowClosed(False)

    if not QSystemTrayIcon.isSystemTrayAvailable():
        QMessageBox.critical(None, "Systray", "Критическая ошибка")
        sys.exit(1)

    window = Window()
    checker = Checker()
    checker.start()
    app.exec_()


# window.show()
# sys.exit(app.exec_())