'''
Created on September , 2012
@author: 
'''
from xml.etree.ElementTree import Element, SubElement, ElementTree
from xml.etree import ElementTree as ET
from xml.dom import minidom

import sys

def prettify(elem):
    """Return a pretty-printed XML string for the Element.
    """
    rough_string = ET.tostring(elem, 'utf-8')
    reparsed = minidom.parseString(rough_string)
    return reparsed.toprettyxml(indent="  ")

class Controller():
    
    id = None
    name = None
    description = None
    readBranchObjList = None
    procBranchObjList = None
    
    def __init__(self):
        
#        self.id = id
#        self.name = name
#        self.description = description
        
        self.readBranchObjList = []
        self.procBranchObjList = []
    
    def setParms(self, id, name, description):
        
        self.id = id
        self.name = name
        self.description = description
    
    def addReadBranch(self, dpath, dataformat, readMode, startDate='', endDate='', startTime='', endTime=''):
        
        id = self.getNewId()
        
        readBranchObj = ReadBranch()
        readBranchObj.setup(id, dpath, dataformat, readMode, startDate, endDate, startTime, endTime)
        
        self.readBranchObjList.append(readBranchObj)
        
        return readBranchObj
    
    def addProcBranch(self, name):
        
        id = self.getNewId()
        
        procBranchObj = ProcBranch()
        procBranchObj.setup(id, name)
        
        self.procBranchObjList.append(procBranchObj)
        
        return procBranchObj
    
    def getNewId(self):
        
        id = int(self.id)*10 + len(self.readBranchObjList) + len(self.procBranchObjList) + 1
        return id
    
    def makeXml(self):    
        
        projectElement = Element('Controller')
        projectElement.set('id', str(self.id))
        projectElement.set('name', self.name)
        projectElement.set('description', self.description)
        
        for readBranchObj in self.readBranchObjList:
            readBranchObj.makeXml(projectElement)
            
        for procBranchObj in self.procBranchObjList:
            procBranchObj.makeXml(projectElement)
            
        self.projectElement = projectElement
    
    def writeXml(self, filename):
        
        self.makeXml()
        ElementTree(self.projectElement).write(filename, method='xml')
        print prettify(self.projectElement)

    def readXml(self, filename):
        
        #tree = ET.parse(filename)
        self.projectElement = None
        self.readBranchObjList = None
        self.procBranchObjList = None
        
        self.projectElement = ElementTree().parse(filename)
        
        self.project = self.projectElement.tag
        
        self.id = self.projectElement.get('id')
        self.name = self.projectElement.get('name')      
        
        self.readBranchObjList = []
        
        readBranchElementList = self.projectElement.getiterator('readBranch')
        
        for readBranchElement in readBranchElementList:
            readBrachObj = ReadBranch()
            readBrachObj.readXml(readBranchElement)
            self.readBranchObjList.append(readBranchObj)
        
        self.procBranchObjList = []
        
        procBranchElementList = self.projectElement.getiterator('procBranch')
        
        for procBranchElement in procBranchElementList:
            procBranchObj = ProcBranch()
            procBranchObj.readXml(procBranchElement)
            self.procBranchObjList.append(procBranchObj)
               
    def printattr(self):
        
        print "Controller[%s]: name = %s, description = %s" %(self.id,
                                                                    self.name,
                                                                    self.description)
        
        for readBranchObj in self.readBranchObjList:
            readBranchObj.printattr()
        
        for procBranchObj in self.procBranchObjList:
            procBranchObj.printattr()
            
class ReadBranch():
    
    id = None
    name = None
#    dpath = None
#    dataformat = None
#    readMode = None
#    startDate = None
#    endDate = None
#    startTime = None
#    endTime = None
    
    parmObjList = []
    
    def __init__(self):
        
        self.parmObjList = []
        
    def setup(self, id, dpath, dataformat, readMode, startDate, endDate, startTime, endTime):
        
        self.id = id
        self.dpath = dpath
        self.dataformat = dataformat
        self.readMode = readMode
        self.startDate = startDate
        self.endDate = endDate
        self.startTime = startTime
        self.endTime = endTime
    
    def addParameter(self, name, value):
        
        id = int(self.id)*10 + len(self.parmObjList) + 1
        
        parmObj = ParameterConf()
        parmObj.setup(id, name, value)
        
        self.parmObjList.append(parmObj)
        
        return parmObj
    
    def makeXml(self, projectElement):
        
        readBranchElement = SubElement(projectElement, 'readBranch')
        readBranchElement.set('id', str(self.id))
            
        self.addParameter(name='dpath'     , value=self.dpath)
        self.addParameter(name='dataformat', value=self.dataformat)
        self.addParameter(name='startDate' , value=self.startDate)
        self.addParameter(name='endDate'   , value=self.endDate)
        self.addParameter(name='startTime' , value=self.startTime)
        self.addParameter(name='endTime'   , value=self.endTime)
        self.addParameter(name='readMode'  , value=str(self.readMode))
        
        for parmObj in self.parmObjList:
            parmObj.makeXml(readBranchElement)
            
    def readXml(self, readBranchElement):
        
        self.id = readBranchElement.get('id')
        
        self.parmObjList = []
        
        parmElementList = readBranchElement.getiterator('Parameter')
        
        for parmElement in parmElementList:
            parmObj = ParameterConf()
            parmObj.readXml(parmElement)
            self.parmObjList.append(parmObj)
    
    def printattr(self):
        
        print "ReadBranch[%s]: name = %s" %(self.id,
                                            self.name)
        
        for parmObj in self.parmObjList:
            parmObj.printattr()
    
class ProcBranch():
    
    id = None
    name = None
    
    upObjList = None
    
    def __init__(self):
        pass
    
    def setup(self, id, name):
        
        self.id = id
        self.name = name
        
        self.upObjList = []
        
    def addUP(self, name, type, inputId):
        
        id = int(self.id)*10 + len(self.upObjList) + 1
        
        upObj = UPConf()
        upObj.setup(id, name, type, inputId)
        
        self.upObjList.append(upObj)
        
        return upObj
          
    def makeXml(self, projectElement):
        
        procBranchElement = SubElement(projectElement, 'procBranch')
        procBranchElement.set('id', str(self.id))
        procBranchElement.set('name', self.name)
        
        for upObj in self.upObjList:
            upObj.makeXml(procBranchElement)
    
    def readXml(self, procBranchElement):
        
        self.id = procBranchElement.get('id')
        self.name = procBranchElement.get('name')
        
        self.upObjList = []
        
        upElementList = procBranchElement.getiterator('UP')
        
        for upElement in upElementList:
            upObj = UPConf()
            upObj.readXml(upElement)
            self.upObjList.append(upObj)
    
    def printattr(self):
        
        print "ProcBranch[%s]: name = %s" %(self.id,
                                                self.name)
        
        for upObj in self.upObjList:
            upObj.printattr()
        
class UPConf():
    
    id = None
    name = None
    type = None
    inputId = None
    
    opObjList = []
    
    def __init__(self):
        pass
    
    def setup(self, id, name, type, inputId):
        
        self.id = id
        self.name = name
        self.type = type
        self.inputId = inputId
        
        self.opObjList = []
        
    def addOperation(self, name, priority, type='self'):
        
        id = int(self.id)*10 + len(self.opObjList) + 1
        
        opObj = OperationConf()
        opObj.setup(id, name, priority, type)
        
        self.opObjList.append(opObj)
        
        return opObj
    
    def makeXml(self, procBranchElement):
        
        upElement = SubElement(procBranchElement, 'UP')
        upElement.set('id', str(self.id))
        upElement.set('name', self.name)
        upElement.set('type', self.type)
        upElement.set('inputId', str(self.inputId))
        
        for opObj in self.opObjList:
            opObj.makeXml(upElement)
    
    def readXml(self, upElement):
        
        self.id = upElement.get('id')
        self.name = upElement.get('name')
        self.type = upElement.get('type')
        self.inputId = upElement.get('inputId')
        
        self.opObjList = []
        
        opElementList = upElement.getiterator('Operation')
        
        for opElement in opElementList:
            opObj = OperationConf()
            opObj.readXml(opElement)
            self.opObjList.append(opObj)
        
    def getOperationObjList(self):
        
        return self.opObjList
    
    def printattr(self):
        
        print "UP[%s]: name = %s, type = %s, inputId = %s" %(self.id,
                                                             self.name,
                                                             self.type,
                                                             self.inputId)
        
        for opObj in self.opObjList:
            opObj.printattr()
    
class OperationConf():
    
    id = 0
    name = None
    priority = None
    type = 'self'
    
    parmObjList = []
    
    def __init__(self):
        pass
    
    def setup(self, id, name, priority, type):
        
        self.id = id
        self.name = name
        self.priority = priority
        self.type = type
        
        self.parmObjList = []
        
    def addParameter(self, name, value):
        
        id = int(self.id)*10 + len(self.parmObjList) + 1
        
        parmObj = ParameterConf()
        parmObj.setup(id, name, value)
        
        self.parmObjList.append(parmObj)
        
        return parmObj
    
    def makeXml(self, upElement):
        
        opElement = SubElement(upElement, 'Operation')
        opElement.set('id', str(self.id))
        opElement.set('name', self.name)
        opElement.set('priority', str(self.priority))
        
        for parmObj in self.parmObjList:
            parmObj.makeXml(opElement)
            
    def readXml(self, opElement):
        
        self.id = opElement.get('id')
        self.name = opElement.get('name')
        self.type = opElement.get('type')
        
        self.parmObjList = []
        
        parmElementList = opElement.getiterator('Parameter')
        
        for parmElement in parmElementList:
            parmObj = ParameterConf()
            parmObj.readXml(parmElement)
            self.parmObjList.append(parmObj)
            
    def getParameterObjList(self):
        
        return self.parmObjList
    
    def printattr(self):
        
        print "Operation[%s]: name = %s, type = %s, priority = %s" %(self.id,
                                                                    self.name,
                                                                    self.type,
                                                                    self.priority)
        
        for parmObj in self.parmObjList:
            parmObj.printattr()
    
class ParameterConf():
    
    id = None
    name = None
    value = None
    
    def __init__(self):
        pass
    
    def setup(self, id, name, value):
        
        self.id = id
        self.name = name
        self.value = value
    
    def makeXml(self, opElement):
        
        parmElement = SubElement(opElement, 'Parameter')
        parmElement.set('id', str(self.id))
        parmElement.set('name', self.name)
        parmElement.set('value', self.value)
    
    def readXml(self, parmElement):
        
        self.id = parmElement.get('id')
        self.name = parmElement.get('name')
        self.value = parmElement.get('value')
    
    def printattr(self):
        
        print "Parameter[%s]: name = %s, value = %s" %(self.id, self.name, self.value)
        
if __name__ == '__main__':
    
    desc = "Este es un test"
    filename = "test.xml"
    
    projectObj = Controller()
    
    projectObj.setParms(id = '191', name='test01', description=desc)
    
    readBranchObj = projectObj.addReadBranch(dpath='mydata', dataformat='rawdata', readMode=0, startDate='1', endDate='3', startTime='4', endTime='5')
    
    procBranchObj = projectObj.addProcBranch(name='Branch1')
    
    procBranchObj1 = projectObj.addProcBranch(name='Branch2')
    
    upObj1 = procBranchObj.addUP(name='UP1', type='Voltage', inputId=readBranchObj.id)
    upObj2 = procBranchObj.addUP(name='UP2', type='Voltage', inputId=upObj1.id)
    
    opObj11 = upObj1.addOperation(name='removeDC', priority=1)
    opObj11.addParameter(name='type', value='1')
    
    opObj12 = upObj1.addOperation(name='decoder', priority=2)
    opObj12.addParameter(name='ncode', value='2')
    opObj12.addParameter(name='nbauds', value='8')
    opObj12.addParameter(name='code0', value='001110011')
    opObj12.addParameter(name='code1', value='001110011')
    
    print "Escribiendo el archivo XML"
    
    projectObj.writeXml(filename)
    
    print "Leyendo el archivo XML"
    projectObj.readXml(filename)
    projectObj.printattr()