Changeset 33098 for branches/meh_branches/ppsub_test/ippToPsps/jython/datastore.py

Timestamp:

Jan 11, 2012, 11:19:21 PM (15 years ago)

Author:

mhuber

Message:

merged latest r33093 trunk changes to meh_branches/ppsub_test

Location:

branches/meh_branches/ppsub_test

Files:

• . (modified) (1 prop)
• ippToPsps/jython/datastore.py (modified) (6 diffs)

branches/meh_branches/ppsub_test/ippToPsps/jython/datastore.py

@@ -5 +5 @@
 import logging
 import sys
+import re
 from xml.etree.ElementTree import ElementTree, Element, tostring
+
+from ipptopspsdb import IppToPspsDb
 
 '''
@@ -16 +19 @@
 
     '''
-    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 +38 @@
     def publish(self, name, path, file, fileType):
 
+        # TODO should handle this better
+        batchID = int(name[1:])
+
         self.logger.infoPair("Attempting to publish",  path + "/" + file)
         tempFile = tempfile.NamedTemporaryFile(mode='w+b')
@@ -39 +46 @@
         tempFile.seek(0)
 
-        # build dsreg command command
+        # build dsreg command
         command  = "dsreg --add " + name + "\
-                    --copy --datapath " + path + "\
+                    --link --datapath " + path + "\
                     --type " + self.type + "\
                     --product " + self.product + "\
@@ -52 +59 @@
             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()
 
-        tempFile.close()
         return ret
-
 
     '''
     Removes an item from the datastore
+
+    NB this will return a success if the batch is not found on the datastore
     '''
     def remove(self, name):
+
+        # TODO should handle this better
+        batchID = int(name[1:])
 
         command = "dsreg \
@@ -71 +84 @@
                    --product " + self.product
 
-        p = Popen(command, shell=True, stdout=PIPE)
+        p = Popen(command, shell=True, stdout=PIPE, stderr=PIPE)
         p.wait()
+        output, errors = p.communicate()
 
         if p.returncode != 0:
-            self.logger.errorPair("Datastore removal failed",  name)
-            ret = False
+            if re.search("not found in", errors):
+                self.logger.infoPair("Datastore batch not found", name)
+                ret = True
+                self.ippToPspsDb.updateDeletedDatastore(batchID, 1)
+            else:
+                self.logger.infoPair("Datastore removal FAILED", name)
+                ret = False
         else:
+            self.logger.infoPair("Datastore removal successful", name)
             ret = True
-            self.logger.infoPair("Datastore removal successful", name)
+            self.ippToPspsDb.updateDeletedDatastore(batchID, 1)
             
         return ret