mirror of
https://github.com/space-syndicate/space-station-14.git
synced 2026-02-15 00:54:51 +01:00
27 lines
923 B
Python
27 lines
923 B
Python
class YAMLEntity:
|
|
def __init__(self, id, name, description, suffix, parent_id = None):
|
|
self.id = id
|
|
self.name = name
|
|
self.description = description
|
|
self.suffix = suffix
|
|
self.parent_id = parent_id
|
|
|
|
|
|
class YAMLElements:
|
|
def __init__(self, items):
|
|
self.elements = list(map(lambda i: self.create_element(i), items))
|
|
|
|
def create_element(self, item):
|
|
if not 'id' in item:
|
|
return None
|
|
|
|
if item['type'] == 'entity':
|
|
entity = YAMLEntity(item['id'], item['name'] if 'name' in item else None,
|
|
item['description'] if 'description' in item else None,
|
|
item['suffix'] if 'suffix' in item else None,
|
|
item['parent'] if 'parent' in item else None
|
|
)
|
|
return entity
|
|
else:
|
|
return None
|