與蟒蛇LXML一個XML文件中寫入一個目錄樹

Question

我想讀一個目錄樹將它寫在一個XML文件，而無需太多sucess：

# -*- coding: utf-8 -*- 
""" 
Created on Tue Jan 31 13:30:22 2012 

@author: Jean-Patrick Pommier 
""" 
import lxml.etree as et 
import os 
'''' 
Lire l'arboresence d'un répertoire projet 
        P 
       /| \ 
       A B C 
      /|\ |\ |\ 
       a b ca b c e 
Stocker dans un fichier xml 
<P> 
    <A> 
     <a> 
     <b> 
     <c> 
    </A> 
    <B> 
     <a> 
     <b> 
    </B> 
    <C> 
     <c> 
     <e> 
    </C> 
</P> 
''' 
def makeNodes(parentxml,leveldirlist): 
     #print 'parent',parentxml 
     print 'chidren',leveldirlist 
     for d in leveldirlist: 
      child=et.Element(d) 
      parentxml.append(child) 

if __name__ == '__main__': 
    topdir='/home/claire/Applications/ProjetPython/testxml/biblio' 
    projetxml=et.Element('Project')#racine  
    parent=projetxml 

    for roots, dirs, files in os.walk(topdir): 
     print roots#, '*',dirs, '*',files,'\n' 
     makeNodes(parent,dirs) 

    print(et.tostring(projetxml,pretty_print=True)) 

所有子目錄成爲「根的chidren」：

<Project> 
    <Roman/> 
    <Cuisine/> 
    <Essais/> 
    <Science/> 
    <r20s/> 
    <r19s/> 
    <Amerique/> 
    <France/> 
    <Asie/> 
    <Religion/> 
    <Politique/> 
    <maths/> 
    <physique/> 
</Project> 

在哪裏，法國阿美里克，應該是亞洲烹飪的傳統。

謝謝你的幫助。 讓 - 帕特里克

Answer 1

你需要跟蹤父文件夾，並觀察你需要添加每個walked目錄的位置。

# -*- coding: utf-8 -*- 
import lxml.etree as et 
import os 
def makeNodes(current, parents, leveldirlist): 
    new = {} 
    for d in leveldirlist: 
     child=et.Element(d) 
     new[os.path.join(current, d)] = child 
     parents[current].append(child) 
    return new 

if __name__ == '__main__': 
    topdir='t1' 
    projectxml=et.Element('Project')  

    parents = {topdir: projectxml} 
    for current, dirs, files in os.walk(topdir): 
     parents.update(makeNodes(current, parents, dirs)) 

    print(et.tostring(projectxml,pretty_print=True)) 

這產生：

<Project> 
    <t2> 
    <t6/> 
    </t2> 
    <t3> 
    <t5/> 
    <t4> 
     <t7/> 
    </t4> 
    </t3> 
</Project>