91 lines
5.2 KiB
Python
91 lines
5.2 KiB
Python
"""PMFlow OS change review models."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Any, Optional
|
|
|
|
from sqlalchemy import BigInteger, Float, ForeignKey, Integer, Text, TIMESTAMP, VARCHAR
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from .base import Base
|
|
from .types import JSONB
|
|
|
|
|
|
class PMFlowChangeSet(Base):
|
|
__tablename__ = "pmflow_change_sets"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
project_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("projects.id"), nullable=False, index=True)
|
|
thread_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_threads.id"), nullable=False, index=True)
|
|
source_message_id: Mapped[int] = mapped_column(
|
|
BigInteger, ForeignKey("pmflow_thread_messages.id"), nullable=False, index=True
|
|
)
|
|
status: Mapped[str] = mapped_column(VARCHAR(32), nullable=False, default="open", index=True)
|
|
summary: Mapped[str] = mapped_column(Text, nullable=False)
|
|
headline: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
|
|
extraction_scope: Mapped[str] = mapped_column(VARCHAR(64), nullable=False, default="latest_message")
|
|
routing_meta: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)
|
|
llm_call_id: Mapped[Optional[str]] = mapped_column(VARCHAR(128), nullable=True)
|
|
created_by: Mapped[int] = mapped_column(BigInteger, ForeignKey("users.id"), nullable=False, index=True)
|
|
reviewed_at: Mapped[Optional[datetime]] = mapped_column(TIMESTAMP, nullable=True)
|
|
committed_at: Mapped[Optional[datetime]] = mapped_column(TIMESTAMP, nullable=True)
|
|
committed_by: Mapped[Optional[int]] = mapped_column(BigInteger, ForeignKey("users.id"), nullable=True, index=True)
|
|
|
|
|
|
class PMFlowChangeItem(Base):
|
|
__tablename__ = "pmflow_change_items"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
change_set_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_change_sets.id"), nullable=False, index=True)
|
|
project_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("projects.id"), nullable=False, index=True)
|
|
group: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
|
|
object_type: Mapped[str] = mapped_column(VARCHAR(64), nullable=False)
|
|
object_id: Mapped[Optional[str]] = mapped_column(VARCHAR(128), nullable=True)
|
|
title: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
|
|
summary: Mapped[str] = mapped_column(Text, nullable=False)
|
|
diff: Mapped[str] = mapped_column(Text, nullable=False)
|
|
payload: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)
|
|
edited_payload: Mapped[Optional[Any]] = mapped_column(JSONB, nullable=True)
|
|
confidence: Mapped[Optional[float]] = mapped_column(Float, nullable=True)
|
|
risk_level: Mapped[str] = mapped_column(VARCHAR(16), nullable=False, default="medium")
|
|
status: Mapped[str] = mapped_column(VARCHAR(32), nullable=False, default="pending_review", index=True)
|
|
reviewed_at: Mapped[Optional[datetime]] = mapped_column(TIMESTAMP, nullable=True)
|
|
|
|
|
|
class PMFlowApproval(Base):
|
|
__tablename__ = "pmflow_approvals"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
project_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("projects.id"), nullable=False, index=True)
|
|
change_set_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_change_sets.id"), nullable=False, index=True)
|
|
change_item_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_change_items.id"), nullable=False, index=True)
|
|
actor_user_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("users.id"), nullable=False, index=True)
|
|
action: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
|
|
decision: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
|
|
before_status: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
|
|
after_status: Mapped[str] = mapped_column(VARCHAR(32), nullable=False)
|
|
before_payload: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)
|
|
after_payload: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)
|
|
reason: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
|
|
|
|
|
class PMFlowEvidenceLink(Base):
|
|
__tablename__ = "pmflow_evidence_links"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
project_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("projects.id"), nullable=False, index=True)
|
|
change_set_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("pmflow_change_sets.id"), nullable=False, index=True)
|
|
change_item_id: Mapped[Optional[int]] = mapped_column(
|
|
BigInteger, ForeignKey("pmflow_change_items.id"), nullable=True, index=True
|
|
)
|
|
source_type: Mapped[str] = mapped_column(VARCHAR(64), nullable=False, index=True)
|
|
source_id: Mapped[str] = mapped_column(VARCHAR(128), nullable=False, index=True)
|
|
title: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
|
|
excerpt: Mapped[str] = mapped_column(Text, nullable=False)
|
|
display_label: Mapped[str] = mapped_column(VARCHAR(255), nullable=False)
|
|
evidence_metadata: Mapped[Any] = mapped_column(JSONB, nullable=False, default=dict)
|
|
|
|
|
|
__all__ = ["PMFlowChangeSet", "PMFlowChangeItem", "PMFlowApproval", "PMFlowEvidenceLink"]
|