Laurent@1330: import os, re, traceback
Laurent@1315: 
Laurent@1315: from copy import deepcopy
Laurent@1315: from lxml import etree
Laurent@1315: from xmlclass import GenerateParserFromXSDstring
laurent@401: 
Laurent@1096: from PLCControler import UndoBuffer
Laurent@1330: from ConfigTreeNode import XSDSchemaErrorMessage
laurent@630: 
Laurent@1124: CODEFILE_XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>
Laurent@1315: <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema"
Laurent@1315:             xmlns:xhtml="http://www.w3.org/1999/xhtml">
Laurent@1124:   <xsd:element name="%(codefile_name)s">
Laurent@1124:     <xsd:complexType>
Laurent@1124:       <xsd:sequence>
Laurent@1124:         %(includes_section)s
Laurent@1124:         <xsd:element name="variables">
Laurent@1124:           <xsd:complexType>
Laurent@1124:             <xsd:sequence>
Laurent@1124:               <xsd:element name="variable" minOccurs="0" maxOccurs="unbounded">
Laurent@1124:                 <xsd:complexType>
Laurent@1124:                   <xsd:attribute name="name" type="xsd:string" use="required"/>
Laurent@1124:                   <xsd:attribute name="type" type="xsd:string" use="required"/>
Laurent@1124:                   <xsd:attribute name="class" use="optional">
Laurent@1124:                     <xsd:simpleType>
Laurent@1124:                       <xsd:restriction base="xsd:string">
Laurent@1124:                         <xsd:enumeration value="input"/>
Laurent@1124:                         <xsd:enumeration value="memory"/>
Laurent@1124:                         <xsd:enumeration value="output"/>
Laurent@1124:                       </xsd:restriction>
Laurent@1124:                     </xsd:simpleType>
Laurent@1124:                   </xsd:attribute>
Laurent@1124:                   <xsd:attribute name="initial" type="xsd:string" use="optional" default=""/>
Laurent@1124:                 </xsd:complexType>
Laurent@1124:               </xsd:element>
Laurent@1124:             </xsd:sequence>
Laurent@1124:           </xsd:complexType>
Laurent@1124:         </xsd:element>
Laurent@1124:         %(sections)s
Laurent@1124:       </xsd:sequence>
Laurent@1124:     </xsd:complexType>
Laurent@1124:   </xsd:element>
Laurent@1124:   <xsd:complexType name="CodeText">
Laurent@1124:     <xsd:annotation>
Laurent@1124:       <xsd:documentation>Formatted text according to parts of XHTML 1.1</xsd:documentation>
Laurent@1124:     </xsd:annotation>
Laurent@1124:     <xsd:sequence>
Laurent@1124:       <xsd:any namespace="http://www.w3.org/1999/xhtml" processContents="lax"/>
Laurent@1124:     </xsd:sequence>
Laurent@1124:   </xsd:complexType>
Laurent@1124: </xsd:schema>"""
Laurent@1124: 
Laurent@1124: SECTION_TAG_ELEMENT = "<xsd:element name=\"%s\" type=\"CodeText\"/>"
lbessard@145: 
Laurent@1096: class CodeFile:
laurent@656:     
Laurent@1124:     CODEFILE_NAME = "CodeFile"
Laurent@1124:     SECTIONS_NAMES = []
Laurent@1124:     
etisserant@31:     def __init__(self):
Laurent@1124:         sections_str = {"codefile_name": self.CODEFILE_NAME}
Laurent@1124:         if "includes" in self.SECTIONS_NAMES:
Laurent@1124:             sections_str["includes_section"] = SECTION_TAG_ELEMENT % "includes"
Laurent@1124:         else:
Laurent@1124:             sections_str["includes_section"] = ""
Laurent@1124:         sections_str["sections"] = "\n".join(
Laurent@1124:             [SECTION_TAG_ELEMENT % name
Laurent@1124:              for name in self.SECTIONS_NAMES if name != "includes"])
Laurent@1124:         
Laurent@1315:         self.CodeFileParser = GenerateParserFromXSDstring(
Laurent@1124:             CODEFILE_XSD % sections_str)
Laurent@1315:         self.CodeFileVariables = etree.XPath("variables/variable")
Laurent@1124:         
Laurent@1096:         filepath = self.CodeFileName()
lbessard@145:         
lbessard@145:         if os.path.isfile(filepath):
lbessard@145:             xmlfile = open(filepath, 'r')
Laurent@1315:             codefile_xml = xmlfile.read()
lbessard@145:             xmlfile.close()
lbessard@145:             
Laurent@1315:             codefile_xml = codefile_xml.replace(
Laurent@1315:                 '<%s>' % self.CODEFILE_NAME, 
Laurent@1315:                 '<%s xmlns:xhtml="http://www.w3.org/1999/xhtml">' % self.CODEFILE_NAME)
Laurent@1315:             for cre, repl in [
Laurent@1315:                 (re.compile("(?<!<xhtml:p>)(?:<!\[CDATA\[)"), "<xhtml:p><![CDATA["),
Laurent@1315:                 (re.compile("(?:]]>)(?!</xhtml:p>)"), "]]></xhtml:p>")]:
Laurent@1315:                 codefile_xml = cre.sub(repl, codefile_xml)
Laurent@1330:             
Laurent@1330:             try:
Laurent@1330:                 self.CodeFile, error = self.CodeFileParser.LoadXMLString(codefile_xml)
Laurent@1330:                 if error is not None:
Laurent@1330:                     self.GetCTRoot().logger.write_warning(
Laurent@1332:                         XSDSchemaErrorMessage % ((self.CODEFILE_NAME,) + error))
Laurent@1330:                 self.CreateCodeFileBuffer(True)
Laurent@1330:             except Exception, exc:
Laurent@1330:                 self.GetCTRoot().logger.write_error(_("Couldn't load confnode parameters %s :\n %s") % (CTNName, unicode(exc)))
Laurent@1330:                 self.GetCTRoot().logger.write_error(traceback.format_exc())
lbessard@145:         else:
Laurent@1315:             self.CodeFile = self.CodeFileParser.CreateRoot()
Laurent@1096:             self.CreateCodeFileBuffer(False)
Edouard@718:             self.OnCTNSave()
lbessard@145: 
lbessard@145:     def GetBaseTypes(self):
Edouard@718:         return self.GetCTRoot().GetBaseTypes()
lbessard@145: 
Laurent@1096:     def GetDataTypes(self, basetypes = False):
Laurent@1096:         return self.GetCTRoot().GetDataTypes(basetypes=basetypes)
lbessard@145: 
Laurent@1146:     def GenerateNewName(self, format, start_idx):
Laurent@1146:         return self.GetCTRoot().GenerateNewName(
Laurent@1146:             None, None, format, start_idx,
Laurent@1153:             dict([(var.getname().upper(), True) 
Laurent@1146:                   for var in self.CodeFile.variables.getvariable()]))
Laurent@1146: 
lbessard@145:     def SetVariables(self, variables):
Laurent@1096:         self.CodeFile.variables.setvariable([])
lbessard@145:         for var in variables:
Laurent@1315:             variable = self.CodeFileParser.CreateElement("variable", "variables")
lbessard@145:             variable.setname(var["Name"])
lbessard@145:             variable.settype(var["Type"])
Laurent@1096:             variable.setinitial(var["Initial"])
Laurent@1096:             self.CodeFile.variables.appendvariable(variable)
lbessard@145:     
lbessard@145:     def GetVariables(self):
lbessard@145:         datas = []
Laurent@1315:         for var in self.CodeFileVariables(self.CodeFile):
Laurent@1124:             datas.append({"Name" : var.getname(), 
Laurent@1124:                           "Type" : var.gettype(), 
Laurent@1124:                           "Initial" : var.getinitial()})
lbessard@145:         return datas
lbessard@145: 
Laurent@1096:     def SetTextParts(self, parts):
Laurent@1124:         for section in self.SECTIONS_NAMES:
Laurent@1124:             section_code = parts.get(section)
Laurent@1124:             if section_code is not None:
Laurent@1315:                 getattr(self.CodeFile, section).setanyText(section_code)
Laurent@1124:     
Laurent@1096:     def GetTextParts(self):
Laurent@1315:         return dict([(section, getattr(self.CodeFile, section).getanyText())
Laurent@1124:                      for section in self.SECTIONS_NAMES])
Laurent@1124:             
Edouard@718:     def CTNTestModified(self):
Laurent@1096:         return self.ChangesToSave or not self.CodeFileIsSaved()    
laurent@630: 
Laurent@1061:     def OnCTNSave(self, from_project_path=None):
Laurent@1096:         filepath = self.CodeFileName()
lbessard@145:         
lbessard@145:         xmlfile = open(filepath,"w")
Laurent@1315:         xmlfile.write(etree.tostring(
Laurent@1315:             self.CodeFile, 
Laurent@1315:             pretty_print=True, 
Laurent@1315:             xml_declaration=True, 
Laurent@1315:             encoding='utf-8'))
lbessard@145:         xmlfile.close()
lbessard@145:         
Laurent@1096:         self.MarkCodeFileAsSaved()
etisserant@31:         return True
etisserant@31: 
Laurent@1095:     def CTNGlobalInstances(self):
Laurent@1095:         current_location = self.GetCurrentLocation()
Laurent@1096:         return [(variable.getname(),
Laurent@1096:                  variable.gettype(),
Laurent@1096:                  variable.getinitial())
Laurent@1315:                 for variable in self.CodeFileVariables(self.CodeFile)]
laurent@656: 
lbessard@145: #-------------------------------------------------------------------------------
lbessard@145: #                      Current Buffering Management Functions
lbessard@145: #-------------------------------------------------------------------------------
lbessard@145: 
lbessard@145:     """
Laurent@1096:     Return a copy of the codefile model
lbessard@145:     """
lbessard@145:     def Copy(self, model):
Laurent@1315:         return deepcopy(model)
lbessard@145: 
Laurent@1096:     def CreateCodeFileBuffer(self, saved):
laurent@658:         self.Buffering = False
Laurent@1315:         self.CodeFileBuffer = UndoBuffer(self.CodeFileParser.Dumps(self.CodeFile), saved)
laurent@651: 
Laurent@1096:     def BufferCodeFile(self):
Laurent@1315:         self.CodeFileBuffer.Buffering(self.CodeFileParser.Dumps(self.CodeFile))
lbessard@145:     
lbessard@145:     def StartBuffering(self):
lbessard@145:         self.Buffering = True
lbessard@145:         
lbessard@145:     def EndBuffering(self):
lbessard@145:         if self.Buffering:
Laurent@1315:             self.CodeFileBuffer.Buffering(self.CodeFileParser.Dumps(self.CodeFile))
lbessard@145:             self.Buffering = False
lbessard@145:     
Laurent@1096:     def MarkCodeFileAsSaved(self):
laurent@651:         self.EndBuffering()
Laurent@1096:         self.CodeFileBuffer.CurrentSaved()
laurent@651:     
Laurent@1096:     def CodeFileIsSaved(self):
Laurent@1096:         return self.CodeFileBuffer.IsCurrentSaved() and not self.Buffering
laurent@658:         
lbessard@145:     def LoadPrevious(self):
laurent@651:         self.EndBuffering()
Laurent@1315:         self.CodeFile = self.CodeFileParser.Loads(self.CodeFileBuffer.Previous())
lbessard@145:     
lbessard@145:     def LoadNext(self):
Laurent@1315:         self.CodeFile = self.CodeFileParser.Loads(self.CodeFileBuffer.Next())
lbessard@145:     
lbessard@145:     def GetBufferState(self):
Laurent@1096:         first = self.CodeFileBuffer.IsFirst() and not self.Buffering
Laurent@1096:         last = self.CodeFileBuffer.IsLast()
lbessard@145:         return not first, not last
lbessard@145: