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"" \ 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"" \ 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"" \ No newline at end of file + return f"" \ 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"" \ No newline at end of file