From d1fa5636750d12a4c6d16054d0504d962204fff2 Mon Sep 17 00:00:00 2001
From: HellsSenju <irisha1234517@gmail.com>
Date: Mon, 14 Oct 2024 19:49:54 +0400
Subject: [PATCH] =?UTF-8?q?=D0=B5=D1=89=D0=B5=20=D0=BC=D0=BE=D0=B4=D0=B5?=
 =?UTF-8?q?=D0=BB=D1=8C=D0=BA=D0=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 db/csv_to_db.py                                  |  2 +-
 .../ch_experimentdb_experiment_data_model.py     | 10 ++++++++++
 db/models/experiment_data_model.py               | 16 ++++++++++++++++
 db/models/experiment_parameters_model.py         |  2 +-
 db/models/recycling_parameters_model.py          | 16 ++++++++++++++++
 5 files changed, 44 insertions(+), 2 deletions(-)
 create mode 100644 db/models/ch_experimentdb_experiment_data_model.py
 create mode 100644 db/models/experiment_data_model.py
 create mode 100644 db/models/recycling_parameters_model.py

diff --git a/db/csv_to_db.py b/db/csv_to_db.py
index abb11e1..a467702 100644
--- a/db/csv_to_db.py
+++ b/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)
 
diff --git a/db/models/ch_experimentdb_experiment_data_model.py b/db/models/ch_experimentdb_experiment_data_model.py
new file mode 100644
index 0000000..a7cc595
--- /dev/null
+++ b/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>"
\ No newline at end of file
diff --git a/db/models/experiment_data_model.py b/db/models/experiment_data_model.py
new file mode 100644
index 0000000..b994a89
--- /dev/null
+++ b/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>"
\ No newline at end of file
diff --git a/db/models/experiment_parameters_model.py b/db/models/experiment_parameters_model.py
index 69318ed..9fb6a85 100644
--- a/db/models/experiment_parameters_model.py
+++ b/db/models/experiment_parameters_model.py
@@ -14,4 +14,4 @@ class ExperimentParameters(Base):
     experiment_hash: Mapped[str]
 
     def __repr__(self):
-        return f"<LoadParameters>"
\ No newline at end of file
+        return f"<ExperimentParameters>"
\ No newline at end of file
diff --git a/db/models/recycling_parameters_model.py b/db/models/recycling_parameters_model.py
new file mode 100644
index 0000000..760975b
--- /dev/null
+++ b/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>"
\ No newline at end of file