proplogicgenerator.py

import random


class Node:
    def __init__(self, name, layer) -> None:
        self.truth = random.choice([True, False])
        self.id = name
        self.children = []
        self.nots = []
        self.ops = []
        self.layer = layer
        self.generate_children()
        
        
    def generate_children(self):
        numlist = [1,1,1,1,2,2,2]
        for l in range(0,self.layer):
            numlist.append(0)
        num = random.choice(numlist)
        if num >= 1:
            self.children.append(Node(self.id * 2, self.layer + 1))
            self.nots.append(random.choice([True, False]))
        if num >= 2:
            self.children.append(Node(self.id * 2 + 1, self.layer + 1))
            self.nots.append(random.choice([True, False]))
            self.ops.append(random.choice(['and', 'or']))
            
            
    def evaluate(self):
        if len(self.children) == 1:
            self.truth =  self.children[0].evaluate() == self.nots[0]
        elif len(self.children) == 2:
            if self.nots[0] == 0:
                a = self.children[0].evaluate()
            else:
                a = self.children[0].evaluate() == False
            if self.nots[0] == 0:
                b = self.children[0].evaluate()
            else:
                b = self.children[0].evaluate() == False
            if self.ops[0] == 'and':
                self.truth =  a == b == True
            elif self.ops[0] == 'or':
                self.truth = a==True or b==True
        return self.truth
    
    def print_statements(self):
        for child in self.children:
            child.print_statements()
            print("")
        if len(self.children) == 0:
            print('{} = {}'.format(self.id, self.truth))
        elif len(self.children) == 1:
            print('if {} = {} then {} = True'.format(self.children[0].id, self.nots[0], self.id))
        elif len(self.children) == 2:
            print('if {} = {} {} {} = {} then {} = True'.format(self.children[0].id, self.nots[0], self.ops[0], self.children[1].id, self.nots[1], self.id))
            
            
            
print("start")
root = Node(1, 0)
root.evaluate()
root.print_statements()
print("")
print("{} is {}".format(root.id, root.truth))