Запись дерева каталогов в XML-файл с помощью Python LXML - PullRequest
Новая
       5

Запись дерева каталогов в XML-файл с помощью Python LXML

2 голосов
/ 06 февраля 2012

Я пытаюсь прочитать дерево каталогов, чтобы записать его в XML-файл без особого успеха: 

    # -*- 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))

Все подкаталоги становятся "корневыми детьми": 

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

где Amerique, France, Asie должны быть подкаталогами кухни.

Спасибо за вашу помощь. Jean-Patrick

1 Ответ

3 голосов
/ 06 февраля 2012

Вам необходимо отслеживать родительские папки и смотреть, куда вам нужно добавить каждый пройденный каталог. 

# -*- 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>
...