27 lines
574 B
Python
27 lines
574 B
Python
|
import pillarsdk
|
||
|
|
|
||
|
|
|
||
|
|
class SpecialFolderNode(pillarsdk.Node):
|
||
|
|
NODE_TYPE = 'SPECIAL'
|
||
|
|
|
||
|
|
|
||
|
|
class UpNode(SpecialFolderNode):
|
||
|
|
NODE_TYPE = 'UP'
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__()
|
||
|
|
self['_id'] = 'UP'
|
||
|
|
self['node_type'] = self.NODE_TYPE
|
||
|
|
|
||
|
|
|
||
|
|
class ProjectNode(SpecialFolderNode):
|
||
|
|
NODE_TYPE = 'PROJECT'
|
||
|
|
|
||
|
|
def __init__(self, project):
|
||
|
|
super().__init__()
|
||
|
|
|
||
|
|
assert isinstance(project, pillarsdk.Project), 'wrong type for project: %r' % type(project)
|
||
|
|
|
||
|
|
self.merge(project.to_dict())
|
||
|
|
self['node_type'] = self.NODE_TYPE
|