From fa52ed6dbc1667b2cadd082351607554b232cd04 Mon Sep 17 00:00:00 2001 From: Dion Moult Date: Wed, 28 Apr 2021 16:54:54 +1000 Subject: [PATCH] Fix bug where you might have an activity that is not part of any WBS from P6 --- src/ifcp6/ifcp6/p62ifc.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/ifcp6/ifcp6/p62ifc.py b/src/ifcp6/ifcp6/p62ifc.py index 72e7da193b..c53a05fe25 100644 --- a/src/ifcp6/ifcp6/p62ifc.py +++ b/src/ifcp6/ifcp6/p62ifc.py @@ -38,7 +38,11 @@ class P62Ifc: for activity in project.findall("pr:Activity", ns): activity_id = activity.find("pr:ObjectId", ns).text - self.wbs[activity.find("pr:WBSObjectId", ns).text]["activities"].append(activity_id) + wbs_id = activity.find("pr:WBSObjectId", ns).text + if not wbs_id: + print("No WBS ID found for activity", activity_id) + continue + self.wbs[wbs_id]["activities"].append(activity_id) self.activities[activity_id] = { "Name": activity.find("pr:Name", ns).text, "Identification": activity.find("pr:Id", ns).text,