QFreqtradeUI/qjsontablemodel.cpp at master · orkblutt/QFreqtradeUI · GitHub

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
#include "qjsontablemodel.h"
#include <QJsonObject>
#include <QColor>
#include <QFont>


QJsonTableModel::QJsonTableModel( const QJsonTableModel::Header& header, QObject * parent )
    : QAbstractTableModel( parent )
    , m_header( header )
{

}

bool QJsonTableModel::setJson(const QJsonDocument &json)
{
    return setJson( json.array() );
}

bool QJsonTableModel::setJson( const QJsonArray& array )
{
    beginResetModel();
    m_json = array;
    endResetModel();
    return true;
}

QVariant QJsonTableModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if( role != Qt::DisplayRole )
    {
        return QVariant();
    }

    switch( orientation )
    {
    case Qt::Horizontal:
        return m_header[section]["title"];
    case Qt::Vertical:
        //return section + 1;
        return QVariant();
    default:
        return QVariant();
    }

}

int QJsonTableModel::rowCount(const QModelIndex &parent ) const
{
    Q_UNUSED(parent);
    return m_json.size();
}

int QJsonTableModel::columnCount(const QModelIndex &parent ) const
{
    Q_UNUSED(parent);
    return m_header.size();
}


QJsonObject QJsonTableModel::getJsonObject( const QModelIndex &index ) const
{
    const QJsonValue& value = m_json[index.row() ];
    return value.toObject();
}

QVariant QJsonTableModel::data( const QModelIndex &index, int role ) const
{
    switch( role )
    {
    case Qt::DisplayRole:
    {
        QJsonObject obj = getJsonObject( index );
        const QString& key = m_header[index.column()]["index"];
        if( obj.contains( key ))
        {
            QJsonValue v = obj[ key ];

            if( v.isString() )
            {
                return v.toString();
            }
            else if( v.isDouble() )
            {
                return v.toDouble();
            }
            else
            {
                return QVariant();
            }
        }
        else
        {
            return QVariant();
        }
    }
    case Qt::BackgroundRole:
    {
        QJsonObject obj = getJsonObject( index );
        if(obj.contains("current_profit"))
        {
            if(obj["current_profit"].toDouble() <= 0)
                return QVariant(QColor(Qt::darkRed));
            else
                return QVariant(QColor(Qt::darkGreen));

        }
        else if(obj.contains("profit"))
        {
            if(obj["profit"].toDouble() <= 0)
                return QVariant(QColor(Qt::darkRed));
            else
                return QVariant(QColor(Qt::darkGreen));
        }
        else if(obj.contains("close_profit"))
        {
            if(obj["close_profit"].toDouble() <= 0)
                return QVariant(QColor(Qt::darkRed));
            else
                return QVariant(QColor(Qt::darkGreen));
        }
        return QVariant();
    }
    case Qt::FontRole:
    {
        QFont font;
        font.setBold(true);
        return font;
    }
    case Qt::ToolTipRole:
        return QVariant();
    default:
        return QVariant();
    }
}