114 lines
4.2 KiB
Python
114 lines
4.2 KiB
Python
|
import uuid
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
from sqlalchemy import Column, String, Integer, BigInteger, Boolean, Text, ARRAY, Numeric
|
||
|
|
from sqlalchemy import DateTime, ForeignKey, func
|
||
|
|
from sqlalchemy.dialects.postgresql import UUID, JSONB, INET
|
||
|
|
from sqlalchemy.orm import DeclarativeBase, relationship
|
||
|
|
|
||
|
|
|
||
|
|
class Base(DeclarativeBase):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class Workspace(Base):
|
||
|
|
__tablename__ = "workspaces"
|
||
|
|
|
||
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
|
|
name = Column(String(255), nullable=False)
|
||
|
|
description = Column(Text)
|
||
|
|
domain = Column(String(100))
|
||
|
|
created_by = Column(String(255))
|
||
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
||
|
|
updated_at = Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())
|
||
|
|
|
||
|
|
files = relationship("File", back_populates="workspace")
|
||
|
|
|
||
|
|
|
||
|
|
class File(Base):
|
||
|
|
__tablename__ = "files"
|
||
|
|
|
||
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
|
|
workspace_id = Column(UUID(as_uuid=True), ForeignKey("workspaces.id", ondelete="CASCADE"))
|
||
|
|
filename = Column(String(500), nullable=False)
|
||
|
|
original_name = Column(String(500), nullable=False)
|
||
|
|
file_type = Column(String(50))
|
||
|
|
file_size = Column(BigInteger)
|
||
|
|
storage_path = Column(Text)
|
||
|
|
parsed_path = Column(Text)
|
||
|
|
status = Column(String(50), default="uploaded")
|
||
|
|
error_msg = Column(Text)
|
||
|
|
metadata = Column(JSONB, default={})
|
||
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
||
|
|
updated_at = Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())
|
||
|
|
|
||
|
|
workspace = relationship("Workspace", back_populates="files")
|
||
|
|
tasks = relationship("Task", back_populates="file")
|
||
|
|
|
||
|
|
|
||
|
|
class Task(Base):
|
||
|
|
__tablename__ = "tasks"
|
||
|
|
|
||
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
|
|
task_type = Column(String(100), nullable=False)
|
||
|
|
status = Column(String(50), default="pending")
|
||
|
|
payload = Column(JSONB, default={})
|
||
|
|
result = Column(JSONB)
|
||
|
|
error_msg = Column(Text)
|
||
|
|
progress = Column(Integer, default=0)
|
||
|
|
file_id = Column(UUID(as_uuid=True), ForeignKey("files.id"))
|
||
|
|
celery_task_id = Column(String(255))
|
||
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
||
|
|
updated_at = Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())
|
||
|
|
completed_at = Column(DateTime(timezone=True))
|
||
|
|
|
||
|
|
file = relationship("File", back_populates="tasks")
|
||
|
|
|
||
|
|
|
||
|
|
class ComplianceReport(Base):
|
||
|
|
__tablename__ = "compliance_reports"
|
||
|
|
|
||
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
|
|
file_id = Column(UUID(as_uuid=True), ForeignKey("files.id"))
|
||
|
|
regulation_domains = Column(ARRAY(Text))
|
||
|
|
overall_risk_level = Column(String(20))
|
||
|
|
risk_score = Column(Numeric(5, 2))
|
||
|
|
findings = Column(JSONB, default=[])
|
||
|
|
recommendations = Column(JSONB, default=[])
|
||
|
|
report_markdown = Column(Text)
|
||
|
|
llm_model = Column(String(100))
|
||
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
||
|
|
|
||
|
|
|
||
|
|
class RegulationSource(Base):
|
||
|
|
__tablename__ = "regulation_sources"
|
||
|
|
|
||
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
|
|
name = Column(String(255), nullable=False)
|
||
|
|
url = Column(Text, nullable=False)
|
||
|
|
source_type = Column(String(50), default="webpage")
|
||
|
|
domain = Column(String(100))
|
||
|
|
fetch_interval = Column(Integer, default=86400)
|
||
|
|
is_active = Column(Boolean, default=True)
|
||
|
|
last_fetched_at = Column(DateTime(timezone=True))
|
||
|
|
last_hash = Column(String(64))
|
||
|
|
fetch_config = Column(JSONB, default={})
|
||
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
||
|
|
|
||
|
|
|
||
|
|
class RegulationUpdate(Base):
|
||
|
|
__tablename__ = "regulation_updates"
|
||
|
|
|
||
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||
|
|
source_id = Column(UUID(as_uuid=True), ForeignKey("regulation_sources.id"))
|
||
|
|
title = Column(String(500))
|
||
|
|
url = Column(Text)
|
||
|
|
change_type = Column(String(50))
|
||
|
|
summary = Column(Text)
|
||
|
|
raw_content = Column(Text)
|
||
|
|
diff_content = Column(Text)
|
||
|
|
is_notified = Column(Boolean, default=False)
|
||
|
|
importance = Column(String(20), default="normal")
|
||
|
|
fetched_at = Column(DateTime(timezone=True), server_default=func.now())
|
||
|
|
published_at = Column(DateTime(timezone=True))
|