store objects to postgresql

.gitignore

@@ -1,2 +1,2 @@
 __pycache__/
-
+.env

experiment.py

@@ -1,85 +1,62 @@
-from dataclasses import dataclass, field
+from sqlalchemy import Column, String, Float, Integer, DateTime, ForeignKey
-from typing import List, Any
+from sqlalchemy.orm import relationship, declarative_base
+from dataclasses import dataclass
 
-Interval = tuple[float, float] | None
+Base = declarative_base()
 
 
 @dataclass
-class Measurement:
+class Measurement(Base):
-    """These are the fields that represent the outputs of the experiment."""
+    __tablename__ = "measurements"
+    name = Column(String, nullable=False)
+    unit = Column(String, nullable=False)
+    experiment = relationship("Experiment", back_populates="measurements")
+    experiment_id = Column(Integer, ForeignKey("experiments.id"))
+    values = relationship("MeasurementValue", back_populates="measurement")
+    id = Column(Integer, primary_key=True, autoincrement=True)
 
-    name: str
+    def take_measurement(self, value: float):
-    unit: str
+        self.values.append(MeasurementValue(value=value))
-    interval: Interval = None
-    values: List[float] = field(default_factory=list)
 
 
 @dataclass
-class Variable:
+class MeasurementValue(Base):
-    """These are the fields that represent the inputs of the experiment."""
+    __tablename__ = "measurement_values"
-
+    value = Column(Float, nullable=False)
-    name: str
+    measurement = relationship("Measurement", back_populates="values")
-    unit: str
+    measurement_id = Column(Integer, ForeignKey("measurements.id"))
-    interval: Interval = None
+    id = Column(Integer, primary_key=True, autoincrement=True)
-    value: Any = None
 
 
 @dataclass
-class Experiment:
+class Variable(Base):
-    """This is the data class that represents the experiment."""
+    __tablename__ = "variables"
+    name = Column(String, nullable=False)
+    unit = Column(String, nullable=False)
+    value = Column(Integer, nullable=False)  # Storing Any type as string
+    experiment = relationship("Experiment", back_populates="variables")
+    experiment_id = Column(Integer, ForeignKey("experiments.id"))
+    id = Column(Integer, primary_key=True, autoincrement=True)
 
-    experiment: str
-    date: str
-    measurements: dict[str, Measurement] = field(default_factory=dict)
-    variables: dict[str, Variable] = field(default_factory=dict)
 
-    def to_json(self):
+@dataclass
-        # Convert the data class to a dictionary, which can be easily converted to JSON.
+class Experiment(Base):
-        import json
+    __tablename__ = "experiments"
+    experiment = Column(String, nullable=False)
+    date = Column(DateTime, nullable=False)
+    measurements = relationship("Measurement", back_populates="experiment")
+    variables = relationship("Variable", back_populates="experiment")
+    id = Column(Integer, primary_key=True, autoincrement=True)
 
-        return json.dumps(self, default=lambda o: o.__dict__, sort_keys=True, indent=4)
+    def get_measurement(self, source: str):
-
+        # Get a measurement from the experiment.
-    def from_json(self, json_str):
+        for m in self.measurements:
-        # Convert a JSON string to a data class.
+            if m.name == source:
-        import json
+                return m
-
-        data = json.loads(json_str)
-        self.__dict__.update(data)
-
-    def add_measurement(self, measurement: Measurement):
-        # Add a measurement to the experiment.
-        if measurement.name in self.measurements:
-            raise ValueError(
-                f"Measurement {measurement.name} already exists in the experiment."
-            )
-        self.measurements[measurement.name] = measurement
-
-    def add_measurements(self, measurements: List[Measurement]):
-        # Add a list of measurements to the experiment.
-        for measurement in measurements:
-            self.add_measurement(measurement)
-
-    def take_measurement(self, source: str, value: float):
-        # Add a measurement to the experiment.
-        if source not in self.measurements:
         raise ValueError(f"Measurement {source} not found in the experiment.")
-        self.measurements[source].values.append(value)
 
-    def add_variable(self, variable: Variable):
+    def take_measurement(self, source: str | Measurement, value: float):
-        # Add a variable to the experiment.
+        measurement: Measurement = (
-        if variable.name in self.variables:
+            source if isinstance(source, Measurement) else self.get_measurement(source)
-            raise ValueError(
-                f"Variable {variable.name} already exists in the experiment."
         )
-        self.variables[variable.name] = variable
+        measurement.take_measurement(value)
-
-    def add_variables(self, variables: List[Variable]):
-        # Add a list of variables to the experiment.
-        for variable in variables:
-            self.add_variable(variable)
-
-    def set_variable(self, source: str, value: Any):
-        # Set a variable in the experiment.
-        if source not in self.variables:
-            raise ValueError(f"Variable {source} not found in the experiment.")
-        self.variables[source].value = value

shelling.py

@@ -1,12 +1,7 @@
-from dataclasses import dataclass
-
 from experiment import Experiment, Measurement, Variable
 
 
-@dataclass
+def make_shelling_experiment(
-class ShellingExperiment(Experiment):
-    def __init__(
-        self,
     date: str,
     drum_rpm: int,
     paddle_shaft_rpm: int,
@@ -14,30 +9,29 @@ class ShellingExperiment(Experiment):
     tilt_angle: float,
     moisture_content: float,
     feed_rate: int = 500,
-        pecan_variety: str = "desirable",
+    pecan_variety_index: int = 0,
-    ):
+) -> Experiment:
 
-        super().__init__(experiment="Shelling Analysis", date=date)
+    return Experiment(
-
+        date=date,
-        variables = [
+        experiment="shelling",
-            Variable("drum-rpm", "rpm", (30, 40), drum_rpm),
+        variables=[
-            Variable("paddle-shaft-rpm", "rpm", (400, 800), paddle_shaft_rpm),
+            Variable(name="drum-rpm", unit="rpm", value=drum_rpm),
-            Variable("ring-gap", "in", value=ring_gap),
+            Variable(name="paddle-shaft-rpm", unit="rpm", value=paddle_shaft_rpm),
-            Variable("tilt-angle", "deg", (2, 5), tilt_angle),
+            Variable(name="ring-gap", unit="in", value=ring_gap),
-            Variable("feed-rate", "lb/hr", (300, 500), feed_rate),
+            Variable(name="tilt-angle", unit="deg", value=tilt_angle),
-            Variable("moisture-content", "%", (5, 9), moisture_content),
+            Variable(name="feed-rate", unit="lb/hr", value=feed_rate),
-            Variable("pecan-variety", "", value=pecan_variety),
+            Variable(name="moisture-content", unit="%", value=moisture_content),
-        ]
+            Variable(name="pecan-variety", unit="", value=pecan_variety_index),
-        self.add_variables(variables)
+        ],
-
+        measurements=[
-        measurements = [
+            Measurement(name="bin1-weight", unit="lb"),
-            Measurement("bin1-weight", "lb"),
+            Measurement(name="bin2-weight", unit="lb"),
-            Measurement("bin2-weight", "lb"),
+            Measurement(name="bin3-weight", unit="lb"),
-            Measurement("bin3-weight", "lb"),
+            Measurement(name="recirculated-weight", unit="lb"),
-            Measurement("recirculated-weight", "lb"),
+            Measurement(name="final-discharge-weight", unit="lb"),
-            Measurement("final-discharge-weight", "lb"),
+            Measurement(name="bin1-half-yield", unit="%"),
-            Measurement("bin1-half-yield", "%"),
+            Measurement(name="bin2-half-yield", unit="%"),
-            Measurement("bin2-half-yield", "%"),
+            Measurement(name="bin3-half-yield", unit="%"),
-            Measurement("bin3-half-yield", "%"),
+        ],
-        ]
+    )
-        self.add_measurements(measurements)