X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=t%2Fmoe%2Fpipeline.py;h=f14a0f5f7956be474151c6148474b94bc86f525f;hb=b07a7c4bf08db2928e1845c4e60798f7c9ad8b65;hp=30c4c587e4417bb71c0d0114b1a1a7c65951729f;hpb=8da1e0880f38cdd74fa1b73339860a46f3fa7227;p=eval.git

diff --git a/t/moe/pipeline.py b/t/moe/pipeline.py
index 30c4c58..f14a0f5 100644
--- a/t/moe/pipeline.py
+++ b/t/moe/pipeline.py
@@ -3,11 +3,17 @@
 import sys
 import bisect
 import imp
+import moe
 import moe.log
 
-class MoePipeError(Exception):
+class MoePipeError(moe.MoeError):
     """Failure of the MoePipeline."""
 
+class MoeAbortPipeline(Exception):
+
+    def __init__(self, skip_to=999):
+	self.skip_to = skip_to
+
 class MoePipeline:
     """Moe pipeline."""
 
@@ -23,23 +29,31 @@ class MoePipeline:
 	    raise MoePipeError, "Pipeline insert cannot alter the past"
 	self.pipe.insert(pos, triple)
 
-    def dump(self, file=sys.stdout):
+    def dump(self, file=sys.stdout, prefix=""):
 	file.write(">>> Pipeline %s\n" % self.name)
 	for pri,name,fun in self.pipe:
-	    file.write("%03d %s\n" % (pri,name))
+	    file.write("%s%03d %s\n" % (prefix,pri,name))
 
     def run(self, *args):
 	self.index = 0
+	min_pri = -1
 	while self.index < len(self.pipe):
 	    (pri,name,fun) = self.pipe[self.index]
-	    moe.log.verbose(">> Running %s:%s\n" % (self.name,name))
-	    fun(*args)
+	    if pri >= min_pri:
+		moe.log.default.verbose(">> Running %s:%s\n" % (self.name,name))
+		try:
+		    fun(*args)
+		except MoeAbortPipeline, err:
+		    min_pri = err.skip_to
+	    else:
+		moe.log.default.verbose(">> Skipping %s:%s\n" % (self.name,name))
 	    self.index += 1
 	self.index = -1
+	moe.log.default.verbose(">> Pipeline %s finished\n" % self.name)
 
     def add_hook(self, name):
 	modname = "moe.hooks." + name
-	moe.log.verbose(">> Loading hook %s\n" % name)
+	moe.log.default.verbose(">> Loading hook %s\n" % name)
 	if not sys.modules.has_key(modname):
 	    ## FIXME: Configuration variable for the hook directory?
 	    try: