Index: trunk/ippToPsps/jython/datastore.py
===================================================================
--- trunk/ippToPsps/jython/datastore.py	(revision 32331)
+++ trunk/ippToPsps/jython/datastore.py	(revision 32466)
@@ -7,4 +7,6 @@
 import re
 from xml.etree.ElementTree import ElementTree, Element, tostring
+
+from ipptopspsdb import IppToPspsDb
 
 '''
@@ -17,9 +19,10 @@
 
     '''
-    def __init__(self, logger, doc):
+    def __init__(self, logger, doc, ippToPspsDb):
     
         # setup logger
         self.logger = logger
         self.doc = doc
+        self.ippToPspsDb = ippToPspsDb
         self.logger.debug("Datastore constructor")
 
@@ -34,4 +37,7 @@
     '''
     def publish(self, name, path, file, fileType):
+
+        # TODO should handle this better
+        batchID = int(name[1:])
 
         self.logger.infoPair("Attempting to publish",  path + "/" + file)
@@ -53,7 +59,9 @@
             self.logger.errorPair("Datastore publish",  "failed")
             ret = False
+            self.ippToPspsDb.updateLoadedToDatastore(batchID, -1)
         else:
             self.logger.infoPair("Datastore publish", "successful")
             ret = True
+            self.ippToPspsDb.updateLoadedToDatastore(batchID, 1)
             
         tempFile.close()
@@ -66,4 +74,7 @@
     '''
     def remove(self, name):
+
+        # TODO should handle this better
+        batchID = int(name[1:])
 
         command = "dsreg \
@@ -80,4 +91,5 @@
                 self.logger.infoPair("Datastore batch not found", name)
                 ret = True
+                self.ippToPspsDb.updateDeletedDatastore(batchID, 1)
             else:
                 self.logger.infoPair("Datastore removal FAILED", name)
@@ -86,4 +98,5 @@
             self.logger.infoPair("Datastore removal successful", name)
             ret = True
+            self.ippToPspsDb.updateDeletedDatastore(batchID, 1)
             
         return ret