from segment import Segment
from config_loader import ConfigLoader

class Rack:
    def __init__(self, capacity_mah=2500):
        self.config = ConfigLoader()
        self.segments = [
            Segment(segment['id'])
            for segment in self.config.config['segments']
        ]

    def measure_voltage(self):
        total_voltage = 0
        for segment in self.segments:
            for subsegment in segment.subsegments:
                total_voltage += subsegment.get_subset_voltage()
        return total_voltage

    def measure_power(self):
        # Logic to measure power
        pass