2012-07-05 12 views
12

Sto analizzando un file xml generato da un esterno program. Vorrei quindi aggiungere annotazioni personalizzate a questo file, utilizzando il mio spazio dei nomi. Il mio ingresso appare come di seguito:lxml: aggiungi spazio nomi al file di input

<sbml xmlns="http://www.sbml.org/sbml/level2/version4" xmlns:celldesigner="http://www.sbml.org/2001/ns/celldesigner" level="2" version="4"> 
    <model metaid="untitled" id="untitled"> 
    <annotation>...</annotation> 
    <listOfUnitDefinitions>...</listOfUnitDefinitions> 
    <listOfCompartments>...</listOfCompartments> 
    <listOfSpecies> 
     <species metaid="s1" id="s1" name="GenA" compartment="default" initialAmount="0"> 
     <annotation> 
      <celldesigner:extension>...</celldesigner:extension> 
     </annotation> 
     </species> 
     <species metaid="s2" id="s2" name="s2" compartment="default" initialAmount="0"> 
     <annotation> 
      <celldesigner:extension>...</celldesigner:extension> 
     </annotation> 
     </species> 
    </listOfSpecies> 
    <listOfReactions>...</listOfReactions> 
    </model> 
</sbml> 

Il problema è che lxml dichiara solo spazi dei nomi quando sono usati, il che significa che la dichiarazione è ripetuto molte volte, in questo modo (semplificato):

<sbml xmlns="namespace" xmlns:celldesigner="morenamespace" level="2" version="4"> 
    <listOfSpecies> 
    <species> 
     <kjw:test xmlns:kjw="http://this.is.some/custom_namespace"/> 
     <celldesigner:data>Some important data which must be kept</celldesigner:data> 
    </species> 
    <species> 
     <kjw:test xmlns:kjw="http://this.is.some/custom_namespace"/> 
    </species> 
    .... 
    </listOfSpecies> 
</sbml> 

E ' possibile forzare lxml a scrivere questa dichiarazione solo una volta in un elemento padre, ad esempio sbml o listOfSpecies? O c'è una buona ragione per non farlo? Il risultato che voglio sarebbe:

<sbml xmlns="namespace" xmlns:celldesigner="morenamespace" level="2" version="4" xmlns:kjw="http://this.is.some/custom_namespace"> 
    <listOfSpecies> 
    <species> 
     <kjw:test/> 
     <celldesigner:data>Some important data which must be kept</celldesigner:data> 
    </species> 
    <species> 
     <kjw:test/> 
    </species> 
    .... 
    </listOfSpecies> 
</sbml> 

Il problema importante è che i dati esistenti che viene letto da un file deve essere mantenuta, quindi non posso solo fare un nuovo elemento principale (credo?).

MODIFICA: Codice allegato sotto.

def annotateSbml(sbml_input): 
    from lxml import etree 

    checkSbml(sbml_input) # Makes sure the input is valid sbml/xml. 

    ns = "http://this.is.some/custom_namespace" 
    etree.register_namespace('kjw', ns) 

    sbml_doc = etree.ElementTree() 
    root = sbml_doc.parse(sbml_input, etree.XMLParser(remove_blank_text=True)) 
    nsmap = root.nsmap 
    nsmap['sbml'] = nsmap[None] # Makes code more readable, but seems ugly. Any alternatives to this? 
    nsmap['kjw'] = ns 
    ns = '{' + ns + '}' 
    sbmlns = '{' + nsmap['sbml'] + '}' 

    for species in root.findall('sbml:model/sbml:listOfSpecies/sbml:species', nsmap): 
    species.append(etree.Element(ns + 'test')) 

    sbml_doc.write("test.sbml.xml", pretty_print=True, xml_declaration=True) 

    return 
+1

Mostrare il codice. – Marcin

+0

@Marcin: done. Qualche consiglio? – kai

+0

@mzjin my input contiene tutto tranne i tag ''. L'obiettivo è inserire tali tag (o simili, ad esempio 'kjw: score' o' kjw: length') in ogni specie in questa lista. Questo ha senso, o dovrei postare l'intero file (pensavo che la mia domanda iniziale fosse abbastanza lunga così com'è)? – kai

risposta

8

La modifica della mappatura dello spazio dei nomi di un nodo non è possibile in lxml. Vedere this open ticket che ha questa caratteristica come elemento della lista di desideri.

È originato da this thread nella mailing list lxml, dove viene fornito un workaround replacing the root node in alternativa. Ci sono alcuni problemi con la sostituzione del nodo radice: vedi il ticket qui sopra.

metto il codice di soluzione sostituzione della radice suggerito per completezza:

>>> DOC = """<sbml xmlns="http://www.sbml.org/sbml/level2/version4" xmlns:celldesigner="http://www.sbml.org/2001/ns/celldesigner" level="2" version="4"> 
... <model metaid="untitled" id="untitled"> 
...  <annotation>...</annotation> 
...  <listOfUnitDefinitions>...</listOfUnitDefinitions> 
...  <listOfCompartments>...</listOfCompartments> 
...  <listOfSpecies> 
...  <species metaid="s1" id="s1" name="GenA" compartment="default" initialAmount="0"> 
...   <annotation> 
...   <celldesigner:extension>...</celldesigner:extension> 
...   </annotation> 
...  </species> 
...  <species metaid="s2" id="s2" name="s2" compartment="default" initialAmount="0"> 
...   <annotation> 
...   <celldesigner:extension>...</celldesigner:extension> 
...   </annotation> 
...  </species> 
...  </listOfSpecies> 
...  <listOfReactions>...</listOfReactions> 
... </model> 
... </sbml>""" 
>>> 
>>> from lxml import etree 
>>> from StringIO import StringIO 
>>> NS = "http://this.is.some/custom_namespace" 
>>> tree = etree.ElementTree(element=None, file=StringIO(DOC)) 
>>> root = tree.getroot() 
>>> nsmap = root.nsmap 
>>> nsmap['kjw'] = NS 
>>> new_root = etree.Element(root.tag, nsmap=nsmap) 
>>> new_root[:] = root[:] 
>>> new_root.append(etree.Element('{%s}%s' % (NS, 'test'))) 
>>> new_root.append(etree.Element('{%s}%s' % (NS, 'test'))) 

>>> print etree.tostring(new_root, pretty_print=True) 
<sbml xmlns:celldesigner="http://www.sbml.org/2001/ns/celldesigner" xmlns:kjw="http://this.is.some/custom_namespace" xmlns="http://www.sbml.org/sbml/level2/version4"><model metaid="untitled" id="untitled"> 
    <annotation>...</annotation> 
    <listOfUnitDefinitions>...</listOfUnitDefinitions> 
    <listOfCompartments>...</listOfCompartments> 
    <listOfSpecies> 
     <species metaid="s1" id="s1" name="GenA" compartment="default" initialAmount="0"> 
     <annotation> 
      <celldesigner:extension>...</celldesigner:extension> 
     </annotation> 
     </species> 
     <species metaid="s2" id="s2" name="s2" compartment="default" initialAmount="0"> 
     <annotation> 
      <celldesigner:extension>...</celldesigner:extension> 
     </annotation> 
     </species> 
    </listOfSpecies> 
    <listOfReactions>...</listOfReactions> 
    </model> 
<kjw:test/><kjw:test/></sbml> 
+1

Per riferimento futuro, ciò richiede una piccola modifica (almeno su Python 3.2), altrimenti restituisce TypeError da '** root.nsmap' quando tocca" None: 'namespace'' dato che 'None' non è una stringa. Usando 'nsmap = root.nsmap;' 'nsmap ['kjw'] = NS;' 'new_root = etree.Element (root.tag, nsmap = nsmap);' funziona. – kai

+0

nice catch, aggiornato – jterrace

+0

è inoltre necessario copiare attributi, testo e (improbabile, ma solo per completezza) coda. 'nsmap = dict (kjw = NS, nsmap = nsmap))' è sbagliato; dovrebbe essere solo 'nsmap = nsmap' – jfs

0

È possibile sostituire l'elemento radice per aggiungere "kjw" alla sua nsmap. Quindi la dichiarazione xmlns sarebbe solo nell'elemento root.

3

Piuttosto che fare direttamente con il codice XML grezzo si potrebbe anche guardare verso LibSBML, una libreria per la manipolazione di documenti SBML con attacchi di lingua per , tra gli altri, python. Lì lo utilizzeresti in questo modo:

 
>>> from libsbml import * 
>>> doc = readSBML('Dropbox/SBML Models/BorisEJB.xml') 
>>> species = doc.getModel().getSpecies('MAPK') 
>>> species.appendAnnotation('<kjw:test xmlns:kjw="http://this.is.some/custom_namespace"/>') 
0 
>>> species.toSBML() 
'<species id="MAPK" compartment="compartment" initialConcentration="280" boundaryCondition="false">\n <annotation>\n 
<kjw:test xmlns:kjw="http://this.is.some/custom_namespace"/>\n </annotation>\n</species>' 
>>> 

1

Se aggiungi temporaneamente un attributo di namespace al nodo radice, questo è il trucco.

ns = '{http://this.is.some/custom_namespace}' 

# add 'kjw:foobar' attribute to root node 
root.set(ns+'foobar', 'foobar') 

# add kjw namespace elements (or attributes) elsewhere 
... get child element species ... 
species.append(etree.Element(ns + 'test')) 

# remove temporary namespaced attribute from root node 
del root.attrib[ns+'foobar'] 
1

So che questa è vecchia questione, ma è ancora valido e, come di lxml 3.5.0, non v'è probabilmente migliore soluzione a questo problema:

cleanup_namespaces() accetta un argomento nuovo che si muove top_nsmap definizioni della mappatura dello spazio dei nomi prefisso fornita nella parte superiore dell'albero.

Così ora la mappa namespace può essere spostata con semplice chiamata a questo:

nsmap = {'kjw': 'http://this.is.some/custom_namespace'} 
etree.cleanup_namespaces(root, top_nsmap=nsmap) 
Problemi correlati