diff --git a/wani.py b/wani.py index 57f2840..80bcd8d 100644 --- a/wani.py +++ b/wani.py @@ -123,6 +123,10 @@ CATEGORY_BASES = { "X": ['.'] * 1 } +class ComponentType(Enum): + Other = 0 + Core = 2 + Core2w = 3 class RestrictionType(Enum): Morphology = 0 @@ -524,6 +528,7 @@ class Component: def __init__(self, info): idx = info['cid'] name = info['name'] if 'name' in info else None + typ = ComponentType.Core if info['type'] == "core" else ComponentType.Other if 'status' not in info: status = ComponentStatus.Required @@ -543,6 +548,7 @@ class Component: self.next_element = [] self.representation = [] self.selection = {} + self.type = typ self.iter_ctr = 0