diff options
-rw-r--r-- | pyenc/model.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/pyenc/model.py b/pyenc/model.py index be491ee..10b2336 100644 --- a/pyenc/model.py +++ b/pyenc/model.py @@ -32,11 +32,21 @@ class Host(db.Model): return { column.name: self.__getattribute__(column.name) for column in self.__table__.columns } +class PuppetFile(db.Model): + __tablename__ = 'puppet_file' + id = db.Column(db.Integer, primary_key=True) + path = db.Column(db.Text, nullable=False) + json = db.Column(db.Text, nullable=False) + last_parse = db.Column(db.Float) + # classes = db.relationship('PuppetClass', back_populates='comes_from') + classes = db.relationship('PuppetClass', backref='comes_from') + class PuppetClass(db.Model): __tablename__ = 'puppet_class' id = db.Column(db.Integer, primary_key=True) class_name = db.Column(db.Text, nullable=False) - source_file = db.Column(db.Text) + # comes_from = db.relationship('PuppetFile', back_populates='classes') + comes_from_id = db.Column(db.Integer, db.ForeignKey(f'{PuppetFile.__tablename__}.id')) hosts = db.relationship('Host', back_populates='classes', secondary=host_classes) |