Coverage for polar/models/notification.py: 88%

1from typing import TYPE_CHECKING 2 ctx1ab

2from uuid import UUID 2 ctx1ab

3 

4from sqlalchemy import ForeignKey, String, Uuid 2 ctx1ab

5from sqlalchemy.dialects.postgresql import JSONB 2 ctx1ab

6from sqlalchemy.orm import Mapped, mapped_column, relationship 2 ctx1ab

7 

8from polar.kit.db.models import RecordModel 2 ctx1ab

9from polar.types import JSONDict 2 ctx1ab

10 

11if TYPE_CHECKING: 11 ↛ 12line 11 didn't jump to line 12 because the condition on line 11 was never true2 ctx1ab

12 from polar.models import User 

13 

14 

15class Notification(RecordModel): 2 ctx1ab

16 __tablename__ = "notifications" 2 ctx1ab

17 

18 user_id: Mapped[UUID] = mapped_column( 2 ctx1ab

19 Uuid, ForeignKey("users.id", ondelete="cascade"), nullable=False, index=True 

20 ) 

21 type: Mapped[str] = mapped_column(String, nullable=False) 2 ctx1ab

22 payload: Mapped[JSONDict] = mapped_column(JSONB, nullable=False, default=dict) 2 ctx1ab

23 

24 user: Mapped["User"] = relationship("User", lazy="raise") 2 ctx1ab