import py defaultconftestpath = py.magic.autopath().dirpath('defaultconftest.py') class Conftest(object): """ the single place for accessing values and interacting towards conftest modules from py.test objects. Note that triggering Conftest instances to import conftest.py files may result in added cmdline options. XXX """ def __init__(self, path=None): self._path2confmods = {} if path is not None: self.setinitial([path]) def setinitial(self, args): """ return a Conftest object initialized with a path obtained from looking at the first (usually cmdline) argument that points to an existing file object. XXX note: conftest files may add command line options and we thus have no completely safe way of determining which parts of the arguments are actually related to options. """ current = py.path.local() for arg in args + [current]: anchor = current.join(arg, abs=1) if anchor.check(): # we found some file object print "initializing conftest from", anchor # conftest-lookups without a path actually mean # lookups with our initial path. self._path2confmods[None] = self.getconftestmodules(anchor) #print " -> ", conftest._path2confmods break def getconftestmodules(self, path): """ return a list of imported conftest modules for the given path. """ try: clist = self._path2confmods[path] except KeyError: dp = path.dirpath() if dp == path: return [importconfig(defaultconftestpath)] clist = self.getconftestmodules(dp) conftestpath = path.join("conftest.py") if conftestpath.check(file=1): clist.append(importconfig(conftestpath)) self._path2confmods[path] = clist # be defensive: avoid changes from caller side to # affect us by always returning a copy of the actual list return clist[:] # XXX no real use case, may probably go #def lget(self, name, path=None): # modules = self.getconftestmodules(path) # return self._get(name, modules) def rget(self, name, path=None): modules = self.getconftestmodules(path) modules.reverse() return self._get(name, modules) def _get(self, name, modules): for mod in modules: try: return getattr(mod, name) except AttributeError: continue raise KeyError, name def importconfig(configpath): if not configpath.dirpath('__init__.py').check(file=1): # HACK: we don't want any "globally" imported conftest.py, # prone to conflicts and subtle problems modname = str(configpath).replace('.', configpath.sep) return configpath.pyimport(modname=modname) else: return configpath.pyimport()