mirror of
https://github.com/space-syndicate/space-station-14.git
synced 2026-10-10 02:41:18 +03:00
c6264ee41a
Co-authored-by: mhamster <81412348+mhamsterr@users.noreply.github.com> Co-authored-by: Morb0 <14136326+Morb0@users.noreply.github.com>
26 lines
926 B
Python
26 lines
926 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
|