mapcat/mapcat/database/atomic_coadd.py at ac9b30d544cfbef9dbdf08d37a9f55719bb321b4 · simonsobs/mapcat · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
"""
Atomic map coadds
"""

from typing import TYPE_CHECKING

from sqlmodel import Field, Relationship, SQLModel

if TYPE_CHECKING:
    from .atomic_map import AtomicMapTable

from .links import AtomicMapToCoaddTable, CoaddMapToCoaddTable  # pragma: no cover


class AtomicMapCoaddTable(SQLModel, table=True):
    __tablename__ = "atomic_map_coadds"

    coadd_id: int = Field(primary_key=True)

    coadd_name: str = Field()
    prefix_path: str = Field()

    platform: str = Field()
    interval: str = Field()
    start_time: float = Field()
    stop_time: float = Field()
    freq_channel: str = Field()
    geom_file_path: str = Field()
    split_label: str = Field()

    atomic_maps: list["AtomicMapTable"] = Relationship(
        back_populates="coadds",
        link_model=AtomicMapToCoaddTable,
    )

    child_coadds: list["AtomicMapCoaddTable"] = Relationship(
        back_populates="parent_coadds",
        link_model=CoaddMapToCoaddTable,
        sa_relationship_kwargs={
            "primaryjoin": "AtomicMapCoaddTable.coadd_id == CoaddMapToCoaddTable.parent_coadd_id",
            "secondaryjoin": "AtomicMapCoaddTable.coadd_id == CoaddMapToCoaddTable.child_coadd_id",
        },
    )

    parent_coadds: list["AtomicMapCoaddTable"] = Relationship(
        back_populates="child_coadds",
        link_model=CoaddMapToCoaddTable,
        sa_relationship_kwargs={
            "primaryjoin": "AtomicMapCoaddTable.coadd_id == CoaddMapToCoaddTable.child_coadd_id",
            "secondaryjoin": "AtomicMapCoaddTable.coadd_id == CoaddMapToCoaddTable.parent_coadd_id",
        },
    )