Question

我想使用SQL Alchemy创建通用决策树。也就是说，每个节点具有零个或多个任何类型的子代，任务是使用树的根来评估某个表达式，这将使用扩展类将逻辑传递给子代。

我定义了以下基类：

from flask_sqlalchemy import Model, SQLAlchemy, DefaultMeta
from abc import ABCMeta, abstractmethod
from sqlalchemy import Column, Integer, String, Date, Boolean, ForeignKey, Text, Float, Unicode

db = SQLAlchemy(model_class=BaseModel)

class BaseModel(Model):
    pass

class ModelABCMeta(DefaultMeta, ABCMeta):
    pass

class RuleBaseNode(db.Model, metaclass=ModelABCMeta):
    """Generic base class representing a node in a decision tree"""

    id = Column(Integer, primary_key=True)
    type = Column(String(50))

    parent_node_type = Column(Unicode(255), nullable=True)
    parent_node_id = Column(Integer, nullable=True)

    parent_node = generic_relationship(parent_node_type, parent_node_id)

    __mapper_args__ = {
        'polymorphic_on': type,
        'polymorphic_identity': 'node'
    }

    @abstractmethod
    def eval(self, input) -> bool:
        """Evaluates an input to a boolean"""
        pass

现在的问题是如何添加节点子代的属性。

通常，我将relationship与backref一起使用，但是在documentation中找不到任何内容。

我想要这样的属性：

class RuleBaseNode(db.Model, metaclass=ModelABCMeta):
    ...
    @property
    def sub_nodes():
        return ...

现在我想我可以实现下面的某种形式，但是我想它不适用于查询抽象类

def get_sub_nodes(session, node):
    session.query(RuleBaseNode).filter(RuleBaseNode.parent_node == node).all()

用SQL Alchemy表示通用决策树

0 个答案: