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)