from time import sleep

from CoreLibrary.Param_Model import Param_Model
from CoreLibrary.Measurement import Measurement


def main(param_model: Param_Model):
    sensor = param_model.sensors["mess1"]
    driver = param_model.drivers["treib1"]
    driver.set_config({"speed": 2})
    status = driver.get_status()

    for speed in [1.0, 2.0, 3.0]:
        print(speed)
        driver.set_config({"speed": speed})

        measurement = param_model.new_measurement(name=f"penis_{speed}", writer="CSV")
        measurement.attach("mess1")

        sensor.start_measuring()
        sleep(1)
        sensor.stop_measuring()
    print("test")
    sensor.exit()
    driver.exit()

if __name__ == '__main__':
    pd = Param_Model()
    main(pd)