Viewing file:  ForgeBlackBox.py (3.47 KB) -rwxr-xr-x
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
#!/usr/bin/python
import Alchemist
import os
import re
import stat
from xml.xslt.Processor import Processor
def getBox(box_cfg):
"""
getBox: get a BlackBox from this module
@box_cfg: a box_cfg conformant AdmList element containing
the following elements:
path: type string, the path to the context file (required)
returns: Forge object
"""
if box_cfg == None:
raise ValueError
return ForgeBlackBox(box_cfg)
class ForgeBlackBox(Alchemist.BlackBox):
"""o
The ForgeBlackBox is the class which handles converting the xml with a XSLT
stylesheet and writing this as a configuration file
"""
def __init__(self,box_cfg):
"""
__init__: Initialize a ForgeBlackBox
@self: the class instance
@box_cfg: a box_cfg conformant AdmList element containing
the following elements:
xslstylesheetpath: a string
configfilepath: a string
mode: an integer
"""
self.status=0
self.me=self.__class__.__module__
self._errNo=0
self._errStr=None
self.readable=0
self.writable=1
Alchemist.validateBoxCfg(box_cfg)
self.sspath=box_cfg.getChildByName("xslstylesheetpath").getValue()
try:
if os.access(self.sspath, os.R_OK):
mode = os.stat(self.sspath)[stat.ST_MODE]
if (not stat.S_ISREG(mode)):
raise ValueError
except:
raise ValueError, "ForgeBlackBox box_cfg must contain a 'xslstylesheetpath' entry pointing at a readable file (" + self.sspath + ")"
try:
self.cfgpath=box_cfg.getChildByName("configfilepath").getValue()
if self.cfgpath=="SPLITFILES":
self.splitfiles=1
else:
self.splitfiles=0
except:
raise ValueError, "ForgeBlackBox box_cfg must contain a 'configfilepath' entry"
try:
self.mode=box_cfg.getChildByName("mode").getValue()
except:
raise ValueError, "ForgeBlackBox box_cfg must contain a 'mode' entry"
def write (self,context):
"""
Extract xml, stylesheetpath and target file from the context
and create the configuration file.
@self: the class instance
@context: the context from which to extract information
"""
xml_string = context.toXML()
if (not xml_string):
return 0
processor=Processor()
processor.appendStylesheetFile(self.sspath)
result=processor.runString(xml_string)
if (self.splitfiles==0):
fd = os.open(self.cfgpath, os.O_WRONLY | os.O_CREAT | os.O_TRUNC, self.mode)
file = os.fdopen(fd, "w")
file.write(result)
file.close()
else:
myreg=re.compile(r'^--------CUTFILEHERE: (?P<filename>.*?)$(?P<content>.*?)^--------ENDFILEHERE?(?P<remaining>.*)',re.MULTILINE|re.DOTALL)
match=1
while match:
match=myreg.search(result)
if not match:
break
else:
result=match.group("remaining")
fd = os.open(match.group("filename"), os.O_WRONLY | os.O_CREAT | os.O_TRUNC, self.mode)
file = os.fdopen(fd, "w")
file.write(match.group("content"))
file.close()
|