from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
from PyQt5.QtCore import *
import notify2
import sys
import os
import time
import threading as thd
import tushare as ts
import Txt2Png


class Mythread(QThread):
    # 定义信号,定义参数为str类型
    _signal = pyqtSignal()

    def __init__(self):
        super(Mythread, self).__init__()

    def run(self):
        while True:
            # 发出信号通知主线程
            self._signal.emit()
            # 让程序休眠
            time.sleep(2)


class Start:
    # 实时行情
    df = []
    # 托盘集合
    tpList = []
    # 最新price集合，用于和前后两次比较
    priceList = []
    # 提示上下限
    myPriceList = []
    # 是否有弹过提示
    notifyList = []
    # 行情时间
    current_time = ''

    def update_ui(self):
        # 计算时间，如果开市时间则去获取行情
        s = self.current_time.split(':')
        if len(s) > 1:
            # print(s[0])
            h = int(s[0])
            if h < 9 or 13 > h >= 12 or h >= 15:
                return

        # print('running')
        # 获取实时行情
        self.df = ts.get_realtime_quotes(self.codes)

        for i in self.df.index:
            # 昨收盘价
            pre = self.df.loc[i]['pre_close']
            # 当前时间
            self.current_time = self.df.loc[i]['time']
            # 当前价
            p = self.df.loc[i]['price']
            # 小数点后有3位，舍弃最后一位
            price = p[0:len(p) - 1]
            # 如果本次得到的price和上一次不一样，则刷新托盘图标
            if price != self.priceList[i]:
                # 记录变动后的新price
                self.priceList[i] = price

                # 转成图标
                Txt2Png.Price2Png(i, price, pre)

                # 刷新托盘图标
                self.tpList[i].setIcon(QIcon(str(i)+'.png'))

                # 判断是否需要提示
                if self.notifyList[i] is False:
                    price = float(self.df.loc[i]['price'])
                    pre = float(self.df.loc[i]['pre_close'])
                    percent = (price - pre) / pre * 100
                    # 检查预警上限
                    if price >= self.myPriceList[i]['max']:
                        self.show_message(self.df.loc[i]['name'], '恭喜！已达预期值' + str(self.myPriceList[i]['max']))
                        self.notifyList[i] = True
                    # 检查预警下限
                    elif price <= self.myPriceList[i]['min']:
                        self.show_message(self.df.loc[i]['name'], '小心！已到预警线' + str(self.myPriceList[i]['min']))
                        self.notifyList[i] = True
                    # 波动大于5个点需要提示
                    elif abs(percent) >= 5:
                        self.show_message(self.df.loc[i]['name'], str(format(percent, '.2f') + '%，注意波动！！！'))
                        self.notifyList[i] = True

    # 弹出提示
    def show_message(self, title, content):
        notify2.init('notify'+title)
        bubble = notify2.Notification(title, content)
        bubble.show()

    # 打开详情窗
    def show_window(self):
        # self.tpList[0].showMessage("提示", "你点了消息", icon=0)

        # 关闭所有窗口,也不关闭应用程序
        QApplication.setQuitOnLastWindowClosed(False)

        # QWidget窗口是PyQt5中所有用户界口对象的基本类.我们使用了QWidget默认的构造器.默认的构造器没有父类.一个没有父类的窗口被称为一个window.
        self.w = QWidget()
        # resize()方法调整了窗口的大小.被调整为250像素宽和250像素高.
        self.w.resize(1000, 100)
        # move()方法移动了窗口到屏幕坐标x=300, y=300的位置.
        self.w.move(300, 300)
        # 在这里我们设置了窗口的标题.标题会被显示在标题栏上.
        self.w.setWindowTitle('List')

        # 创建一个底layout
        form_layout = QFormLayout()

        # 标题
        layout = QHBoxLayout()
        layout.addWidget(QLabel('名称(代码)'))
        layout.addWidget(QLabel('当前价'))
        layout.addWidget(QLabel('百分比'))
        layout.addWidget(QLabel('最高价'))
        layout.addWidget(QLabel('最低价'))
        layout.addWidget(QLabel('开盘价'))
        layout.addWidget(QLabel('昨收价'))
        layout.addWidget(QLabel('成交量'))
        layout.addWidget(QLabel('成交额'))
        # 添加一行
        form_layout.addRow(layout)

        for i in self.df.index:
            price = float(self.df.loc[i]['price'])
            pre = float(self.df.loc[i]['pre_close'])
            percent = format((price - pre) / pre * 100, '.2f')

            name = self.df.loc[i]['name']
            code = self.df.loc[i]['code']
            high = format(float(self.df.loc[i]['high']), '.2f')
            low = format(float(self.df.loc[i]['low']), '.2f')
            open = format(float(self.df.loc[i]['open']), '.2f')
            volume = self.df.loc[i]['volume']
            amount = self.df.loc[i]['amount']

            layout = QHBoxLayout()
            layout.addWidget(QLabel(name+'\n('+code+')'))
            layout.addWidget(QLabel(str(price)))
            layout.addWidget(QLabel(str(percent)+' %'))
            layout.addWidget(QLabel(high))
            layout.addWidget(QLabel(low))
            layout.addWidget(QLabel(open))
            layout.addWidget(QLabel(str(pre)))
            layout.addWidget(QLabel(volume))
            layout.addWidget(QLabel(amount))
            # 间距为0
            layout.setSpacing(0)
            # 添加一行
            form_layout.addRow(layout)

        self.w.setLayout(form_layout)

        # show()方法将窗口显示在屏幕上.一个窗口是先在内存中被创建,然后显示在屏幕上的.
        self.w.show()

    def quit_app(self):
        # 关闭窗体程序
        QCoreApplication.instance().quit()
        # 在应用程序全部关闭后，TrayIcon其实还不会自动消失，
        # 直到你的鼠标移动到上面去后，才会消失，
        # 这是个问题，（如同你terminate一些带TrayIcon的应用程序时出现的状况），
        # 这种问题的解决我是通过在程序退出前将其setVisible(False)来完成的。
        for i in range(len(self.tpList)):
            self.tpList[i].setVisible(False)
            # 删除生成托盘图标
            os.remove(str(i)+'.png')

    def __init__(self, codes, myPriceList):

        # 要关注的代码
        self.codes = codes
        self.myPriceList = myPriceList

        # 初始化提示状态
        for i in range(len(self.codes)):
            self.notifyList.append(False)

        # pyqt窗口必须在QApplication方法中使用
        # 每一个PyQt5应用都必须创建一个应用对象.sys.argv参数是来自命令行的参数列表.Python脚本可以从shell里运行.这是我们如何控制我们的脚本运行的一种方法.
        app = QApplication(sys.argv)

        # 设置系统托盘图标的菜单
        menu = QMenu()
        a1 = QAction('&显示(Show)', triggered=self.show_window)  # 直接退出可以用qApp.quit
        a2 = QAction('&退出(Exit)', triggered=self.quit_app)  # 直接退出可以用qApp.quit
        menu.addAction(a1)
        menu.addAction(a2)

        for i in range(len(self.codes)):
            # 在系统托盘处显示图标
            tp = QSystemTrayIcon()
            tp.setIcon(QIcon('./bg.png'))

            # 设置右键弹出菜单
            tp.setContextMenu(menu)

            # 不调用show不会显示系统托盘
            tp.show()

            # 添加托盘到list中
            self.tpList.append(tp)

            # 定义好要保存的最新price
            self.priceList.append('')

        # 创建线程
        self.thread = Mythread()
        # 注册信号处理函数
        self.thread._signal.connect(self.update_ui)
        # 启动线程
        self.thread.start()

        # sys为了调用sys.exit(0)退出程序
        # 最后,我们进入应用的主循环.事件处理从这里开始.主循环从窗口系统接收事件,分派它们到应用窗口.如果我们调用了exit()方法或者主窗口被销毁,则主循环结束.sys.exit()方法确保一个完整的退出.环境变量会被通知应用是如何结束的.
        # exec_()方法是有一个下划线的.这是因为exec在Python中是关键字.因此,用exec_()代替.
        sys.exit(app.exec_())


# if __name__ == "__main__":
#     Start(['000014', '002358'])