File size: 1,143 Bytes
7a511fb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5cb75d9
 
 
 
7a511fb
 
 
 
 
 
 
 
 
 
 
5cb75d9
7a511fb
 
 
 
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
from enum import Enum as PyEnum

from sqlalchemy import (
    Column,
    DateTime,
    Enum,
    Float,
    ForeignKey,
    Integer,
    String,
    func,
)
from sqlalchemy.dialects.postgresql import UUID
from pydantic import BaseModel
from ._base import Base


class AnalysisType(PyEnum):
    STRENGTHS = "STRENGTHS"
    WEAKNESSES = "WEAKNESSES"
    DEFICIENCIES = "DEFICIENCIES"

class AnalysisStatus(PyEnum):
    ACCEPTED = "ACCEPTED"
    REJECTED = "REJECTED"
    PENDING = "PENDING"

class Analysis(Base):
    __tablename__ = "analysis"
    id = Column(UUID(as_uuid=True), primary_key=True)
    evaluation_id = Column(
        UUID(as_uuid=True),
        ForeignKey("evaluations.id", ondelete="CASCADE"),
        nullable=False,
    )
    insights = Column(String, nullable=False)
    analysis_type = Column(Enum(AnalysisType), nullable=False)
    status = Column(Enum(AnalysisStatus, name="analysisstatus"), nullable=False, default=AnalysisStatus.PENDING)
    created_at = Column(DateTime(), nullable=False, default=func.now())
    updated_at = Column(
        DateTime(), nullable=False, default=func.now(), onupdate=func.now()
    )