Feature_selection_json_columns.py

import json

import _KEYS_DICT


class JsonColumns:
    stock_id = ""
    Op_buy_sell_Cat = _KEYS_DICT.Op_buy_sell.POS
    path = ""
    vgood16 = []
    good9 = []
    reg4 = []
    low1 = []
    def __init__(self, stock_id, Option_Cat : _KEYS_DICT.Op_buy_sell ):
        NUM_VGOOD = 9
        NUM_GOOD = 4
        self.vgood16 = []
        self.good9 = []
        self.reg4 = []
        self.low1 = []
        self.stock_id = stock_id
        self.Op_buy_sell_Cat = Option_Cat
        dict_best_feat = {}
        self.path = "plots_relations/best_selection_" + self.stock_id  + "_" + self.Op_buy_sell_Cat.value + ".json"
        print("plots_relations/best_selection_" + self.stock_id  + "_" + self.Op_buy_sell_Cat.value + ".json")
        with open(self.path) as handle:
            dict_best_feat = json.loads(handle.read())['index']  # TODO remove 'index from json
        self.low1 = dict_best_feat.pop('1', None)  # borro la primera por no saturar
        # For wildcard files that combine multiple actions
        if  "@" in self.path:
            NUM_VGOOD = 26
            NUM_GOOD = 11

        {k: self.vgood16.extend(v) for k, v in dict_best_feat.items() if NUM_VGOOD <= int(k)}
        {k: self.good9.extend(v) for k, v in dict_best_feat.items() if NUM_VGOOD > int(k) >= NUM_GOOD}
        {k: self.reg4.extend(v) for k, v in dict_best_feat.items() if int(k) < NUM_GOOD}

    def get_vGood_and_Good(self):
        return self.vgood16 + self.good9

    def get_Regulars(self):
        return self.vgood16 + self.good9 + self.reg4

    def get_ALL_Good_and_Low(self):
        return self.vgood16 + self.good9 + self.reg4 + self.low1

    def get_Dict_JsonColumns(self):
        Dict_JsonColumns = {
            _KEYS_DICT.MODEL_TYPE_COLM.VGOOD.value:    self.vgood16,
            _KEYS_DICT.MODEL_TYPE_COLM.GOOD.value:     self.get_vGood_and_Good(),
            _KEYS_DICT.MODEL_TYPE_COLM.REG.value:      self.get_Regulars(),
            _KEYS_DICT.MODEL_TYPE_COLM.LOW.value:      self.get_ALL_Good_and_Low()
        }
        return Dict_JsonColumns