# PyP6XER
# Copyright (C) 2020, 2021 Hassan Emam <hassan@constology.com>
#
# This file is part of PyP6XER.
#
# PyP6XER library is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License v2.1 as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# PyP6XER is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with PyP6XER. If not, see <https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html>.
from xerparser.model.tasks import Tasks
from xerparser.model.classes.task import Task
[docs]
class WBS:
def __init__(self, params, data=None):
[docs]
self.wbs_id = int(params.get('wbs_id').strip()) if params.get('wbs_id') else None
[docs]
self.proj_id = int(params.get('proj_id').strip()) if params.get('proj_id') else None
[docs]
self.obs_id = params.get('obs_id').strip()
[docs]
self.seq_num = params.get('seq_num').strip()
[docs]
self.est_wt = params.get('est_wt')
[docs]
self.proj_node_flag = params.get('proj_node_flag').strip()
[docs]
self.sum_data_flag = params.get('sum_data_flag').strip()
[docs]
self.status_code = params.get('status_code').strip()
[docs]
self.wbs_short_name = params.get('wbs_short_name').strip()
[docs]
self.wbs_name = params.get('wbs_name').strip()
[docs]
self.phase_id = params.get('phase_id').strip()
[docs]
self.parent_wbs_id = int(params.get('parent_wbs_id')) if params.get('parent_wbs_id') else None
[docs]
self.ev_user_pct = params.get('ev_user_pct').strip()
[docs]
self.ev_etc_user_value = params.get('ev_etc_user_value').strip()
[docs]
self.orig_cost = params.get('orig_cost').strip()
[docs]
self.indep_remain_total_cost = params.get('indep_remain_total_cost').strip()
[docs]
self.ann_dscnt_rate_pct = params.get('ann_dscnt_rate_pct').strip()
[docs]
self.dscnt_period_type = params.get('dscnt_period_type').strip()
[docs]
self.indep_remain_work_qty = params.get('indep_remain_work_qty').strip()
[docs]
self.anticip_start_date = params.get('anticip_start_date').strip()
[docs]
self.anticip_end_date = params.get('anticip_end_date').strip()
[docs]
self.ev_compute_type = params.get('ev_compute_type').strip()
[docs]
self.ev_etc_compute_type = params.get('ev_etc_compute_type').strip()
[docs]
self.guid = params.get('guid').strip()
[docs]
self.tmpl_guid = params.get('tmpl_guid').strip()
[docs]
self.plan_open_state = params.get('plan_open_state').strip() if params.get('plan_open_state') else None
WBS.obj_list.append(self)
[docs]
def get_id(self):
return self.wbs_id
[docs]
def get_tsv(self):
tsv = ['%R', self.wbs_id, self.proj_id, self.obs_id, self.seq_num, self.est_wt,
self.proj_node_flag, self.sum_data_flag, self.status_code, self.wbs_short_name,
self.wbs_name, self.phase_id, self.parent_wbs_id, self.ev_user_pct, self.ev_etc_user_value,
self.orig_cost, self.indep_remain_total_cost, self.ann_dscnt_rate_pct, self.dscnt_period_type,
self.indep_remain_work_qty, self.anticip_start_date, self.anticip_end_date, self.ev_compute_type,
self.ev_etc_compute_type, self.guid, self.tmpl_guid, self.plan_open_state]
return tsv
@classmethod
[docs]
def get_json(cls):
root_nodes = list(filter(lambda x: WBS.find_by_id(x.parent_wbs_id) is None, cls.obj_list))
print(root_nodes)
json = dict()
for node in root_nodes:
json["node"] = node
json["level"] = 0
json["childs"] = []
json["childs"].append(cls.get_childs(node, 0))
print(json)
return json
@classmethod
[docs]
def get_childs(cls, node, level):
nodes_lst = list(filter(lambda x: x.parent_wbs_id == node.wbs_id, cls.obj_list))
nod = dict()
for node in nodes_lst:
nod["node"] = node
nod["level"] = level + 1
children = cls.get_childs(node, level + 1)
nod["childs"] = []
nod["childs"].append(children)
return nod
@classmethod
[docs]
def find_by_id(cls, ID):
obj = list(filter(lambda x: x.wbs_id == ID, cls.obj_list))
if obj:
return obj[0]
return None
@classmethod
[docs]
def find_by_project_id(cls, project_id):
return [ v for v in cls.obj_list if v.proj_id == project_id]
@property
[docs]
def activities(self):
return self.data.tasks.activities_by_wbs_id(self.wbs_id)
[docs]
def __repr__(self):
return self.wbs_name