+from macros import sequences
+
+# add $HOME in PYTHONPATH so we can import LocalSubstrate.py
+sys.path.append(os.environ['HOME'])
+import LocalSubstrate
+
+class Step:
+
+ natives=TestPlc.__dict__
+
+ def display (self): return self.name.replace('_','-')
+ def internal (self): return self.name.replace('-','_')
+
+ def __init__ (self, name):
+ self.name=name
+ # a native step is implemented as a method on TestPlc
+ self.native = self.internal() in Step.natives
+ if self.native:
+ self.method=Step.natives[self.internal()]
+ else:
+ try:
+ self.substeps=sequences[self.internal()]
+ except Exception,e:
+ print "macro step %s not found in macros.py (%s) - exiting"%(self.display(),e)
+ raise
+
+ def print_doc (self,level=0):
+ tab=32
+ trail=8
+ if self.native:
+ start=level*' '+'* '
+ # 2 is the len of '* '
+ width=tab-level-2
+ format="%%-%ds"%width
+ line=start+format%self.display()
+ print line,
+ try:
+ print self.method.__doc__
+ except:
+ print "*** no doc found"
+ else:
+ beg_start=level*' '+'>>> '
+ end_start=level*' '+'<<< '
+ trailer=trail*'-'
+ # 4 is the len of '>>> '
+ width=tab-level-4-trail
+ format=("%%-%ds"%width)
+ beg_line=beg_start+format%self.display()+trail*'>'
+ end_line=end_start+format%self.display()+trail*'<'
+ print beg_line
+ for step in self.substeps:
+ Step(step).print_doc(level+1)
+ print end_line
+
+ # return a list of (name, method) for all native steps involved
+ def tuples (self):
+ if self.native: return [ (self.internal(), self.method,) ]
+ else:
+ result=[]
+ for substep in [ Step(name) for name in self.substeps ] :
+ result += substep.tuples()
+ return result
+
+ # convenience for listing macros
+ # just do a listdir, hoping we're in the right directory...
+ @staticmethod
+ def list_macros ():
+ names= sequences.keys()
+ names.sort()
+ return names