add fields to DC model object

ispyb/model/datacollection.py

@@ -29,7 +29,7 @@ def __init__(self, dcid, db_area, preload=None):
 
   def reload(self):
     '''Load/update information from the database.'''
-    self._data = self._db.retrieve_data_collection_main(self._dcid)[0]
+    self._data = self._db.retrieve_data_collection(self._dcid)[0]
 
   @property
   def dcid(self):
@@ -107,6 +107,11 @@ def __str__(self):
     ('image_count', 'noImages', None),
     ('image_start_number', 'startImgNumber', None),
     ('status', 'status', 'Returns a string representing the current data collection status.'),
+    ('comment', 'comments', 'A free comment field for the data collection.'),
+    ('snapshot1', 'snapshot1', 'One of four possible fields to store file paths to image files relating to the data collection'),
+    ('snapshot2', 'snapshot2', 'One of four possible fields to store file paths to image files relating to the data collection'),
+    ('snapshot3', 'snapshot3', 'One of four possible fields to store file paths to image files relating to the data collection'),
+    ('snapshot4', 'snapshot4', 'One of four possible fields to store file paths to image files relating to the data collection'),
 ))