diff --git a/day07.py b/day07.py index 51d4279..0329efa 100644 --- a/day07.py +++ b/day07.py @@ -21,11 +21,8 @@ def getBagContent(bag_line): return return_dict for content in contents.split(", "): - match_groups = re.match(content_re, content) - if not match_groups: - print("ERROR: content '%s' cannot be matched" % content) - - return_dict[color][match_groups.group(2)] = int(match_groups.group(1)) + content_count, content_color = re.match(content_re, content).groups() + return_dict[color][content_color] = int(content_count) return return_dict