Coverage for middle_layer/testdata/domain_layer/entities/context_entity.py: 92.86%

1# Copyright 2024 Associated Universities, Inc. 

2# 

3# This file is part of Telescope Time Allocation Tools (TTAT). 

4# 

5# TTAT is free software: you can redistribute it and/or modify 

6# it under the terms of the GNU General Public License as published by 

7# the Free Software Foundation, either version 3 of the License, or 

8# any later version. 

9# 

10# TTAT is distributed in the hope that it will be useful, 

11# but WITHOUT ANY WARRANTY; without even the implied warranty of 

12# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

13# GNU General Public License for more details. 

14# 

15# You should have received a copy of the GNU General Public License 

16# along with TTAT. If not, see <https://www.gnu.org/licenses/>. 

17# 

18 

19from sqlalchemy import ForeignKey 

20from sqlalchemy.orm import Mapped, mapped_column, relationship 

21 

22from common.domain_layer.entities.base import Base 

23from solicit.domain_layer.entities.solicitation import Solicitation 

24 

25 

26class ContextEntity(Base): 

27 __tablename__ = "contexts" 

28 

29 solicitation_id: Mapped[int] = mapped_column( 

30 ForeignKey("solicitations.solicitation_id", ondelete="CASCADE"), primary_key=True 

31 ) 

32 solicitation: Mapped[Solicitation] = relationship( 

33 Solicitation, back_populates="contexts", single_parent=True, cascade="all, delete-orphan" 

34 ) 

35 seed: Mapped[int] 

36 do_notify: Mapped[bool] 

37 

38 def __init__(self, solicitation: Solicitation, seed: int = None, do_notify: bool = True): 

39 super().__init__(solicitation=solicitation, seed=seed, do_notify=do_notify) 

40 

41 def __json__(self): 

42 return {"solicitationId": self.solicitation_id, "seed": self.seed, "doNotify": self.do_notify}