import pickle


def decision(prediction):
    if prediction == 0:
        return "Nie_podejmuj_działania"
    elif prediction == 1:
        return "Zastosuj_nawóz"
    elif prediction == 2:
        return "Zastosuj_środek"
    elif prediction == 4:
        return "Zbierz"
    elif prediction == 5:
        return "Roślina_już_zgniła-zbierz_i_wyrzuć"


def test():
    for n in range(0, 2):
        if n == 0:
            print("############# Nie ma nawozu #############")
        else:
            print("############# Zastosowano nawóz #############")
        for s in range(0, 2):
            if s == 0:
                print("############# Nie ma środka ochrony #############")
            else:
                print("############# Zastosowano środek ochrony #############")
            for st in range(0, 101):
                val = tree.predict([[1, n, s, st]])
                print("Stan roślinki: ", st, " ", decision(val))


if __name__ == '__main__':
    filename = 'decisionTree.sav'
    tree = pickle.load(open(filename, 'rb'))
    test()