Commit 64be2367 authored by Andrea Giannetti's avatar Andrea Giannetti
Browse files

Added relationships to all tables.

parent e0b1ceb0

etl/stg/stg_build_db_structure.py

+14 −0
Original line number Diff line number Diff line
@@ -21,7 +21,9 @@ engine = get_pg_engine(logger=logger) 
class GridPars(Base):

    __tablename__ = "grid_parameters"

    zipped_grid_name = Column(String(150), primary_key=True)

    species_and_partners = relationship("SpeciesAndPartners", cascade="all, delete-orphan")

    fits_cube_name = relationship("ModelPars", cascade="all, delete-orphan")

    line_pars = relationship("LinePars", cascade="all, delete-orphan")

    grid_type = Column(String)

    coordinate_system = Column(String)

    central_density = Column(Float)
@@ -52,6 +54,12 @@ class GridPars(Base): 


class LinePars(Base):

    __tablename__ = "lines_parameters"

    __table_args__ = (

        ForeignKeyConstraint(

            ('zipped_grid_name', 'run_id'),

            ['grid_parameters.zipped_grid_name', 'grid_parameters.run_id']

        ),

    )

    zipped_grid_name = Column(String(150), primary_key=True)

    lines_mode = Column(String(20))

    created_on = Column(DateTime)
@@ -60,6 +68,12 @@ class LinePars(Base): 


class SpeciesAndPartners(Base):

    __tablename__ = "species_and_partners"

    __table_args__ = (

        ForeignKeyConstraint(

            ('zipped_grid_name', 'run_id'),

            ['grid_parameters.zipped_grid_name', 'grid_parameters.run_id']

        ),

    )

    zipped_grid_name = Column(String(150), primary_key=True)

    species_to_include = Column(String(100), primary_key=True)

    molecular_abundance = Column(Float)