使用PyQt绘制精美的QTabWidget页签式股票代码一览表

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
# -*- coding: utf-8 -*-

import os
import sys
from PyQt4.QtCore import Qt
from pyqtgraph.Qt import QtCore, QtGui

import tushare as ts
import pandas as pd
from PyQt4.QtGui import QApplication

reload(sys)
sys.setdefaultencoding('utf8')

try:
    _fromUtf8 = QtCore.QString.fromUtf8
except AttributeError:
    def _fromUtf8(s):
        return s

try:
    _encoding = QtGui.QApplication.UnicodeUTF8
    def _translate(context, text, disambig):
        return QtGui.QApplication.translate(context, text, disambig, _encoding)
except AttributeError:
    def _translate(context, text, disambig):
        return QtGui.QApplication.translate(context, text, disambig)

class StockTableWidget():
    def __init__(self, *arg):
        pass

    def createStockTableWidget(self):

        self.tabWidget = QtGui.QTabWidget()
        sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding)
        sizePolicy.setHorizontalStretch(1)
        sizePolicy.setVerticalStretch(0)
        sizePolicy.setHeightForWidth(self.tabWidget.sizePolicy().hasHeightForWidth())
        self.tabWidget.setSizePolicy(sizePolicy)
        self.tabWidget.setCursor(QtGui.QCursor(QtCore.Qt.ArrowCursor))
        self.tabWidget.setMouseTracking(False)
        self.tabWidget.setAutoFillBackground(False)
        self.tabWidget.setObjectName(_fromUtf8("tabWidget"))
        self.tabWidget.setTabPosition(QtGui.QTabWidget.West)
        self.tab = QtGui.QWidget()
        self.tab.setObjectName(_fromUtf8("tab"))
        self.verticalLayout_2 = QtGui.QVBoxLayout(self.tab)
        self.verticalLayout_2.setObjectName(_fromUtf8("verticalLayout_2"))
        self.verticalLayout_2.setContentsMargins(0, 0, 0, 0)
        self.gridLayout = QtGui.QGridLayout()
        self.gridLayout.setObjectName(_fromUtf8("gridLayout"))
        self.verticalLayout_2.addLayout(self.gridLayout)
        self.tabWidget.addTab(self.tab, _fromUtf8(""))
        self.tab_2 = QtGui.QWidget()
        self.tab_2.setObjectName(_fromUtf8("tab_2"))
        self.tabWidget.addTab(self.tab_2, _fromUtf8(""))
        self.verticalLayout_3 = QtGui.QVBoxLayout(self.tab_2)
        self.verticalLayout_3.setObjectName(_fromUtf8("verticalLayout_3"))
        self.verticalLayout_3.setContentsMargins(0, 0, 0, 0)
        self.tab_3 = QtGui.QWidget()
        self.tab_3.setObjectName(_fromUtf8("tab_3"))
        self.tabWidget.addTab(self.tab_3, _fromUtf8(""))
        self.verticalLayout_4 = QtGui.QVBoxLayout(self.tab_3)
        self.verticalLayout_4.setObjectName(_fromUtf8("verticalLayout_4"))
        self.verticalLayout_4.setContentsMargins(0, 0, 0, 0)
        self.tab_4 = QtGui.QWidget()
        self.tab_4.setObjectName(_fromUtf8("tab_4"))
        self.tabWidget.addTab(self.tab_4, _fromUtf8(""))
        self.verticalLayout_5 = QtGui.QVBoxLayout(self.tab_4)
        self.verticalLayout_5.setObjectName(_fromUtf8("verticalLayout_5"))
        self.verticalLayout_5.setContentsMargins(0, 0, 0, 0)
        self.tab_5 = QtGui.QWidget()
        self.tab_5.setObjectName(_fromUtf8("tab_5"))
        self.tabWidget.addTab(self.tab_5, _fromUtf8(""))
        self.verticalLayout_6 = QtGui.QVBoxLayout(self.tab_5)
        self.verticalLayout_6.setObjectName(_fromUtf8("verticalLayout_6"))
        self.verticalLayout_6.setContentsMargins(0, 0, 0, 0)
        self.tab_6 = QtGui.QWidget()
        self.tab_6.setObjectName(_fromUtf8("tab_6"))
        self.tabWidget.addTab(self.tab_6, _fromUtf8(""))
        self.verticalLayout_7 = QtGui.QVBoxLayout(self.tab_6)
        self.verticalLayout_7.setObjectName(_fromUtf8("verticalLayout_7"))
        self.verticalLayout_7.setContentsMargins(0, 0, 0, 0)
        self.tab_7 = QtGui.QWidget()
        self.tab_7.setObjectName(_fromUtf8("tab_7"))
        self.tabWidget.addTab(self.tab_7, _fromUtf8(""))
        self.verticalLayout_8 = QtGui.QVBoxLayout(self.tab_7)
        self.verticalLayout_8.setObjectName(_fromUtf8("verticalLayout_8"))
        self.verticalLayout_8.setContentsMargins(0, 0, 0, 0)
        self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab), _translate("MainWindow", "沪深", None))
        self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_2), _translate("MainWindow", "港股", None))
        self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_3), _translate("MainWindow", "美股", None))
        self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_4), _translate("MainWindow", "期货", None))
        self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_5), _translate("MainWindow", "基金", None))
        self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_6), _translate("MainWindow", "全球", None))
        self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_7), _translate("MainWindow", "自选", None))

        self.treeWidget = QtGui.QTreeWidget()
        self.treeWidget.setObjectName(_fromUtf8("treeWidget"))
        self.treeWidget.setHeaderItem(QtGui.QTreeWidgetItem(
            [u'代码', u'名称', u'所属行业', u'地区', u'市盈率', u'流通股本(亿)', u'总股本(亿)', u'总资产(万)', u'流动资产', u'固定资产', u'公积金', u'每股公积金',
             u'每股收益', u'每股净资', u'市净率', u'上市日期', u'未分利润', u'每股未分配', u'收入同比(%)', u'利润同比(%)', u'毛利率(%)', u'净利润率(%)',
             u'股东人数']))
        self.treeWidget.header().setClickable(True)
        self.treeWidget.setSortingEnabled(True)
        self.treeWidget.sortItems(0, Qt.AscendingOrder)
        self.generateParameterTreeNode()
        self.verticalLayout_2.addWidget(self.treeWidget)
        return self.tabWidget

    def generateParameterTreeNode(self):
        self.treeWidget.setColumnCount(23)

        if os.path.exists('index.csv'):
            self.index = pd.read_csv('index.csv', dtype={'code': str})
        else:
            self.index = ts.get_index()
            self.index.to_csv('index.csv')
            self.index = pd.read_csv('index.csv', dtype={'code': str})

        if os.path.exists('stock_basics.csv'):
            self.stock_basics = pd.read_csv('stock_basics.csv', dtype={'code': str})
        else:
            self.stock_basics = ts.get_stock_basics()
            self.stock_basics.to_csv('stock_basics.csv')
            self.stock_basics = pd.read_csv('stock_basics.csv', dtype={'code': str})
        for indexs in self.stock_basics.index:
            try:
                code, name, industry, area, pe, outstanding, totals, totalAssets, liquidAssets, fixedAssets, reserved, reservedPerShare, esp, bvps, pb, timeToMarket, undp, perundp, rev, profit, gpr, npr, holders = self.stock_basics.loc[indexs].values[0:23]
                '''code,name,industry,area,pe,outstanding,totals,totalAssets,liquidAssets,fixedAssets,reserved,reservedPerShare,esp,bvps,pb,timeToMarket,undp,perundp, rev,profit,gpr,npr,holders'''
                self.insertTreeWidget(0, [code, name.decode('utf8'), industry.decode('utf8'), area.decode('utf8'), str(pe), str(outstanding), str(totals), str(totalAssets), str(liquidAssets), str(fixedAssets), str(reserved), str(reservedPerShare), str(esp), str(bvps), str(pb), str(timeToMarket), str(undp), str(perundp), str(rev), str(profit), str(gpr), str(npr), str(holders)], topLevel=None)
            except Exception:
                print Exception
        return

    def insertTreeWidget(self, level, datas, topLevel):
        if level == 0:
            topLevelItem = QtGui.QTreeWidgetItem(datas)
            self.treeWidget.addTopLevelItem(topLevelItem)
            return topLevelItem

if __name__ == '__main__':
    app = QApplication(sys.argv)
    stockTable = StockTableWidget().createStockTableWidget()
    stockTable.setWindowTitle(u'沪深证券代码一览表')
    stockTable.setStyleSheet('''QWidget{font-size: 12px;}''')
    stockTable.show()
    sys.exit(app.exec_())
    app.exec_()