+class Step:
+
+ natives=TestPlc.__dict__
+
+ def __init__ (self, name):
+ self.name=name.replace('-','_')
+ # a native step is implemented as a method on TestPlc
+ self.native = name in Step.natives
+ if self.native:
+ self.method=Step.natives[self.name]
+ else:
+ # non-native steps (macros) are implemented as a 'Step'
+ try:
+ modulename = 'macro_' + self.name
+ module = __import__ (modulename)
+ self.substeps = module.sequence
+ except Exception,e:
+ print "Cannot load macro step %s (%s) - exiting"%(self.name,e)
+ raise
+
+ def norm_name (self): return self.name.replace('_','-')
+
+ def print_doc (self):
+ if self.native:
+ print '*',self.norm_name(),"\r",4*"\t",
+ try:
+ print self.method.__doc__
+ except:
+ print "*** no doc found"
+ else:
+ print '*',self.norm_name(),"\r",3*"\t","========== BEG MACRO step"
+ for step in self.substeps:
+ Step(step).print_doc()
+ print '*',self.norm_name(),"\r",3*"\t","========== END MACRO step"
+
+ # return a list of (name, method) for all native steps involved
+ def tuples (self):
+ if self.native: return [ (self.name, 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= [ filename.replace('macro_','').replace('.py','') for filename in glob.glob ('macro_*.py')]
+ names.sort()
+ return names
+