mirror of https://github.com/logos-co/open-law.git
29 lines
973 B
Python
29 lines
973 B
Python
|
from app import db
|
||
|
from app.models.utils import BaseModel
|
||
|
|
||
|
|
||
|
class Collection(BaseModel):
|
||
|
__tablename__ = "collections"
|
||
|
|
||
|
# need to redeclare id to use it in the parrent relationship
|
||
|
id = db.Column(db.Integer, primary_key=True)
|
||
|
label = db.Column(db.String(1024), unique=False, nullable=False)
|
||
|
about = db.Column(db.Text, unique=False, nullable=False)
|
||
|
is_root = db.Column(db.Boolean, default=False)
|
||
|
is_leaf = db.Column(db.Boolean, default=False)
|
||
|
|
||
|
# Foreign keys
|
||
|
version_id = db.Column(db.ForeignKey("book_versions.id"))
|
||
|
parrent_id = db.Column(db.ForeignKey("collections.id"))
|
||
|
|
||
|
# Relationships
|
||
|
version = db.relationship("BookVersion")
|
||
|
parrent = db.relationship("Collection", remote_side=[id])
|
||
|
children = db.relationship(
|
||
|
"Collection", backref=db.backref("parent", remote_side=[id]), viewonly=True
|
||
|
)
|
||
|
sections = db.relationship("Section")
|
||
|
|
||
|
def __repr__(self):
|
||
|
return f"<{self.id}: {self.label}>"
|