Add revision and correct fields to snapshot class; should simplify most queries

git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@5133 ce84e21b-d406-0410-9b95-82705330c041

1 files changed, 5 insertions, 2 deletions

diff --git a/src/lib/Server/Snapshots/model.py b/src/lib/Server/Snapshots/model.py
index a913a25de..8209eeab7 100644
--- a/src/lib/Server/Snapshots/model.py
+++ b/src/lib/Server/Snapshots/model.py
@@ -200,6 +200,8 @@ action_snap = Table('action_snap', Base.metadata,
 class Snapshot(Base):
     __tablename__ = 'snapshot'
     id = Column(Integer, primary_key=True)
+    correct = Column(Boolean)
+    revision = Column(Unicode(36))
     metadata_id = Column(Integer, ForeignKey('metadata.id'))
     client_metadata = relation(Metadata, primaryjoin=metadata_id==Metadata.id)
     timestamp = Column(DateTime, default=datetime.datetime.now)
@@ -221,9 +223,10 @@ class Snapshot(Base):
                        ('Path', ('extra_files', File))])
 
     @classmethod
-    def from_data(cls, session, metadata, entries, extra):
+    def from_data(cls, session, correct, revision, metadata, entries, extra):
         dbm = Metadata.from_metadata(session, metadata)
-        snap = cls(client_metadata=dbm, timestamp=datetime.datetime.now())
+        snap = cls(correct=correct, client_metadata=dbm, revision=revision,
+                   timestamp=datetime.datetime.now())
         for (dispatch, data) in [(cls.c_dispatch, entries),
                                  (cls.e_dispatch, extra)]:
             for key in dispatch: