hiderfong
69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
from datetime import datetime
|
|
from sqlalchemy import Column, Integer, String, Text, ForeignKey, DateTime, JSON, Float, Boolean
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from app.core.database import Base
|
|
|
|
|
|
class Category(Base):
|
|
__tablename__ = "category"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
parent_id = Column(Integer, ForeignKey("category.id"), nullable=True)
|
|
level = Column(Integer, default=1) # 1, 2, 3
|
|
code = Column(String(50), unique=True, nullable=False)
|
|
name = Column(String(100), nullable=False)
|
|
description = Column(Text)
|
|
sort_order = Column(Integer, default=0)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
parent = relationship("Category", remote_side=[id], backref="children")
|
|
|
|
|
|
class DataLevel(Base):
|
|
__tablename__ = "data_level"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
code = Column(String(20), unique=True, nullable=False) # L1, L2, L3, L4, L5
|
|
name = Column(String(50), nullable=False)
|
|
description = Column(Text)
|
|
color = Column(String(20), default="#999999")
|
|
control_requirements = Column(JSON)
|
|
sort_order = Column(Integer, default=0)
|
|
|
|
|
|
class RecognitionRule(Base):
|
|
__tablename__ = "recognition_rule"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
template_id = Column(Integer, ForeignKey("classification_template.id"), nullable=False)
|
|
category_id = Column(Integer, ForeignKey("category.id"), nullable=True)
|
|
level_id = Column(Integer, ForeignKey("data_level.id"), nullable=True)
|
|
rule_type = Column(String(20), nullable=False) # regex, keyword, enum, ml
|
|
rule_name = Column(String(100))
|
|
rule_content = Column(Text, nullable=False) # regex pattern / keyword list / enum values
|
|
target_field = Column(String(20), default="column_name") # column_name, comment, sample_data
|
|
priority = Column(Integer, default=100)
|
|
hit_count = Column(Integer, default=0)
|
|
is_active = Column(Boolean, default=True)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
template = relationship("ClassificationTemplate", back_populates="rules")
|
|
category = relationship("Category")
|
|
level = relationship("DataLevel")
|
|
|
|
|
|
class ClassificationTemplate(Base):
|
|
__tablename__ = "classification_template"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String(100), nullable=False)
|
|
industry_type = Column(String(50), default="insurance_property")
|
|
version = Column(String(20), default="1.0")
|
|
description = Column(Text)
|
|
is_builtin = Column(Boolean, default=False)
|
|
is_active = Column(Boolean, default=True)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
rules = relationship("RecognitionRule", back_populates="template", cascade="all, delete-orphan")
|