segment.py 540 B

123456789101112131415
  1. from subsegment import Subsegment
  2. from config_loader import ConfigLoader
  3. class Segment:
  4. def __init__(self, segment_id):
  5. self.segment_id = segment_id
  6. self.config = ConfigLoader()
  7. segment_config = self.config.get_segment_config(segment_id)
  8. if not segment_config:
  9. raise ValueError(f"No configuration found for segment {segment_id}")
  10. self.subsegments = [
  11. Subsegment(self.segment_id, subseg['id'])
  12. for subseg in segment_config['subsegments']
  13. ]