db/csv_to_db.py

@@ -1,7 +1,7 @@
 import pandas as pd
 
 
-df = pd.read_csv('./files/experiment_parameters.csv')
+df = pd.read_csv('./files/recycling_parameters.csv')
 headers = df.columns.tolist()
 print(headers)
 

db/models/ch_experimentdb_experiment_data_model.py

@@ -0,0 +1,10 @@
+from db.db import Base
+from sqlalchemy.orm import Mapped, mapped_column
+
+class ChExperimentDBExperimentData(Base):
+    __tablename__ = 'ChExperimentDBExperimentData'
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+
+    def __repr__(self):
+        return f"<ChExperimentDBExperimentData>"

db/models/experiment_data_model.py

@@ -0,0 +1,16 @@
+from db.db import Base
+from sqlalchemy.orm import Mapped, mapped_column
+
+class ExperimentData(Base):
+    __tablename__ = 'ExperimentData'
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    direction: Mapped[float]
+    temperature: Mapped[float]
+    nox: Mapped[float]
+    co2: Mapped[float]
+    co: Mapped[float]
+    file_id: Mapped[str]
+
+    def __repr__(self):
+        return f"<ExperimentData>"

db/models/experiment_parameters_model.py

@@ -14,4 +14,4 @@ class ExperimentParameters(Base):
     experiment_hash: Mapped[str]
 
     def __repr__(self):
-        return f"<LoadParameters>"
+        return f"<ExperimentParameters>"

db/models/recycling_parameters_model.py

@@ -0,0 +1,16 @@
+from db.db import Base
+from sqlalchemy.orm import Mapped, mapped_column
+
+class RecyclingParameters(Base):
+    __tablename__ = 'RecyclingParameters'
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    load_id: Mapped[int]
+    recycling_level: Mapped[int]
+    co2: Mapped[float]
+    n2: Mapped[float]
+    h2o: Mapped[float]
+    o2: Mapped[float]
+
+    def __repr__(self):
+        return f"<RecyclingParameters>"