from enum import Enum BagDesc_oper = Enum( 'BagDesc_oper', 'entity true l_and l_or l_not lt le eq ge gt') class BagDescription(object): def __init__(self, f_type, L_oper=None, R_oper=None, entity=""): self.f_type = f_type self.left_operand = L_oper self.right_operand = R_oper self.entity = entity self.sanity_check() def __repr__(self): if self.f_type == BagDesc_oper.entity: return self.entity if self.f_type == BagDesc_oper.true: return "TRUE" if self.f_type == BagDesc_oper.l_and: return "( " + repr(self.left_operand) + " & " + repr(self.right_operand) + " )" if self.f_type == BagDesc_oper.l_or: return "( " + repr(self.left_operand) + " | " + repr(self.right_operand) + " )" if self.f_type == BagDesc_oper.l_not: return "~" + repr(self.left_operand) if self.f_type == BagDesc_oper.lt: return repr(self.left_operand) + " < " + repr(self.right_operand) if self.f_type == BagDesc_oper.le: return repr(self.left_operand) + " <= " + repr(self.right_operand) if self.f_type == BagDesc_oper.eq: return repr(self.left_operand) + " == " + repr(self.right_operand) if self.f_type == BagDesc_oper.ge: return repr(self.left_operand) + " >= " + repr(self.right_operand) if self.f_type == BagDesc_oper.gt: return repr(self.left_operand) + " > " + repr(self.right_operand) def sanity_check(self): """Sanity checks""" for operand in (self.left_operand, self.right_operand): if operand: if not( isinstance(operand, BagDescription) or isinstance(operand, int)): raise RuntimeError( "Unexpected operand type for a bag: " + str(operand)) @classmethod def f_entity(cls, entity_name): return cls(BagDesc_oper.entity, entity=entity_name) @classmethod def f_TRUE(cls): return cls(BagDesc_oper.true) def __lt__(self, other): return BagDescription(BagDesc_oper.lt, L_oper=self, R_oper=other) def __le__(self, other): return BagDescription(BagDesc_oper.le, L_oper=self, R_oper=other) def __eq__(self, other): return BagDescription(BagDesc_oper.eq, L_oper=self, R_oper=other) def __ge__(self, other): return BagDescription(BagDesc_oper.ge, L_oper=self, R_oper=other) def __gt__(self, other): return BagDescription(BagDesc_oper.gt, L_oper=self, R_oper=other) def __and__(self, other): return BagDescription(BagDesc_oper.l_and, L_oper=self, R_oper=other) def __or__(self, other): return BagDescription(BagDesc_oper.l_or, L_oper=self, R_oper=other) def __invert__(self): return BagDescription(BagDesc_oper.l_not, L_oper=self) # EOF