Coverage for polar/models/organization_review.py: 92%

1from datetime import datetime 2 ctx1ab

2from enum import StrEnum 2 ctx1ab

3from typing import TYPE_CHECKING, Any 2 ctx1ab

4from uuid import UUID 2 ctx1ab

5 

6from sqlalchemy import TIMESTAMP, ForeignKey, String, Text, Uuid 2 ctx1ab

7from sqlalchemy.dialects.postgresql import JSONB 2 ctx1ab

8from sqlalchemy.orm import Mapped, declared_attr, mapped_column, relationship 2 ctx1ab

9 

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

11 

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

13 from polar.models.organization import Organization 

14 

15 

16class OrganizationReview(RecordModel): 2 ctx1ab

17 """Model to store AI validation responses for organizations.""" 

18 

19 class Verdict(StrEnum): 2 ctx1ab

20 PASS = "PASS" 2 ctx1ab

21 FAIL = "FAIL" 2 ctx1ab

22 UNCERTAIN = "UNCERTAIN" 2 ctx1ab

23 

24 class AppealDecision(StrEnum): 2 ctx1ab

25 APPROVED = "approved" 2 ctx1ab

26 REJECTED = "rejected" 2 ctx1ab

27 

28 __tablename__ = "organization_reviews" 2 ctx1ab

29 

30 organization_id: Mapped[UUID] = mapped_column( 2 ctx1ab

31 Uuid, 

32 ForeignKey("organizations.id", ondelete="cascade"), 

33 nullable=False, 

34 unique=True, 

35 index=True, 

36 ) 

37 

38 verdict: Mapped[Verdict] = mapped_column(String, nullable=False) 2 ctx1ab

39 risk_score: Mapped[float] = mapped_column(nullable=False) 2 ctx1ab

40 violated_sections: Mapped[list[str]] = mapped_column( 2 ctx1ab

41 JSONB, nullable=False, default=list 

42 ) 

43 reason: Mapped[str] = mapped_column(Text, nullable=False) 2 ctx1ab

44 

45 timed_out: Mapped[bool] = mapped_column(nullable=False, default=False) 2 ctx1ab

46 model_used: Mapped[str] = mapped_column(String, nullable=False) 2 ctx1ab

47 

48 organization_details_snapshot: Mapped[dict[str, Any]] = mapped_column( 2 ctx1ab

49 JSONB, nullable=False, default=dict 

50 ) 

51 

52 validated_at: Mapped[datetime] = mapped_column( 2 ctx1ab

53 TIMESTAMP(timezone=True), nullable=False, default=lambda: datetime.now() 

54 ) 

55 

56 # Appeal fields 

57 appeal_submitted_at: Mapped[datetime | None] = mapped_column( 2 ctx1ab

58 TIMESTAMP(timezone=True), nullable=True, default=None 

59 ) 

60 appeal_reason: Mapped[str | None] = mapped_column(Text, nullable=True, default=None) 2 ctx1ab

61 appeal_reviewed_at: Mapped[datetime | None] = mapped_column( 2 ctx1ab

62 TIMESTAMP(timezone=True), nullable=True, default=None 

63 ) 

64 appeal_decision: Mapped[AppealDecision | None] = mapped_column( 2 ctx1ab

65 String, nullable=True, default=None 

66 ) 

67 

68 @declared_attr 2 ctx1ab

69 def organization(cls) -> Mapped["Organization"]: 2 ctx1ab

70 return relationship("Organization", lazy="raise", back_populates="review") 2 ctx1ab

71 

72 def __repr__(self) -> str: 2 ctx1ab

73 return f"OrganizationReview(id={self.id}, organization_id={self.organization_id}, verdict={self.verdict})"